2 * Copyright (c) 2022 PANTHEON.tech, s.r.o. 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.assertEquals;
11 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
12 import static org.mockito.ArgumentMatchers.anyString;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
17 import com.google.common.collect.ImmutableSet;
18 import com.google.gson.stream.JsonWriter;
19 import org.junit.jupiter.api.AfterAll;
20 import org.junit.jupiter.api.BeforeAll;
21 import org.junit.jupiter.api.Test;
22 import org.mockito.ArgumentCaptor;
23 import org.opendaylight.yangtools.yang.common.QName;
24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
25 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
27 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
28 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
29 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
30 import org.opendaylight.yangtools.yang.model.api.type.InstanceIdentifierTypeDefinition;
31 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
34 private static final String FOO_NS = "foons"; // namespace for prefix 'foo'
35 private static final QName FOO_FOO = QName.create(FOO_NS, "foo"); // list with key 'str'
36 private static final QName FOO_BAR = QName.create(FOO_NS, "bar"); // list with key 'qname'
37 private static final QName FOO_BAZ = QName.create(FOO_NS, "baz"); // list with key 'id'
38 private static final QName FOO_BEE = QName.create(FOO_NS, "bee"); // list with key 'bts'
39 private static final QName FOO_ONE = QName.create(FOO_NS, "one"); // identity
40 private static final QName FOO_STR = QName.create(FOO_NS, "str"); // key of type 'string'
41 private static final QName FOO_QNAME = QName.create(FOO_NS, "qname"); // key of type 'one' based
42 private static final QName FOO_ID = QName.create(FOO_NS, "id"); // key of type 'instance-identifier'
43 private static final QName FOO_BTS = QName.create(FOO_NS, "bts"); // key of type 'bts' (bits)
45 private static final String BAR_NS = "barns"; // namespace for prefix 'bar'
46 private static final QName BAR_STR = QName.create(BAR_NS, "str"); // leaf-list of type 'string'
47 private static final QName BAR_FOO = QName.create(BAR_NS, "foo"); // leaf-list of type 'foo:one' based
48 private static final QName BAR_BAR = QName.create(BAR_NS, "bar"); // leaf-list of type 'instance-identifier'
49 private static final QName BAR_BEE = QName.create(BAR_NS, "bee"); // leaf-list of type 'foo:bts' (bits)
50 private static final QName BAR_TWO = QName.create(BAR_NS, "two"); // identity inheriting 'foo:one'
52 private static JSONCodec<YangInstanceIdentifier> CODEC;
55 static void beforeAll() {
56 final var modelContext = YangParserTestUtils.parseYangResourceDirectory("/yt1473");
57 final var baz = assertInstanceOf(ListSchemaNode.class, modelContext.getDataChildByName(FOO_BAZ));
58 final var id = assertInstanceOf(LeafSchemaNode.class, baz.getDataChildByName(FOO_ID));
59 final var type = assertInstanceOf(InstanceIdentifierTypeDefinition.class, id.getType());
60 CODEC = JSONCodecFactorySupplier.RFC7951.getShared(modelContext).instanceIdentifierCodec(type);
64 static void afterAll() {
69 void testSerializeSimple() throws Exception {
70 // No escaping needed, use single quotes
71 assertSerdes("/bar:str[.='str\"']", buildYangInstanceIdentifier(BAR_STR, "str\""));
72 assertSerdes("/bar:str[.='str\\']", buildYangInstanceIdentifier(BAR_STR, "str\\"));
73 assertSerdes("/bar:str[.='str\r']", buildYangInstanceIdentifier(BAR_STR, "str\r"));
74 assertSerdes("/bar:str[.='str\n']", buildYangInstanceIdentifier(BAR_STR, "str\n"));
75 assertSerdes("/bar:str[.='str\t']", buildYangInstanceIdentifier(BAR_STR, "str\t"));
77 assertSerdes("/foo:foo[str='str\"\\']", buildYangInstanceIdentifier(FOO_FOO, FOO_STR, "str\"\\"));
78 assertSerdes("/foo:foo[str='str\r\n\t']", buildYangInstanceIdentifier(FOO_FOO, FOO_STR, "str\r\n\t"));
82 void testSerializeEscaped() throws Exception {
83 // Escaping is needed, use double quotes and escape
84 assertSerdes("/bar:str[.=\"str'\\\"\"]", buildYangInstanceIdentifier(BAR_STR, "str'\""));
85 assertSerdes("/bar:str[.=\"str'\\n\"]", buildYangInstanceIdentifier(BAR_STR, "str'\n"));
86 assertSerdes("/bar:str[.=\"str'\\t\"]", buildYangInstanceIdentifier(BAR_STR, "str'\t"));
87 assertSerdes("/bar:str[.=\"str'\r\"]", buildYangInstanceIdentifier(BAR_STR, "str'\r"));
89 assertSerdes("/foo:foo[str=\"str'\\\"\\n\"]", buildYangInstanceIdentifier(FOO_FOO, FOO_STR, "str'\"\n"));
90 assertSerdes("/foo:foo[str=\"str'\\t\r\"]", buildYangInstanceIdentifier(FOO_FOO, FOO_STR, "str'\t\r"));
94 void testSerializeIdentityRefSame() throws Exception {
95 assertSerdes("/foo:bar[qname='one']", buildYangInstanceIdentifier(FOO_BAR, FOO_QNAME, FOO_ONE));
99 void testSerializeIdentityRefOther() throws Exception {
100 // No escaping is needed, use double quotes and escape
101 assertSerdes("/foo:bar[qname='bar:two']", buildYangInstanceIdentifier(FOO_BAR, FOO_QNAME, BAR_TWO));
105 void testSerializeInstanceIdentifierRef() throws Exception {
106 assertSerdes("/foo:baz[id=\"/foo:bar[qname='bar:two']\"]",
107 buildYangInstanceIdentifier(FOO_BAZ, FOO_ID, buildYangInstanceIdentifier(FOO_BAR, FOO_QNAME, BAR_TWO)));
111 void testSerializeIdentityValue() throws Exception {
112 assertSerdes("/bar:foo[.='foo:one']", buildYangInstanceIdentifier(BAR_FOO, FOO_ONE));
113 assertSerdes("/bar:foo[.='two']", buildYangInstanceIdentifier(BAR_FOO, BAR_TWO));
117 void testSerializeInstanceIdentifierValue() throws Exception {
118 assertSerdes("/bar:bar[.=\"/foo:bar[qname='bar:two']\"]",
119 buildYangInstanceIdentifier(BAR_BAR, buildYangInstanceIdentifier(FOO_BAR, FOO_QNAME, BAR_TWO)));
120 assertSerdes("/bar:bar[.=\"/foo:bar[qname='one']\"]",
121 buildYangInstanceIdentifier(BAR_BAR, buildYangInstanceIdentifier(FOO_BAR, FOO_QNAME, FOO_ONE)));
125 void testSerializeBits() throws Exception {
126 assertSerdes("/foo:bee[bts='']", buildYangInstanceIdentifier(FOO_BEE, FOO_BTS, ImmutableSet.of()));
127 assertSerdes("/foo:bee[bts='one']", buildYangInstanceIdentifier(FOO_BEE, FOO_BTS, ImmutableSet.of("one")));
128 assertSerdes("/foo:bee[bts='two three']",
129 buildYangInstanceIdentifier(FOO_BEE, FOO_BTS, ImmutableSet.of("two", "three")));
133 void testSerializeBitsValue() throws Exception {
134 assertSerdes("/bar:bee[.='']", buildYangInstanceIdentifier(BAR_BEE, ImmutableSet.of()));
135 assertSerdes("/bar:bee[.='one']", buildYangInstanceIdentifier(BAR_BEE, ImmutableSet.of("one")));
136 assertSerdes("/bar:bee[.='two three']", buildYangInstanceIdentifier(BAR_BEE, ImmutableSet.of("two", "three")));
139 private static void assertSerdes(final String expected, final YangInstanceIdentifier id) throws Exception {
140 final var writer = mock(JsonWriter.class);
141 final var captor = ArgumentCaptor.forClass(String.class);
142 doReturn(writer).when(writer).value(anyString());
143 CODEC.writeValue(writer, id);
144 verify(writer).value(captor.capture());
146 assertEquals(expected, captor.getValue());
147 assertEquals(id, CODEC.parseValue(null, expected));
150 private static YangInstanceIdentifier buildYangInstanceIdentifier(final QName node, final QName key,
151 final Object value) {
152 return YangInstanceIdentifier.create(
153 new NodeIdentifier(node), NodeIdentifierWithPredicates.of(node, key, value));
156 private static YangInstanceIdentifier buildYangInstanceIdentifier(final QName node, final Object value) {
157 return YangInstanceIdentifier.create(new NodeIdentifier(node), new NodeWithValue<>(node, value));