Refactor mdsal-binding-generator artifacts
[mdsal.git] / binding / mdsal-binding-generator / src / test / java / org / opendaylight / mdsal / binding / generator / impl / AugmentRelativeXPathTest.java
diff --git a/binding/mdsal-binding-generator/src/test/java/org/opendaylight/mdsal/binding/generator/impl/AugmentRelativeXPathTest.java b/binding/mdsal-binding-generator/src/test/java/org/opendaylight/mdsal/binding/generator/impl/AugmentRelativeXPathTest.java
new file mode 100644 (file)
index 0000000..9c2c03e
--- /dev/null
@@ -0,0 +1,116 @@
+/*
+ * 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.mdsal.binding.generator.impl;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+
+import java.util.List;
+import org.junit.Test;
+import org.opendaylight.mdsal.binding.model.api.GeneratedProperty;
+import org.opendaylight.mdsal.binding.model.api.GeneratedTransferObject;
+import org.opendaylight.mdsal.binding.model.api.GeneratedType;
+import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
+import org.opendaylight.mdsal.binding.model.api.MethodSignature;
+import org.opendaylight.mdsal.binding.spec.naming.BindingMapping;
+import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
+
+public class AugmentRelativeXPathTest {
+    @Test
+    public void testAugmentationWithRelativeXPath() {
+        final List<GeneratedType> genTypes = DefaultBindingGenerator.generateFor(
+            YangParserTestUtils.parseYangResourceDirectory("/augment-relative-xpath-models"));
+        assertNotNull("genTypes is null", genTypes);
+        assertEquals(26, genTypes.size());
+
+        GeneratedTransferObject gtInterfaceKey = null;
+        GeneratedType gtInterface = null;
+        GeneratedType gtTunnel = null;
+        GeneratedTransferObject gtTunnelKey = null;
+
+        for (final GeneratedType type : genTypes) {
+            if (!type.getPackageName().contains("augment._abstract.topology")) {
+                continue;
+            }
+
+            if (type.getName().equals("InterfaceKey")) {
+                gtInterfaceKey = (GeneratedTransferObject) type;
+
+                final List<GeneratedProperty> properties = gtInterfaceKey.getProperties();
+                assertNotNull("InterfaceKey properties are null", properties);
+                assertEquals(1, properties.size());
+
+                final GeneratedProperty property = properties.get(0);
+                assertEquals("interfaceId", property.getName());
+                assertNotNull("interfaceId return type is null", property.getReturnType());
+                assertEquals(JavaTypeName.create(String.class), property.getReturnType().getIdentifier());
+            } else if (type.getName().equals("Interface")) {
+                gtInterface = type;
+
+                final List<MethodSignature> gtInterfaceMethods = gtInterface.getMethodDefinitions();
+                assertNotNull("Interface methods are null", gtInterfaceMethods);
+                assertEquals(9, gtInterfaceMethods.size());
+
+                MethodSignature getIfcKeyMethod = null;
+                for (final MethodSignature method : gtInterfaceMethods) {
+                    if (BindingMapping.IDENTIFIABLE_KEY_NAME.equals(method.getName())) {
+                        getIfcKeyMethod = method;
+                        break;
+                    }
+                }
+                assertNotNull("getKey method is null", getIfcKeyMethod);
+                assertNotNull("getKey method return type is null", getIfcKeyMethod.getReturnType());
+                assertEquals(JavaTypeName.create(
+                    "org.opendaylight.yang.gen.v1.urn.model.augment._abstract.topology.rev130503.topology.interfaces",
+                    "InterfaceKey"),
+                    getIfcKeyMethod.getReturnType().getIdentifier());
+            } else if (type.getName().equals("Tunnel")) {
+                gtTunnel = type;
+
+                final List<MethodSignature> tunnelMethods = gtTunnel.getMethodDefinitions();
+                assertNotNull("Tunnel methods are null", tunnelMethods);
+                assertEquals(7, tunnelMethods.size());
+
+                MethodSignature getTunnelKeyMethod = null;
+                for (MethodSignature method : tunnelMethods) {
+                    if (BindingMapping.IDENTIFIABLE_KEY_NAME.equals(method.getName())) {
+                        getTunnelKeyMethod = method;
+                        break;
+                    }
+                }
+                assertNotNull("getKey method is null", getTunnelKeyMethod);
+                assertNotNull("getKey method return type", getTunnelKeyMethod.getReturnType());
+                assertEquals(JavaTypeName.create("org.opendaylight.yang.gen.v1.urn.model.augment._abstract.topology"
+                    + ".rev130503.topology.network.links.network.link.tunnels", "TunnelKey"),
+                    getTunnelKeyMethod.getReturnType().getIdentifier());
+            } else if (type.getName().equals("TunnelKey")) {
+                assertThat(type, instanceOf(GeneratedTransferObject.class));
+
+                gtTunnelKey = (GeneratedTransferObject) type;
+
+                final List<GeneratedProperty> properties = gtTunnelKey.getProperties();
+                assertNotNull("TunnelKey properties are null", properties);
+                assertEquals(1, properties.size());
+
+                final GeneratedProperty property = properties.get(0);
+                assertEquals("tunnelId", property.getName());
+                assertNotNull("tunnelId return type is null", property.getReturnType());
+                assertEquals(
+                    JavaTypeName.create("org.opendaylight.yang.gen.v1.urn.model._abstract.topology.rev130208", "Uri"),
+                    property.getReturnType().getIdentifier());
+            }
+        }
+
+        assertNotNull("Interface is null", gtInterface);
+        assertNotNull("InterfaceKey is null", gtInterfaceKey);
+        assertNotNull("Tunnel is null", gtTunnel);
+        assertNotNull("TunnelKey is null", gtTunnelKey);
+    }
+}