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
  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.HashSet;
  27 import java.util.List;
  28 import org.junit.Test;
  29 import org.mockito.Mockito;
  30 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  31 import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
  32 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  33
  34 /**
  35  * Test correct functionality of copy constructor of generated builder classes.
  36  */
  37 public class Bug532Test extends BaseCompilationTest {
  38
  39     @Test
  40     public void test() throws Exception {
  41         final File sourcesOutputDir = new File(GENERATOR_OUTPUT_PATH + FS + "bug532");
  42         assertTrue("Failed to create test file '" + sourcesOutputDir + "'", sourcesOutputDir.mkdir());
  43         final File compiledOutputDir = new File(COMPILER_OUTPUT_PATH + FS + "bug532");
  44         assertTrue("Failed to create test file '" + compiledOutputDir + "'", compiledOutputDir.mkdir());
  45
  46         generateTestSources("/compilation/list-gen", sourcesOutputDir);
  47
  48         // Test if sources are compilable
  49         testCompilation(sourcesOutputDir, compiledOutputDir);
  50
  51         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
  52         Class<?> linksKeyClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.LinksKey", true, loader);
  53         Class<?> linksClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.Links", true, loader);
  54         Class<?> linksBuilderClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.LinksBuilder", true,
  55                 loader);
  56         Class<?> levelClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.Level", true, loader);
  57         Class<?> nodeClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.Node", true, loader);
  58         Class<?> nodeListClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.links.NodeList", true,
  59                 loader);
  60
  61         // init default values
  62         Byte expectedId = Byte.valueOf("5");
  63         String expectedName = "test-link";
  64         Integer expectedSize = Integer.valueOf(10);
  65         Object expectedLevel = Mockito.mock(levelClass);
  66         Integer expectedLinksId = Integer.valueOf(11);
  67         Object expectedNode = Mockito.mock(nodeClass);
  68         List<?> expectedNodeList = Lists.newArrayList(Mockito.mock(nodeListClass), Mockito.mock(nodeListClass));
  69         Constructor<?> keyConstructor = linksKeyClass.getDeclaredConstructor(Byte.class, String.class, Integer.class);
  70         Object expectedKey = keyConstructor.newInstance(expectedId, expectedName, expectedSize);
  71
  72         // create Links object
  73         Object linksBuilder = linksBuilderClass.newInstance();
  74         linksBuilderClass.getDeclaredMethod("setKey", linksKeyClass).invoke(linksBuilder, expectedKey);
  75         linksBuilderClass.getDeclaredMethod("setLevel", levelClass).invoke(linksBuilder, expectedLevel);
  76         linksBuilderClass.getDeclaredMethod("setLinksId", Integer.class).invoke(linksBuilder, expectedLinksId);
  77         linksBuilderClass.getDeclaredMethod("setNode", nodeClass).invoke(linksBuilder, expectedNode);
  78         linksBuilderClass.getDeclaredMethod("setNodeList", List.class).invoke(linksBuilder, expectedNodeList);
  79         Object links = linksBuilderClass.getDeclaredMethod("build").invoke(linksBuilder);
  80
  81         // create LinksBuilder object with constructor with Links object
  82         // argument
  83         Constructor<?> linksBuilderConstructor = linksBuilderClass.getDeclaredConstructor(linksClass);
  84         assertNotNull(linksBuilderConstructor);
  85         Object linksBuilderTested = linksBuilderConstructor.newInstance(links);
  86
  87         // get values from LinksBuilder
  88         Object actualKey = linksBuilderClass.getDeclaredMethod("getKey").invoke(linksBuilderTested);
  89         Object actualId = linksBuilderClass.getDeclaredMethod("getId").invoke(linksBuilderTested);
  90         Object actualName = linksBuilderClass.getDeclaredMethod("getName").invoke(linksBuilderTested);
  91         Object actualSize = linksBuilderClass.getDeclaredMethod("getSize").invoke(linksBuilderTested);
  92         Object actualLevel = linksBuilderClass.getDeclaredMethod("getLevel").invoke(linksBuilderTested);
  93         Object actualLinksId = linksBuilderClass.getDeclaredMethod("getLinksId").invoke(linksBuilderTested);
  94         Object actualNode = linksBuilderClass.getDeclaredMethod("getNode").invoke(linksBuilderTested);
  95         Object actualNodeList = linksBuilderClass.getDeclaredMethod("getNodeList").invoke(linksBuilderTested);
  96
  97         // test
  98         assertEquals(expectedKey, actualKey);
  99         assertEquals(expectedId, actualId);
 100         assertEquals(expectedName, actualName);
 101         assertEquals(expectedSize, actualSize);
 102         assertEquals(expectedLevel, actualLevel);
 103         assertEquals(expectedLinksId, actualLinksId);
 104         assertEquals(expectedNode, actualNode);
 105         assertEquals(expectedNodeList, actualNodeList);
 106
 107         cleanUp(sourcesOutputDir, compiledOutputDir);
 108     }
 109
 110     private void generateTestSources(String resourceDirPath, File sourcesOutputDir) throws Exception {
 111         final List<File> sourceFiles = getSourceFiles(resourceDirPath);
 112         final SchemaContext context = parser.parseFiles(sourceFiles);
 113         final List<Type> types = bindingGenerator.generateTypes(context);
 114         final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
 115         generator.generateToFile(sourcesOutputDir);
 116     }
 117
 118 }