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.api.schema.stream;
10 import static org.junit.jupiter.api.Assertions.assertEquals;
11 import static org.junit.jupiter.api.Assertions.assertNotNull;
12 import static org.junit.jupiter.api.Assertions.assertThrows;
13 import static org.junit.jupiter.api.Assertions.assertTrue;
14 import static org.mockito.ArgumentMatchers.any;
15 import static org.mockito.Mockito.doCallRealMethod;
16 import static org.mockito.Mockito.doReturn;
17 import static org.mockito.Mockito.mock;
19 import java.io.IOException;
21 import javax.xml.transform.dom.DOMSource;
22 import org.junit.jupiter.api.BeforeEach;
23 import org.junit.jupiter.api.Test;
24 import org.opendaylight.yangtools.yang.common.QName;
25 import org.opendaylight.yangtools.yang.common.QNameModule;
26 import org.opendaylight.yangtools.yang.common.Revision;
27 import org.opendaylight.yangtools.yang.common.XMLNamespace;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
29 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
31 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
32 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
35 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
36 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
37 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
38 import org.opendaylight.yangtools.yang.data.api.schema.SystemLeafSetNode;
39 import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
40 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
41 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
42 import org.opendaylight.yangtools.yang.data.api.schema.UserLeafSetNode;
43 import org.opendaylight.yangtools.yang.data.api.schema.UserMapNode;
45 public class NormalizedNodeWriterTest {
46 private QNameModule bazModule;
47 private QName myKeyedList;
48 private QName myKeyLeaf;
49 private QName myLeafList;
53 bazModule = QNameModule.create(XMLNamespace.of("baz-namespace"), Revision.of("1970-01-01"));
54 myKeyedList = QName.create(bazModule, "my-keyed-list");
55 myKeyLeaf = QName.create(bazModule, "my-key-leaf");
56 myLeafList = QName.create(bazModule, "my-leaf-list");
60 public void testNormalizedNodeWriter() throws IOException {
61 final NormalizedNodeStreamWriter loggingNormalizedNodeStreamWriter = new LoggingNormalizedNodeStreamWriter();
62 final NormalizedNodeWriter orderedNormalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
63 loggingNormalizedNodeStreamWriter);
65 assertEquals(loggingNormalizedNodeStreamWriter, orderedNormalizedNodeWriter.getWriter());
67 final IllegalStateException ex = assertThrows(IllegalStateException.class,
68 () -> orderedNormalizedNodeWriter.write(mock(NormalizedNode.class)));
69 assertTrue(ex.getMessage().startsWith("It wasn't possible to serialize node"));
71 final LeafSetEntryNode<?> mockedLeafSetEntryNode = mock(LeafSetEntryNode.class);
72 doReturn(new NodeWithValue<>(myLeafList, "leaflist-value-1")).when(mockedLeafSetEntryNode).getIdentifier();
73 doReturn("leaflist-value-1").when(mockedLeafSetEntryNode).body();
74 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafSetEntryNode));
76 final LeafNode<?> mockedLeafNode = mock(LeafNode.class);
77 doReturn("leaf-value-1").when(mockedLeafNode).body();
78 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafNode));
80 final DOMSourceAnyxmlNode mockedAnyXmlNode = mock(DOMSourceAnyxmlNode.class);
81 doCallRealMethod().when(mockedAnyXmlNode).bodyObjectModel();
82 doReturn(new DOMSource()).when(mockedAnyXmlNode).body();
83 assertNotNull(orderedNormalizedNodeWriter.write(mockedAnyXmlNode));
85 final NormalizedNode mockedContainerNode = mock(ContainerNode.class);
86 assertNotNull(orderedNormalizedNodeWriter.write(mockedContainerNode));
88 final MapEntryNode mockedMapEntryNode = mock(MapEntryNode.class);
89 doReturn(NodeIdentifierWithPredicates.of(myKeyedList, myKeyLeaf, "list-key-value-1"))
90 .when(mockedMapEntryNode).getIdentifier();
91 doReturn(null).when(mockedMapEntryNode).childByArg(any(NodeIdentifier.class));
92 assertNotNull(orderedNormalizedNodeWriter.write(mockedMapEntryNode));
94 final UnkeyedListEntryNode mockedUnkeyedListEntryNode = mock(UnkeyedListEntryNode.class);
95 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListEntryNode));
97 assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
99 final UnkeyedListNode mockedUnkeyedListNode = mock(UnkeyedListNode.class);
100 final Set<?> value = Set.of(mockedUnkeyedListEntryNode);
101 doReturn(value).when(mockedUnkeyedListNode).body();
102 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListNode));
104 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserMapNode.class)));
106 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemMapNode.class)));
108 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserLeafSetNode.class)));
110 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemLeafSetNode.class)));
112 orderedNormalizedNodeWriter.flush();
113 orderedNormalizedNodeWriter.close();
115 try (var nnWriter = NormalizedNodeWriter.forStreamWriter(loggingNormalizedNodeStreamWriter, false)) {
116 assertNotNull(nnWriter.write(mockedMapEntryNode));