code-generator/binding-java-api-generator/src/test/java/org/opendaylight/yangtools/sal/java/api/generator/test/Bug532Test.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.sal.java.api.generator.test;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.BASE_PKG;
  14 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.COMPILER_OUTPUT_PATH;
  15 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.FS;
  16 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.GENERATOR_OUTPUT_PATH;
  17 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.cleanUp;
  18 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.getSourceFiles;
  19 import static org.opendaylight.yangtools.sal.java.api.generator.test.CompilationTestUtils.testCompilation;
  20 import com.google.common.collect.ImmutableSet;
  21 import com.google.common.collect.Lists;
  22 import java.io.File;
  23 import java.lang.reflect.Constructor;
  24 import java.net.URL;
  25 import java.net.URLClassLoader;
  26 import java.util.List;
  27 import org.junit.Test;
  28 import org.mockito.Mockito;
  29 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  30 import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
  31 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  32
  33 /**
  34  * Test correct functionality of copy constructor of generated builder classes.
  35  */
  36 public class Bug532Test extends BaseCompilationTest {
  37
  38     @Test
  39     public void test() throws Exception {
  40         final File sourcesOutputDir = new File(GENERATOR_OUTPUT_PATH + FS + "bug532");
  41         assertTrue("Failed to create test file '" + sourcesOutputDir + "'", sourcesOutputDir.mkdir());
  42         final File compiledOutputDir = new File(COMPILER_OUTPUT_PATH + FS + "bug532");
  43         assertTrue("Failed to create test file '" + compiledOutputDir + "'", compiledOutputDir.mkdir());
  44
  45         generateTestSources("/compilation/list-gen", sourcesOutputDir);
  46
  47         // Test if sources are compilable
  48         testCompilation(sourcesOutputDir, compiledOutputDir);
  49
  50         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
  51         Class<?> linksKeyClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.LinksKey", true, loader);
  52         Class<?> linksClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.Links", true, loader);
  53         Class<?> linksBuilderClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.LinksBuilder", true,
  54                 loader);
  55         Class<?> levelClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.Level", true, loader);
  56         Class<?> nodeClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.Node", true, loader);
  57         Class<?> nodeListClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.NodeList", true,
  58                 loader);
  59
  60         // init default values
  61         Byte expectedId = Byte.valueOf("5");
  62         String expectedName = "test-link";
  63         Integer expectedSize = Integer.valueOf(10);
  64         Object expectedLevel = Mockito.mock(levelClass);
  65         Integer expectedLinksId = Integer.valueOf(11);
  66         Object expectedNode = Mockito.mock(nodeClass);
  67         List<?> expectedNodeList = Lists.newArrayList(Mockito.mock(nodeListClass), Mockito.mock(nodeListClass));
  68         Constructor<?> keyConstructor = linksKeyClass.getDeclaredConstructor(Byte.class, String.class, Integer.class);
  69         Object expectedKey = keyConstructor.newInstance(expectedId, expectedName, expectedSize);
  70
  71         // create Links object
  72         Object linksBuilder = linksBuilderClass.newInstance();
  73         linksBuilderClass.getDeclaredMethod("setKey", linksKeyClass).invoke(linksBuilder, expectedKey);
  74         linksBuilderClass.getDeclaredMethod("setLevel", levelClass).invoke(linksBuilder, expectedLevel);
  75         linksBuilderClass.getDeclaredMethod("setLinksId", Integer.class).invoke(linksBuilder, expectedLinksId);
  76         linksBuilderClass.getDeclaredMethod("setNode", nodeClass).invoke(linksBuilder, expectedNode);
  77         linksBuilderClass.getDeclaredMethod("setNodeList", List.class).invoke(linksBuilder, expectedNodeList);
  78         Object links = linksBuilderClass.getDeclaredMethod("build").invoke(linksBuilder);
  79
  80         // create LinksBuilder object with constructor with Links object
  81         // argument
  82         Constructor<?> linksBuilderConstructor = linksBuilderClass.getDeclaredConstructor(linksClass);
  83         assertNotNull(linksBuilderConstructor);
  84         Object linksBuilderTested = linksBuilderConstructor.newInstance(links);
  85
  86         // get values from LinksBuilder
  87         Object actualKey = linksBuilderClass.getDeclaredMethod("getKey").invoke(linksBuilderTested);
  88         Object actualId = linksBuilderClass.getDeclaredMethod("getId").invoke(linksBuilderTested);
  89         Object actualName = linksBuilderClass.getDeclaredMethod("getName").invoke(linksBuilderTested);
  90         Object actualSize = linksBuilderClass.getDeclaredMethod("getSize").invoke(linksBuilderTested);
  91         Object actualLevel = linksBuilderClass.getDeclaredMethod("getLevel").invoke(linksBuilderTested);
  92         Object actualLinksId = linksBuilderClass.getDeclaredMethod("getLinksId").invoke(linksBuilderTested);
  93         Object actualNode = linksBuilderClass.getDeclaredMethod("getNode").invoke(linksBuilderTested);
  94         Object actualNodeList = linksBuilderClass.getDeclaredMethod("getNodeList").invoke(linksBuilderTested);
  95
  96         // test
  97         assertEquals(expectedKey, actualKey);
  98         assertEquals(expectedId, actualId);
  99         assertEquals(expectedName, actualName);
 100         assertEquals(expectedSize, actualSize);
 101         assertEquals(expectedLevel, actualLevel);
 102         assertEquals(expectedLinksId, actualLinksId);
 103         assertEquals(expectedNode, actualNode);
 104         assertEquals(expectedNodeList, actualNodeList);
 105
 106         cleanUp(sourcesOutputDir, compiledOutputDir);
 107     }
 108
 109     private void generateTestSources(final String resourceDirPath, final File sourcesOutputDir) throws Exception {
 110         final List<File> sourceFiles = getSourceFiles(resourceDirPath);
 111         final SchemaContext context = parser.parseFiles(sourceFiles);
 112         final List<Type> types = bindingGenerator.generateTypes(context);
 113         final GeneratorJavaFile generator = new GeneratorJavaFile(ImmutableSet.copyOf(types));
 114         generator.generateToFile(sourcesOutputDir);
 115     }
 116
 117 }