binding/mdsal-binding-java-api-generator/src/test/java/org/opendaylight/mdsal/binding/java/api/generator/test/Bug532Test.java

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