Added getParent() method to DataSchemaNode and DataNodeContainer. Fixed Bugs.
[mdsal.git] / code-generator / binding-java-api-generator / src / test / java / org / opendaylight / yangtools / sal / java / api / generator / test / CompilationTest.java
index 0581b95ddc96f73d50f2923ac41f0c55fcc83230..59d4523affb5e4e93104a1f994350b16f8c5c70c 100644 (file)
@@ -43,14 +43,17 @@ import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
 
 /**
  * Test correct code generation.
- * 
+ *
  */
 public class CompilationTest {
     private static final String FS = File.separator;
-    private static final String NS_TEST = "org" + FS + "opendaylight" + FS + "yang" + FS + "gen" + FS + "v1" + FS
-            + "urn" + FS + "opendaylight" + FS + "test" + FS + "rev131008";
-    private static final String NS_FOO = "org" + FS + "opendaylight" + FS + "yang" + FS + "gen" + FS + "v1" + FS
-            + "urn" + FS + "opendaylight" + FS + "foo" + FS + "rev131008";
+    private static final String BASE_PATH = "org" + FS + "opendaylight" + FS + "yang" + FS + "gen" + FS + "v1";
+    private static final String NS_TEST = BASE_PATH + FS + "urn" + FS + "opendaylight" + FS + "test" + FS + "rev131008";
+    private static final String NS_FOO = BASE_PATH + FS + "urn" + FS + "opendaylight" + FS + "foo" + FS + "rev131008";
+    private static final String NS_BAR = BASE_PATH + FS + "urn" + FS + "opendaylight" + FS + "bar" + FS + "rev131008";
+    private static final String NS_BAZ = BASE_PATH + FS + "urn" + FS + "opendaylight" + FS + "baz" + FS + "rev131008";
+
+    private static final String BASE_PKG = "org.opendaylight.yang.gen.v1";
 
     private static final String TEST_PATH = "target" + FS + "test";
     private static final File TEST_DIR = new File(TEST_PATH);
@@ -62,6 +65,8 @@ public class CompilationTest {
 
     private YangParserImpl parser;
     private BindingGenerator bindingGenerator;
+    private JavaCompiler compiler;
+    private StandardJavaFileManager fileManager;
 
     @BeforeClass
     public static void createTestDirs() {
@@ -76,6 +81,8 @@ public class CompilationTest {
     public void init() {
         parser = new YangParserImpl();
         bindingGenerator = new BindingGeneratorImpl();
+        compiler = ToolProvider.getSystemJavaCompiler();
+        fileManager = compiler.getStandardFileManager(null, null, null);
     }
 
     /**
@@ -107,32 +114,63 @@ public class CompilationTest {
 
         // Test if all sources are generated
         File parent = new File(sourcesOutputDir, NS_TEST);
-        File linksKeyFile = new File(parent, "LinksKey.java");
-        assertTrue(new File(parent, "KeyArgs.java").exists());
-        assertTrue(new File(parent, "Links.java").exists());
-        assertTrue(new File(parent, "LinksBuilder.java").exists());
-        assertTrue(linksKeyFile.exists());
-        assertTrue(new File(parent, "TestData.java").exists());
-        assertTrue(new File(parent, "links" + FS + "Level.java").exists());
-        assertTrue(new File(parent, "links" + FS + "LinkGroup.java").exists());
-        assertTrue(new File(parent, "links" + FS + "Node.java").exists());
-        assertTrue(new File(parent, "links" + FS + "NodeBuilder.java").exists());
-        assertTrue(new File(parent, "links" + FS + "NodeList.java").exists());
-        assertTrue(new File(parent, "links" + FS + "NodeListBuilder.java").exists());
-        assertTrue(linksKeyFile.exists());
+        assertEquals(6, parent.listFiles().length);
+        File keyArgs = new File(parent, "KeyArgs.java");
+        File links = new File(parent, "Links.java");
+        File linksBuilder = new File(parent, "LinksBuilder.java");
+        File linksKey = new File(parent, "LinksKey.java");
+        File testData = new File(parent, "TestData.java");
+        assertTrue(keyArgs.exists());
+        assertTrue(links.exists());
+        assertTrue(linksBuilder.exists());
+        assertTrue(linksKey.exists());
+        assertTrue(testData.exists());
+
+        parent = new File(sourcesOutputDir, NS_TEST + FS + "links");
+        assertEquals(7, parent.listFiles().length);
+        File level = new File(parent, "Level.java");
+        File linkGroup = new File(parent, "LinkGroup.java");
+        File node = new File(parent, "Node.java");
+        File nodeBuilder = new File(parent, "NodeBuilder.java");
+        File nodeList = new File(parent, "NodeList.java");
+        File nodeListBuilder = new File(parent, "NodeListBuilder.java");
+        File nodesType = new File(parent, "NodesType.java");
+        assertTrue(level.exists());
+        assertTrue(linkGroup.exists());
+        assertTrue(node.exists());
+        assertTrue(nodeBuilder.exists());
+        assertTrue(nodeList.exists());
+        assertTrue(nodeListBuilder.exists());
+        assertTrue(nodesType.exists());
 
         // Test if sources are compilable
-        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
-        StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
-        List<File> filesList = getJavaFiles(sourcesOutputDir);
-        Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
-        Iterable<String> options = Arrays.asList("-d", compiledOutputDir.getAbsolutePath());
-        boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
-        assertTrue(compiled);
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
-        Class<?> linksKeyClass = Class.forName("org.opendaylight.yang.gen.v1.urn.opendaylight.test.rev131008.LinksKey",
-                true, loader);
+        Class<?> keyArgsClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.KeyArgs", true, loader);
+        Class<?> linksClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.Links", true, loader);
+        Class<?> linksKeyClass = Class.forName(BASE_PKG + ".urn.opendaylight.test.rev131008.LinksKey", true, loader);
+
+        // Test generated 'grouping key-args'
+        try {
+            assertTrue(keyArgsClass.isInterface());
+            assertEquals(3, keyArgsClass.getDeclaredMethods().length);
+
+            Method getId = keyArgsClass.getMethod("getId");
+            assertEquals(Byte.class, getId.getReturnType());
+            Method getName = keyArgsClass.getMethod("getName");
+            assertEquals(String.class, getName.getReturnType());
+            Method getSize = keyArgsClass.getMethod("getSize");
+            assertEquals(Integer.class, getSize.getReturnType());
+        } catch (NoSuchMethodException e) {
+            throw new AssertionError("Required method not found in " + keyArgsClass, e);
+        }
+
+        // test generated 'list links'
+        assertTrue(linksClass.isInterface());
+        // FIXME: anyxml
+        assertEquals(5, linksClass.getDeclaredMethods().length);
+        testImplementIfc(linksClass, keyArgsClass);
 
         // Test list key constructor arguments ordering
         try {
@@ -158,26 +196,218 @@ public class CompilationTest {
         final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
         generator.generateToFile(sourcesOutputDir);
 
-        // Test if all sources are generated
-        File parent = new File(sourcesOutputDir, NS_TEST);
+        // Test if all sources were generated from 'module foo'
+        File parent = new File(sourcesOutputDir, NS_FOO);
+        assertEquals(7, parent.listFiles().length);
         assertTrue(new File(parent, "Object.java").exists());
         assertTrue(new File(parent, "OpenObject.java").exists());
-        assertTrue(new File(parent, "object" + FS + "Nodes.java").exists());
-        assertTrue(new File(parent, "object" + FS + "NodesBuilder.java").exists());
-        assertTrue(new File(parent, "open" + FS + "object" + FS + "Nodes1.java").exists());
-        assertTrue(new File(parent, "open" + FS + "object" + FS + "Nodes1Builder.java").exists());
-        assertTrue(new File(parent, "open" + FS + "object" + FS + "nodes" + FS + "Links.java").exists());
-        assertTrue(new File(parent, "open" + FS + "object" + FS + "nodes" + FS + "LinksBuilder.java").exists());
+        assertTrue(new File(parent, "ExplicitRouteObject.java").exists());
+        assertTrue(new File(parent, "PathKeySubobject.java").exists());
+
+        parent = new File(parent, "object");
+        assertEquals(2, parent.listFiles().length);
+        assertTrue(new File(parent, "Nodes.java").exists());
+        assertTrue(new File(parent, "NodesBuilder.java").exists());
+
+        parent = new File(sourcesOutputDir, NS_FOO + FS + "open");
+        assertEquals(1, parent.listFiles().length);
+
+        parent = new File(parent, "object");
+        assertEquals(3, parent.listFiles().length);
+        assertTrue(new File(parent, "Nodes1.java").exists());
+        assertTrue(new File(parent, "Nodes1Builder.java").exists());
+
+        parent = new File(parent, "nodes");
+        assertEquals(2, parent.listFiles().length);
+        assertTrue(new File(parent, "Links.java").exists());
+        assertTrue(new File(parent, "LinksBuilder.java").exists());
+
+        parent = new File(sourcesOutputDir, NS_FOO + FS + "explicit");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "route");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "object");
+        assertEquals(3, parent.listFiles().length);
+        assertTrue(new File(parent, "Subobjects.java").exists());
+        assertTrue(new File(parent, "SubobjectsBuilder.java").exists());
+
+        parent = new File(parent, "subobjects");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "subobject");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "type");
+        assertEquals(3, parent.listFiles().length);
+        assertTrue(new File(parent, "PathKey.java").exists());
+        assertTrue(new File(parent, "PathKeyBuilder.java").exists());
+
+        parent = new File(parent, "path");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "key");
+        assertEquals(2, parent.listFiles().length);
+        assertTrue(new File(parent, "PathKey.java").exists());
+        assertTrue(new File(parent, "PathKeyBuilder.java").exists());
+
+        // Test if all sources were generated from 'module bar'
+        parent = new File(sourcesOutputDir, NS_BAR);
+        assertEquals(3, parent.listFiles().length);
+        assertTrue(new File(parent, "BasicExplicitRouteSubobjects.java").exists());
+        assertTrue(new File(parent, "ExplicitRouteSubobjects.java").exists());
+
+        parent = new File(parent, "basic");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "explicit");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "route");
+        assertEquals(1, parent.listFiles().length);
+
+        parent = new File(parent, "subobjects");
+        assertEquals(2, parent.listFiles().length);
+        assertTrue(new File(parent, "SubobjectType.java").exists());
+
+        parent = new File(parent, "subobject");
+        assertEquals(1, parent.listFiles().length);
+
+        parent = new File(parent, "type");
+        assertEquals(4, parent.listFiles().length);
+        assertTrue(new File(parent, "IpPrefix.java").exists());
+        assertTrue(new File(parent, "IpPrefixBuilder.java").exists());
+        assertTrue(new File(parent, "Label.java").exists());
+        assertTrue(new File(parent, "LabelBuilder.java").exists());
 
         // Test if sources are compilable
-        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
-        StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
-        List<File> filesList = getJavaFiles(sourcesOutputDir);
-        Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
-        Iterable<String> options = Arrays.asList("-d", compiledOutputDir.getAbsolutePath());
-        boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
-        assertTrue(compiled);
+        cleanUp(sourcesOutputDir, compiledOutputDir);
+    }
+
+    @Test
+    public void testAugmentOfAugmentGeneration() throws Exception {
+        final File sourcesOutputDir = new File(GENERATOR_OUTPUT_PATH + FS + "aug-of-aug");
+        assertTrue("Failed to create test file '" + sourcesOutputDir + "'", sourcesOutputDir.mkdir());
+        final File compiledOutputDir = new File(COMPILER_OUTPUT_PATH + FS + "aug-of-aug");
+        assertTrue("Failed to create test file '" + compiledOutputDir + "'", compiledOutputDir.mkdir());
+
+        final List<File> sourceFiles = getSourceFiles("/compilation/augment-of-augment");
+        final Set<Module> modulesToBuild = parser.parseYangModels(sourceFiles);
+        final SchemaContext context = parser.resolveSchemaContext(modulesToBuild);
+        final List<Type> types = bindingGenerator.generateTypes(context);
+        final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
+        generator.generateToFile(sourcesOutputDir);
+
+        // Test if all sources were generated from 'module foo'
+        File parent = new File(sourcesOutputDir, NS_FOO);
+        assertEquals(6, parent.listFiles().length);
+        File fooListener = new File(parent, "FooListener.java");
+        File pathAttributes = new File(parent, "PathAttributes.java");
+        File update = new File(parent, "Update.java");
+        File updateBuilder = new File(parent, "UpdateBuilder.java");
+        assertTrue(fooListener.exists());
+        assertTrue(pathAttributes.exists());
+        assertTrue(update.exists());
+        assertTrue(updateBuilder.exists());
+
+        parent = new File(sourcesOutputDir, NS_FOO + FS + "path");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "attributes");
+        assertEquals(2, parent.listFiles().length);
+        File origin = new File(parent, "Origin.java");
+        File originBuilder = new File(parent, "OriginBuilder.java");
+        assertTrue(origin.exists());
+        assertTrue(originBuilder.exists());
+
+        parent = new File(sourcesOutputDir, NS_FOO + FS + "update");
+        assertEquals(2, parent.listFiles().length);
+        pathAttributes = new File(parent, "PathAttributes.java");
+        File pathAttributesBuilder = new File(parent, "PathAttributesBuilder.java");
+        assertTrue(pathAttributes.exists());
+        assertTrue(pathAttributesBuilder.exists());
+
+        // Test if all sources were generated from 'module bar'
+        parent = new File(sourcesOutputDir, NS_BAR);
+        assertEquals(5, parent.listFiles().length);
+        File destination = new File(parent, "Destination.java");
+        File pathAttributes1 = new File(parent, "PathAttributes1.java");
+        File pathAttributes1Builder = new File(parent, "PathAttributes1Builder.java");
+        assertTrue(destination.exists());
+        assertTrue(pathAttributes1.exists());
+        assertTrue(pathAttributes1Builder.exists());
+
+        parent = new File(sourcesOutputDir, NS_BAR + FS + "destination");
+        assertEquals(2, parent.listFiles().length);
+        File destinationType = new File(parent, "DestinationType.java");
+        assertTrue(destinationType.exists());
+
+        parent = new File(parent, "destination");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "type");
+        assertEquals(2, parent.listFiles().length);
+        File destinationIpv4 = new File(parent, "DestinationIp.java");
+        File destinationIpv4Builder = new File(parent, "DestinationIpBuilder.java");
+        assertTrue(destinationIpv4.exists());
+        assertTrue(destinationIpv4Builder.exists());
+
+        parent = new File(sourcesOutputDir, NS_BAR + FS + "update");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "path");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "attributes");
+        assertEquals(3, parent.listFiles().length);
+        File mpUnreachNlri = new File(parent, "MpUnreachNlri.java");
+        File mpUnreachNlriBuilder = new File(parent, "MpUnreachNlriBuilder.java");
+        assertTrue(mpUnreachNlri.exists());
+        assertTrue(mpUnreachNlriBuilder.exists());
+
+        parent = new File(parent, "mp");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "unreach");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "nlri");
+        assertEquals(3, parent.listFiles().length);
+        File withdrawnRoutes = new File(parent, "WithdrawnRoutes.java");
+        File withdrawnRoutesBuilder = new File(parent, "WithdrawnRoutesBuilder.java");
+        assertTrue(withdrawnRoutes.exists());
+        assertTrue(withdrawnRoutesBuilder.exists());
+
+        parent = new File(parent, "withdrawn");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "routes");
+        assertEquals(1, parent.listFiles().length);
+        destinationType = new File(parent, "DestinationType.java");
+        assertTrue(destinationType.exists());
+
+        // Test if all sources were generated from 'module baz'
+        parent = new File(sourcesOutputDir, NS_BAZ);
+        assertEquals(2, parent.listFiles().length);
+        File linkstateDestination = new File(parent, "LinkstateDestination.java");
+        assertTrue(linkstateDestination.exists());
+
+        parent = new File(sourcesOutputDir, NS_BAZ + FS + "update");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "path");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "attributes");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "mp");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "unreach");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "nlri");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "withdrawn");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "routes");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "destination");
+        assertEquals(1, parent.listFiles().length);
+        parent = new File(parent, "type");
+        assertEquals(2, parent.listFiles().length);
+        File destinationLinkstate = new File(parent, "DestinationLinkstate.java");
+        File destinationLinkstateBuilder = new File(parent, "DestinationLinkstateBuilder.java");
+        assertTrue(destinationLinkstate.exists());
+        assertTrue(destinationLinkstateBuilder.exists());
+
+        // Test if sources are compilable
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
         cleanUp(sourcesOutputDir, compiledOutputDir);
     }
@@ -202,40 +432,34 @@ public class CompilationTest {
         assertTrue(new File(parent, "NodesBuilder.java").exists());
         assertTrue(new File(parent, "Alg.java").exists());
 
-        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
-        StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
-
-        List<File> filesList = getJavaFiles(sourcesOutputDir);
-        Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
-        Iterable<String> options = Arrays.asList("-d", compiledOutputDir.getAbsolutePath());
-        boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
-        assertTrue(compiled);
+        // Test if sources are compilable
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
-        String pkg = "org.opendaylight.yang.gen.v1.urn.opendaylight.test.rev131008";
+        String pkg = BASE_PKG + ".urn.opendaylight.test.rev131008";
         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
         Class<?> nodesClass = Class.forName(pkg + ".Nodes", true, loader);
 
         // Test methods return type
         byte[] b = new byte[] {};
         testReturnType(nodesClass, "getIdBinary", b.getClass());
-        testReturnType(loader, nodesClass, "getIdBits", pkg + ".Nodes$IdBits");
-        testReturnType(loader, nodesClass, "isIdBoolean", "java.lang.Boolean");
-        testReturnType(loader, nodesClass, "getIdDecimal64", "java.math.BigDecimal");
-        testReturnType(loader, nodesClass, "isIdEmpty", "java.lang.Boolean");
-        testReturnType(loader, nodesClass, "getIdEnumeration", pkg + ".Nodes$IdEnumeration");
+        testReturnType(nodesClass, "getIdBits", pkg + ".Nodes$IdBits", loader);
+        testReturnType(nodesClass, "isIdBoolean", "java.lang.Boolean", loader);
+        testReturnType(nodesClass, "getIdDecimal64", "java.math.BigDecimal", loader);
+        testReturnType(nodesClass, "isIdEmpty", "java.lang.Boolean", loader);
+        testReturnType(nodesClass, "getIdEnumeration", pkg + ".Nodes$IdEnumeration", loader);
         testReturnTypeIdentityref(nodesClass, "getIdIdentityref", pkg + ".Alg");
         testReturnTypeInstanceIdentitifer(loader, nodesClass, "getIdInstanceIdentifier");
-        testReturnType(loader, nodesClass, "getId8", "java.lang.Byte");
-        testReturnType(loader, nodesClass, "getId16", "java.lang.Short");
-        testReturnType(loader, nodesClass, "getId32", "java.lang.Integer");
-        testReturnType(loader, nodesClass, "getId64", "java.lang.Long");
-        testReturnType(loader, nodesClass, "getIdLeafref", "java.lang.Long");
-        testReturnType(loader, nodesClass, "getIdString", "java.lang.String");
-        testReturnType(loader, nodesClass, "getIdU8", "java.lang.Short");
-        testReturnType(loader, nodesClass, "getIdU16", "java.lang.Integer");
-        testReturnType(loader, nodesClass, "getIdU32", "java.lang.Long");
-        testReturnType(loader, nodesClass, "getIdU64", "java.math.BigInteger");
-        testReturnType(loader, nodesClass, "getIdUnion", pkg + ".Nodes$IdUnion");
+        testReturnType(nodesClass, "getId8", "java.lang.Byte", loader);
+        testReturnType(nodesClass, "getId16", "java.lang.Short", loader);
+        testReturnType(nodesClass, "getId32", "java.lang.Integer", loader);
+        testReturnType(nodesClass, "getId64", "java.lang.Long", loader);
+        testReturnType(nodesClass, "getIdLeafref", "java.lang.Long", loader);
+        testReturnType(nodesClass, "getIdString", "java.lang.String", loader);
+        testReturnType(nodesClass, "getIdU8", "java.lang.Short", loader);
+        testReturnType(nodesClass, "getIdU16", "java.lang.Integer", loader);
+        testReturnType(nodesClass, "getIdU32", "java.lang.Long", loader);
+        testReturnType(nodesClass, "getIdU64", "java.math.BigInteger", loader);
+        testReturnType(nodesClass, "getIdUnion", pkg + ".Nodes$IdUnion", loader);
 
         cleanUp(sourcesOutputDir, compiledOutputDir);
     }
@@ -256,25 +480,31 @@ public class CompilationTest {
 
         // Test if all sources are generated
         File fooParent = new File(sourcesOutputDir, NS_FOO);
-        File nodes = new File(fooParent, "Nodes.java");
-        File nodesBuilder = new File(fooParent, "NodesBuilder.java");
+        assertEquals(3, fooParent.listFiles().length);
         assertTrue(new File(fooParent, "FooData.java").exists());
-        assertTrue(nodes.exists());
-        assertTrue(nodesBuilder.exists());
+        assertTrue(new File(fooParent, "Nodes.java").exists());
+        assertTrue(new File(fooParent, "NodesBuilder.java").exists());
 
-        // Test if sources are compilable
-        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
-        StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
+        File barParent = new File(sourcesOutputDir, NS_BAR);
+        assertEquals(1, barParent.listFiles().length);
+        assertTrue(new File(barParent, "IdentityClass.java").exists());
 
-        List<File> filesList = getJavaFiles(sourcesOutputDir);
-        Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
-        Iterable<String> options = Arrays.asList("-d", compiledOutputDir.getAbsolutePath());
-        boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
-        assertTrue(compiled);
+        // Test if sources are compilable
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
         ClassLoader loader = new URLClassLoader(new URL[] { compiledOutputDir.toURI().toURL() });
-        Class<?> nodesClass = Class.forName("org.opendaylight.yang.gen.v1.urn.opendaylight.foo.rev131008.Nodes", true,
-                loader);
+        Class<?> nodesClass = Class.forName(BASE_PKG + ".urn.opendaylight.foo.rev131008.Nodes", true, loader);
+        Class<?> identityClass = Class
+                .forName(BASE_PKG + ".urn.opendaylight.bar.rev131008.IdentityClass", true, loader);
+
+        // test identity
+        try {
+            identityClass.getConstructor();
+            Class<?> baseIdentity = Class.forName("org.opendaylight.yangtools.yang.binding.BaseIdentity", true, loader);
+            assertEquals(baseIdentity, identityClass.getSuperclass());
+        } catch (NoSuchMethodException e) {
+            throw new AssertionError("IdentityClass must have no-arg constructor");
+        }
 
         // Test annotation
         try {
@@ -290,6 +520,37 @@ public class CompilationTest {
         cleanUp(sourcesOutputDir, compiledOutputDir);
     }
 
+    @Test
+    public void testAugmentToUsesInAugment() throws Exception {
+        final File sourcesOutputDir = new File(GENERATOR_OUTPUT_PATH + FS + "augment-uses-to-augment");
+        assertTrue("Failed to create test file '" + sourcesOutputDir + "'", sourcesOutputDir.mkdir());
+        final File compiledOutputDir = new File(COMPILER_OUTPUT_PATH + FS + "augment-uses-to-augment");
+        assertTrue("Failed to create test file '" + compiledOutputDir + "'", compiledOutputDir.mkdir());
+
+        final List<File> sourceFiles = getSourceFiles("/compilation/augment-uses-to-augment");
+        final Set<Module> modulesToBuild = parser.parseYangModels(sourceFiles);
+        final SchemaContext context = parser.resolveSchemaContext(modulesToBuild);
+        final List<Type> types = bindingGenerator.generateTypes(context);
+        final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
+        generator.generateToFile(sourcesOutputDir);
+
+        // Test if all sources are generated
+        File fooParent = new File(sourcesOutputDir, NS_FOO);
+        assertEquals(4, fooParent.listFiles().length);
+        assertTrue(new File(fooParent, "IgpLinkAttributes.java").exists());
+        assertTrue(new File(fooParent, "Link1.java").exists());
+        assertTrue(new File(fooParent, "Link1Builder.java").exists());
+
+        File bazParent = new File(sourcesOutputDir, NS_BAZ);
+        assertEquals(4, bazParent.listFiles().length);
+        assertTrue(new File(bazParent, "IgpLinkAttributes1.java").exists());
+        assertTrue(new File(bazParent, "IgpLinkAttributes1Builder.java").exists());
+        assertTrue(new File(bazParent, "LinkAttributes.java").exists());
+
+        // Test if sources are compilable
+        testCompilation(sourcesOutputDir, compiledOutputDir);
+    }
+
     @Test
     public void compilationTest() throws Exception {
         final File sourcesOutputDir = new File(GENERATOR_OUTPUT_PATH + FS + "yang");
@@ -304,16 +565,41 @@ public class CompilationTest {
         final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
         generator.generateToFile(sourcesOutputDir);
 
-        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
-        StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
+        // Test if sources are compilable
+        testCompilation(sourcesOutputDir, compiledOutputDir);
 
+        cleanUp(sourcesOutputDir, compiledOutputDir);
+    }
+
+    /**
+     * Test if source code is compilable.
+     *
+     * @param sourcesOutputDir
+     *            directory containing source files
+     * @param compiledOutputDir
+     *            compiler output directory
+     */
+    private void testCompilation(File sourcesOutputDir, File compiledOutputDir) {
         List<File> filesList = getJavaFiles(sourcesOutputDir);
         Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
         Iterable<String> options = Arrays.asList("-d", compiledOutputDir.getAbsolutePath());
         boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
         assertTrue(compiled);
+    }
 
-        cleanUp(sourcesOutputDir, compiledOutputDir);
+    private void testImplementIfc(Class<?> classToTest, String ifc, ClassLoader loader) throws ClassNotFoundException {
+        Class<?>[] interfaces = classToTest.getInterfaces();
+        List<Class<?>> ifcsList = Arrays.asList(interfaces);
+        Class<?> ifcClass = Class.forName(ifc, true, loader);
+        testImplementIfc(classToTest, ifcClass);
+    }
+
+    private void testImplementIfc(Class<?> classToTest, Class<?> ifcClass) throws ClassNotFoundException {
+        Class<?>[] interfaces = classToTest.getInterfaces();
+        List<Class<?>> ifcsList = Arrays.asList(interfaces);
+        if (!ifcsList.contains(ifcClass)) {
+            throw new AssertionError(classToTest + " should implement " + ifcClass);
+        }
     }
 
     private void testReturnType(Class<?> clazz, String methodName, Class<?> returnType) throws Exception {
@@ -326,7 +612,7 @@ public class CompilationTest {
         }
     }
 
-    private void testReturnType(ClassLoader loader, Class<?> clazz, String methodName, String returnTypeStr)
+    private void testReturnType(Class<?> clazz, String methodName, String returnTypeStr, ClassLoader loader)
             throws Exception {
         Class<?> returnType;
         try {