2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.yang.data.codec.gson;
10 import static org.junit.Assert.assertTrue;
12 import com.google.gson.JsonArray;
13 import com.google.gson.JsonElement;
14 import com.google.gson.JsonObject;
15 import com.google.gson.JsonParser;
16 import com.google.gson.JsonPrimitive;
17 import java.io.BufferedReader;
19 import java.io.FileReader;
20 import java.io.IOException;
21 import java.io.InputStream;
22 import java.io.StringWriter;
23 import java.io.Writer;
24 import java.net.URISyntaxException;
25 import java.nio.charset.StandardCharsets;
26 import javax.xml.stream.XMLStreamReader;
27 import org.opendaylight.yangtools.util.xml.UntrustedXML;
28 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
29 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
30 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
31 import org.opendaylight.yangtools.yang.data.codec.xml.XmlParserStream;
32 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
33 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
34 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
35 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
37 public final class TestUtils {
42 static String loadTextFile(final File file) throws IOException {
43 final StringBuilder result = new StringBuilder();
44 try (BufferedReader bufReader = new BufferedReader(new FileReader(file, StandardCharsets.UTF_8))) {
46 while ((line = bufReader.readLine()) != null) {
50 return result.toString();
53 static String loadTextFile(final String relativePath) throws IOException, URISyntaxException {
54 return loadTextFile(new File(TestUtils.class.getResource(relativePath).toURI()));
57 static void loadXmlToNormalizedNodes(final InputStream xmlInputStream, final NormalizedNodeResult result,
58 final EffectiveModelContext schemaContext) throws Exception {
59 final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(xmlInputStream);
60 final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
61 final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext);
62 xmlParser.parse(reader);
65 static String normalizedNodesToJsonString(final NormalizedNode data,
66 final EffectiveModelContext schemaContext, final SchemaPath rootPath) throws IOException {
67 final Writer writer = new StringWriter();
68 final NormalizedNodeStreamWriter jsonStream = JSONNormalizedNodeStreamWriter.createExclusiveWriter(
69 JSONCodecFactorySupplier.DRAFT_LHOTKA_NETMOD_YANG_JSON_02.getShared(schemaContext), rootPath, null,
70 JsonWriterFactory.createJsonWriter(writer, 2));
71 try (NormalizedNodeWriter nodeWriter = NormalizedNodeWriter.forStreamWriter(jsonStream)) {
72 nodeWriter.write(data);
74 return writer.toString();
77 static JsonObject childObject(final JsonObject jsonObject, final String... names) {
78 for (final String name : names) {
79 final JsonObject childJsonObject = jsonObject.getAsJsonObject(name);
80 if (childJsonObject != null) {
81 return childJsonObject;
87 static JsonPrimitive childPrimitive(final JsonObject jsonObject, final String... names) {
88 for (final String name : names) {
89 final JsonPrimitive childJsonPrimitive = jsonObject.getAsJsonPrimitive(name);
90 if (childJsonPrimitive != null) {
91 return childJsonPrimitive;
97 static JsonArray childArray(final JsonObject jsonObject, final String... names) {
98 for (final String name : names) {
99 final JsonArray childJsonArray = jsonObject.getAsJsonArray(name);
100 if (childJsonArray != null) {
101 return childJsonArray;
107 static JsonObject resolveCont1(final String jsonOutput) {
108 final JsonElement rootElement = JsonParser.parseString(jsonOutput);
109 assertTrue(rootElement.isJsonObject());
110 final JsonObject rootObject = rootElement.getAsJsonObject();
111 final JsonObject cont1 = childObject(rootObject, "complexjson:cont1", "cont1");
115 static JsonObject resolveCont2(final String jsonOutput) {
116 final JsonElement rootElement = JsonParser.parseString(jsonOutput);
117 assertTrue(rootElement.isJsonObject());
118 final JsonObject rootObject = rootElement.getAsJsonObject();
119 final JsonObject cont2 = childObject(rootObject, "complexjson:cont2", "cont2");