eb4e0104299455a6d89f0f9629dfdbf173febf01
[mdsal.git] / binding / mdsal-binding-java-api-generator / src / test / java / org / opendaylight / mdsal / binding / java / api / generator / test / UnionWithIdentityrefTest.java
1 /*
2  * Copyright (c) 2016 Intel corporation 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.assertArrayEquals;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNull;
13
14 import com.google.common.collect.ImmutableSet;
15 import java.io.File;
16 import java.lang.reflect.Constructor;
17 import java.lang.reflect.Method;
18 import java.net.URL;
19 import java.net.URLClassLoader;
20 import java.util.List;
21 import org.junit.Test;
22 import org.opendaylight.mdsal.binding.java.api.generator.GeneratorJavaFile;
23 import org.opendaylight.mdsal.binding.model.api.Type;
24 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
25 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
26
27 /**
28  * union constructor with indentityref
29  * previously identityref was ignored so that there is no constructor
30  * for indentityref
31  *
32  */
33 public class UnionWithIdentityrefTest extends BaseCompilationTest {
34
35     @Test
36     public void test() throws Exception {
37         final File sourcesOutputDir = CompilationTestUtils.generatorOutput("union-with-identityref");
38         final File compiledOutputDir = CompilationTestUtils.compilerOutput("union-with-identityref");
39         generateTestSources("/compilation/union-with-identityref", sourcesOutputDir);
40
41         // Test if sources are compilable
42         CompilationTestUtils.testCompilation(sourcesOutputDir, compiledOutputDir);
43
44         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
45         Class<?> identBaseClass = Class.forName(CompilationTestUtils.BASE_PKG + ".urn.opendaylight.yang.union.test.rev160509.IdentBase", true, loader);
46         Class<?> identOneClass = Class.forName(CompilationTestUtils.BASE_PKG + ".urn.opendaylight.yang.union.test.rev160509.IdentOne", true, loader);
47         Class<?> unionTypeClass = Class.forName(CompilationTestUtils.BASE_PKG + ".urn.opendaylight.yang.union.test.rev160509.UnionType", true, loader);
48
49         // test UnionType with IdentOne argument
50         Constructor<?> unionTypeIdentBaseConstructor = CompilationTestUtils.assertContainsConstructor(unionTypeClass, Class.class);
51         Object unionType = unionTypeIdentBaseConstructor.newInstance(identOneClass);
52
53         Method getUint8 = unionTypeClass.getDeclaredMethod("getUint8");
54         Object actualUint8 = getUint8.invoke(unionType);
55         assertNull(actualUint8);
56
57         Method getIdentityref = unionTypeClass.getDeclaredMethod("getIdentityref");
58         Object actualIdentityref = getIdentityref.invoke(unionType);
59         assertEquals(identOneClass, actualIdentityref);
60
61         Method getValue = unionTypeClass.getDeclaredMethod("getValue");
62         Object actualValue = getValue.invoke(unionType);
63         assertArrayEquals(identOneClass.toString().toCharArray(), (char[])actualValue);
64
65         CompilationTestUtils.cleanUp(sourcesOutputDir, compiledOutputDir);
66     }
67
68     private void generateTestSources(final String resourceDirPath, final File sourcesOutputDir) throws Exception {
69         final List<File> sourceFiles = CompilationTestUtils.getSourceFiles(resourceDirPath);
70         final SchemaContext context = YangParserTestUtils.parseYangFiles(sourceFiles);
71         final List<Type> types = bindingGenerator.generateTypes(context);
72         final GeneratorJavaFile generator = new GeneratorJavaFile(ImmutableSet.copyOf(types));
73         generator.generateToFile(sourcesOutputDir);
74     }
75 }