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