yang/yang-data-codec-gson/src/test/java/org/opendaylight/yangtools/yang/data/codec/gson/AnyXmlSupportTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. 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
   9 package org.opendaylight.yangtools.yang.data.codec.gson;
  10
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.opendaylight.yangtools.yang.data.codec.gson.TestUtils.loadTextFile;
  14
  15 import com.google.gson.JsonElement;
  16 import com.google.gson.JsonParser;
  17 import com.google.gson.stream.JsonReader;
  18 import java.io.IOException;
  19 import java.io.StringReader;
  20 import java.io.StringWriter;
  21 import java.io.Writer;
  22 import java.net.URISyntaxException;
  23 import javax.xml.transform.dom.DOMSource;
  24 import org.junit.BeforeClass;
  25 import org.junit.Test;
  26 import org.opendaylight.yangtools.util.xml.UntrustedXML;
  27 import org.opendaylight.yangtools.yang.common.QName;
  28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  29 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  30 import org.opendaylight.yangtools.yang.data.api.schema.AnyXmlNode;
  31 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  32 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  33 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  34 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  35 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
  36 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  37 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  38 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  39 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  40 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  41 import org.w3c.dom.Document;
  42 import org.w3c.dom.Element;
  43 import org.w3c.dom.Text;
  44
  45 public class AnyXmlSupportTest {
  46
  47     private static final QName CONT_1 = QName.create("ns:complex:json", "2014-08-11", "cont1");
  48     private static final QName LF12_ANY = QName.create(CONT_1, "lf12-any");
  49     private static final QName LF13_ANY = QName.create(CONT_1, "lf13-any");
  50     private static final QName LF14_ANY = QName.create(CONT_1, "lf14-any");
  51
  52     private static SchemaContext schemaContext;
  53
  54     @BeforeClass
  55     public static void setup() {
  56         schemaContext = YangParserTestUtils.parseYangResourceDirectory("/complexjson/yang");
  57     }
  58
  59     @Test
  60     public void anyXmlNodeWithSimpleValueInContainer() throws IOException, URISyntaxException {
  61         final String inputJson = loadTextFile("/complexjson/anyxml-node-with-simple-value-in-container.json");
  62
  63         // deserialization
  64         final NormalizedNodeResult result = new NormalizedNodeResult();
  65         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
  66         final JsonParserStream jsonParser = JsonParserStream.create(streamWriter, schemaContext);
  67         jsonParser.parse(new JsonReader(new StringReader(inputJson)));
  68         final NormalizedNode<?, ?> transformedInput = result.getResult();
  69         assertNotNull(transformedInput);
  70
  71         // lf12-any check
  72         final DOMSource Lf12AnyActualValue = getParsedAnyXmlValue(transformedInput, LF12_ANY);
  73         final DOMSource Lf12AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf12-any", "100.5");
  74         verifyTransformedAnyXmlNodeValue(Lf12AnyExpectedValue, Lf12AnyActualValue);
  75
  76         // lf13-any check
  77         final DOMSource Lf13AnyActualValue = getParsedAnyXmlValue(transformedInput, LF13_ANY);
  78         final DOMSource Lf13AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf13-any", "true");
  79         verifyTransformedAnyXmlNodeValue(Lf13AnyExpectedValue, Lf13AnyActualValue);
  80
  81         // lf14-any check
  82         final DOMSource Lf14AnyActualValue = getParsedAnyXmlValue(transformedInput, LF14_ANY);
  83         final DOMSource Lf14AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf14-any", "null");
  84         verifyTransformedAnyXmlNodeValue(Lf14AnyExpectedValue, Lf14AnyActualValue);
  85
  86         // serialization
  87         final Writer writer = new StringWriter();
  88         final NormalizedNodeStreamWriter jsonStream = JSONNormalizedNodeStreamWriter.createExclusiveWriter(
  89             JSONCodecFactory.getShared(schemaContext), SchemaPath.ROOT, null,
  90             JsonWriterFactory.createJsonWriter(writer, 2));
  91         final NormalizedNodeWriter nodeWriter = NormalizedNodeWriter.forStreamWriter(jsonStream);
  92         nodeWriter.write(transformedInput);
  93         nodeWriter.close();
  94         final String serializationResult = writer.toString();
  95
  96         final JsonParser parser = new JsonParser();
  97         final JsonElement expected = parser.parse(inputJson);
  98         final JsonElement actual = parser.parse(serializationResult);
  99         assertTrue(expected.equals(actual));
 100     }
 101
 102     @Test
 103     public void anyXmlNodeWithCompositeValueInContainer() throws IOException, URISyntaxException {
 104         final String inputJson = loadTextFile("/complexjson/anyxml-node-with-composite-value-in-container.json");
 105
 106         final NormalizedNodeResult result = new NormalizedNodeResult();
 107         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
 108         final JsonParserStream jsonParser = JsonParserStream.create(streamWriter, schemaContext);
 109         jsonParser.parse(new JsonReader(new StringReader(inputJson)));
 110         final NormalizedNode<?, ?> transformedInput = result.getResult();
 111         assertNotNull(transformedInput);
 112
 113         // lf12-any check
 114         final DOMSource Lf12AnyActualValue = getParsedAnyXmlValue(transformedInput, LF12_ANY);
 115         final DOMSource Lf12AnyExpectedValue = createLf12AnyXmlCompositeValue("ns:complex:json", "lf12-any");
 116         verifyTransformedAnyXmlNodeValue(Lf12AnyExpectedValue, Lf12AnyActualValue);
 117
 118         // lf13-any check
 119         final DOMSource Lf13AnyActualValue = getParsedAnyXmlValue(transformedInput, LF13_ANY);
 120         final DOMSource Lf13AnyExpectedValue = createLf13AnyXmlCompositeValue("ns:complex:json", "lf13-any");
 121         verifyTransformedAnyXmlNodeValue(Lf13AnyExpectedValue, Lf13AnyActualValue);
 122
 123         // lf14-any check
 124         final DOMSource Lf14AnyActualValue = getParsedAnyXmlValue(transformedInput, LF14_ANY);
 125         final DOMSource Lf14AnyExpectedValue = createLf14AnyXmlCompositeValue("ns:complex:json", "lf14-any");
 126         verifyTransformedAnyXmlNodeValue(Lf14AnyExpectedValue, Lf14AnyActualValue);
 127
 128         // serialization
 129         final Writer writer = new StringWriter();
 130         final NormalizedNodeStreamWriter jsonStream = JSONNormalizedNodeStreamWriter.createExclusiveWriter(
 131             JSONCodecFactory.getShared(schemaContext), SchemaPath.ROOT, null,
 132             JsonWriterFactory.createJsonWriter(writer, 2));
 133         final NormalizedNodeWriter nodeWriter = NormalizedNodeWriter.forStreamWriter(jsonStream);
 134         nodeWriter.write(transformedInput);
 135         nodeWriter.close();
 136         final String serializationResult = writer.toString();
 137
 138         final JsonParser parser = new JsonParser();
 139         final JsonElement expected = parser.parse(inputJson);
 140         final JsonElement actual = parser.parse(serializationResult);
 141         assertTrue(expected.equals(actual));
 142     }
 143
 144     private static DOMSource getParsedAnyXmlValue(final NormalizedNode<?, ?> transformedInput, final QName anyxmlName) {
 145         assertTrue(transformedInput instanceof ContainerNode);
 146         final ContainerNode cont1 = (ContainerNode) transformedInput;
 147         final DataContainerChild<? extends PathArgument, ?> child = cont1.getChild(new NodeIdentifier(anyxmlName))
 148                 .get();
 149         assertTrue(child instanceof AnyXmlNode);
 150         final AnyXmlNode anyXmlNode = (AnyXmlNode) child;
 151         return anyXmlNode.getValue();
 152     }
 153
 154     private static void verifyTransformedAnyXmlNodeValue(final DOMSource expectedValue, final DOMSource actualValue) {
 155         assertTrue(expectedValue.getNode().isEqualNode(actualValue.getNode()));
 156     }
 157
 158     private static DOMSource createAnyXmlSimpleValue(final String ns, final String name, final String value) {
 159         final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
 160         final Element rootElement = doc.createElementNS(ns, name);
 161         doc.appendChild(rootElement);
 162         final Text textNode = doc.createTextNode(value);
 163         rootElement.appendChild(textNode);
 164         return new DOMSource(doc.getDocumentElement());
 165     }
 166
 167     private static DOMSource createLf12AnyXmlCompositeValue(final String ns, final String name) {
 168         final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
 169         final Element rootElement = doc.createElementNS(ns, name);
 170
 171         final Element arrayElement1 = doc.createElement("array-element");
 172         final Text arrayElement1Text = doc.createTextNode("true");
 173         arrayElement1.appendChild(arrayElement1Text);
 174
 175         final Element arrayElement2 = doc.createElement("array-element");
 176
 177         final Element arrayElement2Baz = doc.createElement("baz");
 178         final Element bazArrayElement1 = doc.createElement("array-element");
 179         final Text bazArrayElement1Text = doc.createTextNode("120");
 180         bazArrayElement1.appendChild(bazArrayElement1Text);
 181         final Element bazArrayElement2 = doc.createElement("array-element");
 182         final Text bazArrayElement2Text = doc.createTextNode("str-val");
 183         bazArrayElement2.appendChild(bazArrayElement2Text);
 184         final Element bazArrayElement3 = doc.createElement("array-element");
 185         final Text bazArrayElement3Text = doc.createTextNode("false");
 186         bazArrayElement3.appendChild(bazArrayElement3Text);
 187
 188         arrayElement2Baz.appendChild(bazArrayElement1);
 189         arrayElement2Baz.appendChild(bazArrayElement2);
 190         arrayElement2Baz.appendChild(bazArrayElement3);
 191
 192         arrayElement2.appendChild(arrayElement2Baz);
 193
 194         final Element arrayElement3 = doc.createElement("array-element");
 195         final Element arrayElement3Foo = doc.createElement("foo");
 196         final Text fooText = doc.createTextNode("null");
 197         arrayElement3Foo.appendChild(fooText);
 198         arrayElement3.appendChild(arrayElement3Foo);
 199
 200         rootElement.appendChild(arrayElement1);
 201         rootElement.appendChild(arrayElement2);
 202         rootElement.appendChild(arrayElement3);
 203
 204         doc.appendChild(rootElement);
 205
 206         return new DOMSource(doc.getDocumentElement());
 207     }
 208
 209     private static DOMSource createLf13AnyXmlCompositeValue(final String ns, final String name) {
 210         final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
 211         final Element rootElement = doc.createElementNS(ns, name);
 212
 213         final Element anyXmlArrayA = doc.createElement("anyxml-array-a");
 214
 215         final Element arrayAElement1 = doc.createElement("array-element");
 216         final Element arrayAElement1Foo = doc.createElement("foo");
 217         final Text fooText = doc.createTextNode("true");
 218         arrayAElement1Foo.appendChild(fooText);
 219         arrayAElement1.appendChild(arrayAElement1Foo);
 220
 221         final Element arrayAElement2 = doc.createElement("array-element");
 222         final Text arrayAElement2Text = doc.createTextNode("10");
 223         arrayAElement2.appendChild(arrayAElement2Text);
 224
 225         final Element arrayAElement3 = doc.createElement("array-element");
 226         final Element arrayAElement3Bar = doc.createElement("bar");
 227         final Text barText = doc.createTextNode("false");
 228         arrayAElement3Bar.appendChild(barText);
 229         arrayAElement3.appendChild(arrayAElement3Bar);
 230
 231         anyXmlArrayA.appendChild(arrayAElement1);
 232         anyXmlArrayA.appendChild(arrayAElement2);
 233         anyXmlArrayA.appendChild(arrayAElement3);
 234
 235         final Element anyXmlArrayB = doc.createElement("anyxml-array-b");
 236
 237         final Element arrayBElement1 = doc.createElement("array-element");
 238         final Text arrayBElement1Text = doc.createTextNode("1");
 239         arrayBElement1.appendChild(arrayBElement1Text);
 240
 241         final Element arrayBElement2 = doc.createElement("array-element");
 242         final Text arrayBElement2Text = doc.createTextNode("2");
 243         arrayBElement2.appendChild(arrayBElement2Text);
 244
 245         final Element arrayBElement3 = doc.createElement("array-element");
 246
 247         final Element arrayBElement3Element1 = doc.createElement("array-element");
 248         final Text arrayBElement3Element1Text = doc.createTextNode("4");
 249         arrayBElement3Element1.appendChild(arrayBElement3Element1Text);
 250         final Element arrayBElement3Element2 = doc.createElement("array-element");
 251         final Text arrayBElement3Element2Text = doc.createTextNode("5");
 252         arrayBElement3Element2.appendChild(arrayBElement3Element2Text);
 253
 254         arrayBElement3.appendChild(arrayBElement3Element1);
 255         arrayBElement3.appendChild(arrayBElement3Element2);
 256
 257         final Element arrayBElement4 = doc.createElement("array-element");
 258         final Text arrayBElement4Text = doc.createTextNode("7");
 259         arrayBElement4.appendChild(arrayBElement4Text);
 260
 261         anyXmlArrayB.appendChild(arrayBElement1);
 262         anyXmlArrayB.appendChild(arrayBElement2);
 263         anyXmlArrayB.appendChild(arrayBElement3);
 264         anyXmlArrayB.appendChild(arrayBElement4);
 265
 266         rootElement.appendChild(anyXmlArrayA);
 267         rootElement.appendChild(anyXmlArrayB);
 268
 269         doc.appendChild(rootElement);
 270
 271         return new DOMSource(doc.getDocumentElement());
 272     }
 273
 274     private static DOMSource createLf14AnyXmlCompositeValue(final String ns, final String name) {
 275         final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
 276         final Element rootElement = doc.createElementNS(ns, name);
 277
 278         final Element anyXmlObjectA = doc.createElement("anyxml-object-a");
 279         final Element dataA1 = doc.createElement("data-a1");
 280         final Text dataA1Text = doc.createTextNode("10");
 281         dataA1.appendChild(dataA1Text);
 282         final Element dataA2 = doc.createElement("data-a2");
 283         final Text dataA2Text = doc.createTextNode("11");
 284         dataA2.appendChild(dataA2Text);
 285
 286         anyXmlObjectA.appendChild(dataA1);
 287         anyXmlObjectA.appendChild(dataA2);
 288
 289         final Element anyXmlObjectB = doc.createElement("anyxml-object-b");
 290
 291         final Element childObjectB1 = doc.createElement("child-object-b1");
 292         final Element dataB1 = doc.createElement("data-b1");
 293         final Text dataB1Text = doc.createTextNode("5.5");
 294         dataB1.appendChild(dataB1Text);
 295         childObjectB1.appendChild(dataB1);
 296
 297         final Element childObjectB2 = doc.createElement("child-object-b2");
 298         final Element dataB2 = doc.createElement("data-b2");
 299         final Text dataB2Text = doc.createTextNode("b2-val");
 300         dataB2.appendChild(dataB2Text);
 301         childObjectB2.appendChild(dataB2);
 302
 303         anyXmlObjectB.appendChild(childObjectB1);
 304         anyXmlObjectB.appendChild(childObjectB2);
 305
 306         rootElement.appendChild(anyXmlObjectA);
 307         rootElement.appendChild(anyXmlObjectB);
 308
 309         doc.appendChild(rootElement);
 310
 311         return new DOMSource(doc.getDocumentElement());
 312     }
 313 }