2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.sal.java.api.generator.test;
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;
21 import com.google.common.collect.ImmutableSet;
22 import com.google.common.collect.Lists;
24 import java.lang.reflect.Constructor;
26 import java.net.URLClassLoader;
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 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
36 * Test correct functionality of copy constructor of generated builder classes.
38 public class Bug532Test extends BaseCompilationTest {
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());
47 generateTestSources("/compilation/list-gen-test", sourcesOutputDir);
49 // Test if sources are compilable
50 testCompilation(sourcesOutputDir, compiledOutputDir);
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,
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,
62 // init default values
63 Byte expectedId = Byte.valueOf("5");
64 String expectedName = "test-link";
65 Integer expectedSize = 10;
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);
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);
84 // create LinksBuilder object with constructor with Links object
86 Constructor<?> linksBuilderConstructor = linksBuilderClass.getDeclaredConstructor(linksClass);
87 assertNotNull(linksBuilderConstructor);
88 Object linksBuilderTested = linksBuilderConstructor.newInstance(links);
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);
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);
110 cleanUp(sourcesOutputDir, compiledOutputDir);
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);