Bug 5085: Clean-up test and retest JUnit tests

[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug4456Test.java

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4456Test.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4456Test.java
new file mode 100644 (file)
index 0000000..01cea68
--- /dev/null
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4456Test.java
@@ -0,0 +1,78 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.yangtools.yang.stmt;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.fail;
+
+import java.io.IOException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.List;
+import java.util.Set;
+import org.junit.Test;
+import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
+import org.opendaylight.yangtools.yang.model.api.Module;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
+import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
+import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
+
+public class Bug4456Test {
+    @Test
+    public void test() throws IOException, URISyntaxException, SourceException, ReactorException {
+        SchemaContext schema = StmtTestUtils.parseYangSources("/bugs/bug4456");
+        assertNotNull(schema);
+
+        Set<Module> modules = schema.findModuleByNamespace(new URI("foo"));
+        assertEquals(1, modules.size());
+        Module moduleFoo = modules.iterator().next();
+
+        List<ExtensionDefinition> extensionSchemaNodes = moduleFoo.getExtensionSchemaNodes();
+        assertEquals(5, extensionSchemaNodes.size());
+        for (ExtensionDefinition extensionDefinition : extensionSchemaNodes) {
+
+            List<UnknownSchemaNode> unknownSchemaNodes = extensionDefinition.getUnknownSchemaNodes();
+            assertEquals(1, unknownSchemaNodes.size());
+            UnknownSchemaNode unknownSchemaNode = unknownSchemaNodes.iterator().next();
+            String unknownNodeExtensionDefName = unknownSchemaNode.getExtensionDefinition().getQName().getLocalName();
+
+            List<UnknownSchemaNode> subUnknownSchemaNodes = unknownSchemaNode.getUnknownSchemaNodes();
+            assertEquals(1, subUnknownSchemaNodes.size());
+            UnknownSchemaNode subUnknownSchemaNode = subUnknownSchemaNodes.iterator().next();
+            String subUnknownNodeExtensionDefName = subUnknownSchemaNode.getExtensionDefinition().getQName()
+                    .getLocalName();
+
+            switch (extensionDefinition.getQName().getLocalName()) {
+            case "a":
+                assertEquals("b", unknownNodeExtensionDefName);
+                assertEquals("c", subUnknownNodeExtensionDefName);
+                break;
+            case "b":
+                assertEquals("c", unknownNodeExtensionDefName);
+                assertEquals("a", subUnknownNodeExtensionDefName);
+                break;
+            case "c":
+                assertEquals("a", unknownNodeExtensionDefName);
+                assertEquals("b", subUnknownNodeExtensionDefName);
+                break;
+            case "r":
+                assertEquals("r", unknownNodeExtensionDefName);
+                assertEquals("r2", subUnknownNodeExtensionDefName);
+                break;
+            case "r2":
+                assertEquals("r2", unknownNodeExtensionDefName);
+                assertEquals("r", subUnknownNodeExtensionDefName);
+                break;
+            default:
+                fail("Unexpected extension definition");
+            }
+        }
+    }
+}