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.AugmentationNode;
32 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
35 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
36 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
37 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
38 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
39 import org.opendaylight.yangtools.yang.data.api.schema.SystemLeafSetNode;
40 import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
41 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
42 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
43 import org.opendaylight.yangtools.yang.data.api.schema.UserLeafSetNode;
44 import org.opendaylight.yangtools.yang.data.api.schema.UserMapNode;
46 public class NormalizedNodeWriterTest {
47 private QNameModule bazModule;
48 private QName myKeyedList;
49 private QName myKeyLeaf;
50 private QName myLeafList;
54 bazModule = QNameModule.create(XMLNamespace.of("baz-namespace"), Revision.of("1970-01-01"));
55 myKeyedList = QName.create(bazModule, "my-keyed-list");
56 myKeyLeaf = QName.create(bazModule, "my-key-leaf");
57 myLeafList = QName.create(bazModule, "my-leaf-list");
61 public void testNormalizedNodeWriter() throws IOException {
62 final NormalizedNodeStreamWriter loggingNormalizedNodeStreamWriter = new LoggingNormalizedNodeStreamWriter();
63 final NormalizedNodeWriter orderedNormalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
64 loggingNormalizedNodeStreamWriter);
66 assertEquals(loggingNormalizedNodeStreamWriter, orderedNormalizedNodeWriter.getWriter());
68 final IllegalStateException ex = assertThrows(IllegalStateException.class,
69 () -> orderedNormalizedNodeWriter.write(mock(NormalizedNode.class)));
70 assertTrue(ex.getMessage().startsWith("It wasn't possible to serialize node"));
72 final LeafSetEntryNode<?> mockedLeafSetEntryNode = mock(LeafSetEntryNode.class);
73 doReturn(new NodeWithValue<>(myLeafList, "leaflist-value-1")).when(mockedLeafSetEntryNode).getIdentifier();
74 doReturn("leaflist-value-1").when(mockedLeafSetEntryNode).body();
75 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafSetEntryNode));
77 final LeafNode<?> mockedLeafNode = mock(LeafNode.class);
78 doReturn("leaf-value-1").when(mockedLeafNode).body();
79 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafNode));
81 final DOMSourceAnyxmlNode mockedAnyXmlNode = mock(DOMSourceAnyxmlNode.class);
82 doCallRealMethod().when(mockedAnyXmlNode).bodyObjectModel();
83 doReturn(new DOMSource()).when(mockedAnyXmlNode).body();
84 assertNotNull(orderedNormalizedNodeWriter.write(mockedAnyXmlNode));
86 final NormalizedNode mockedContainerNode = mock(ContainerNode.class);
87 assertNotNull(orderedNormalizedNodeWriter.write(mockedContainerNode));
89 final MapEntryNode mockedMapEntryNode = mock(MapEntryNode.class);
90 doReturn(NodeIdentifierWithPredicates.of(myKeyedList, myKeyLeaf, "list-key-value-1"))
91 .when(mockedMapEntryNode).getIdentifier();
92 doReturn(null).when(mockedMapEntryNode).childByArg(any(NodeIdentifier.class));
93 assertNotNull(orderedNormalizedNodeWriter.write(mockedMapEntryNode));
95 final UnkeyedListEntryNode mockedUnkeyedListEntryNode = mock(UnkeyedListEntryNode.class);
96 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListEntryNode));
98 assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
100 assertNotNull(orderedNormalizedNodeWriter.write(mock(AugmentationNode.class)));
102 final UnkeyedListNode mockedUnkeyedListNode = mock(UnkeyedListNode.class);
103 final Set<?> value = Set.of(mockedUnkeyedListEntryNode);
104 doReturn(value).when(mockedUnkeyedListNode).body();
105 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListNode));
107 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserMapNode.class)));
109 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemMapNode.class)));
111 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserLeafSetNode.class)));
113 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemLeafSetNode.class)));
115 orderedNormalizedNodeWriter.flush();
116 orderedNormalizedNodeWriter.close();
118 final NormalizedNodeWriter normalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
119 loggingNormalizedNodeStreamWriter, false);
121 assertNotNull(normalizedNodeWriter.write(mockedMapEntryNode));
123 normalizedNodeWriter.flush();
124 normalizedNodeWriter.close();