Bug 6112 - UnionTypeCodec fails to non-identityref value

[mdsal.git] / binding / mdsal-binding-dom-codec / src / test / java / org / opendaylight / yangtools / binding / data / codec / test / UnionTypeWithIdentityrefTest.java

diff --git a/binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/yangtools/binding/data/codec/test/UnionTypeWithIdentityrefTest.java b/binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/yangtools/binding/data/codec/test/UnionTypeWithIdentityrefTest.java
index d69c183e8b44a3aa16e5bec1c266cadb34f536ca..2c3e0f220e8a9814b8f350902324bfa5127d73c0 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/yangtools/binding/data/codec/test/UnionTypeWithIdentityrefTest.java
+++ b/binding/mdsal-binding-dom-codec/src/test/java/org/opendaylight/yangtools/binding/data/codec/test/UnionTypeWithIdentityrefTest.java
@@ -50,9 +50,8 @@ public class UnionTypeWithIdentityrefTest extends AbstractBindingRuntimeTest {
         registry.onBindingRuntimeContextUpdated(getRuntimeContext());
     }
 
-    @Test
-    public void bug6006Test() {
-        UnionType unionType = UnionTypeBuilder.getDefaultInstance(identOneString);
+    private DataObject createValueNode(final String valueString) {
+        UnionType unionType = UnionTypeBuilder.getDefaultInstance(valueString);
         UnionNode unionNode = new UnionNodeBuilder().setValue(unionType).build();
         NormalizedNode<?, ?> normalizedUnionNode = registry
             .toNormalizedNode(InstanceIdentifier.builder(UnionNode.class).build(), unionNode)
@@ -62,8 +61,22 @@ public class UnionTypeWithIdentityrefTest extends AbstractBindingRuntimeTest {
                 YangInstanceIdentifier.of(normalizedUnionNode.getNodeType()), normalizedUnionNode);
         DataObject unionNodeObj = unionNodeEntry.getValue();
         assertTrue(unionNodeObj instanceof UnionNode);
+        return unionNodeObj;
+    }
+
+    @Test
+    public void bug6006Test() {
+        DataObject unionNodeObj = createValueNode(identOneString);
         UnionType unionTypeObj = ((UnionNode) unionNodeObj).getValue();
         assertEquals(null, unionTypeObj.getUint8());
         assertEquals(IdentOne.class, unionTypeObj.getIdentityref());
     }
+
+    @Test
+    public void bug6112Test() {
+        DataObject unionNodeObj = createValueNode("1");
+        UnionType unionTypeObj = ((UnionNode) unionNodeObj).getValue();
+        assertEquals(Short.valueOf((short)1), unionTypeObj.getUint8());
+        assertEquals(null, unionTypeObj.getIdentityref());
+    }
 }