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.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertThrows;
13 import static org.junit.Assert.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.Before;
23 import org.junit.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 {
48 private QNameModule bazModule;
50 private QName myKeyedList;
51 private QName myKeyLeaf;
52 private QName myLeafList;
56 bazModule = QNameModule.create(XMLNamespace.of("baz-namespace"), Revision.of("1970-01-01"));
57 myKeyedList = QName.create(bazModule, "my-keyed-list");
58 myKeyLeaf = QName.create(bazModule, "my-key-leaf");
59 myLeafList = QName.create(bazModule, "my-leaf-list");
63 public void testNormalizedNodeWriter() throws IOException {
64 final NormalizedNodeStreamWriter loggingNormalizedNodeStreamWriter = new LoggingNormalizedNodeStreamWriter();
65 final NormalizedNodeWriter orderedNormalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
66 loggingNormalizedNodeStreamWriter);
68 assertEquals(loggingNormalizedNodeStreamWriter, orderedNormalizedNodeWriter.getWriter());
70 final IllegalStateException ex = assertThrows(IllegalStateException.class,
71 () -> orderedNormalizedNodeWriter.write(mock(NormalizedNode.class)));
72 assertTrue(ex.getMessage().startsWith("It wasn't possible to serialize node"));
74 final LeafSetEntryNode<?> mockedLeafSetEntryNode = mock(LeafSetEntryNode.class);
75 doReturn(new NodeWithValue<>(myLeafList, "leaflist-value-1")).when(mockedLeafSetEntryNode).getIdentifier();
76 doReturn("leaflist-value-1").when(mockedLeafSetEntryNode).body();
77 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafSetEntryNode));
79 final LeafNode<?> mockedLeafNode = mock(LeafNode.class);
80 doReturn("leaf-value-1").when(mockedLeafNode).body();
81 assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafNode));
83 final DOMSourceAnyxmlNode mockedAnyXmlNode = mock(DOMSourceAnyxmlNode.class);
84 doCallRealMethod().when(mockedAnyXmlNode).bodyObjectModel();
85 doReturn(new DOMSource()).when(mockedAnyXmlNode).body();
86 assertNotNull(orderedNormalizedNodeWriter.write(mockedAnyXmlNode));
88 final NormalizedNode mockedContainerNode = mock(ContainerNode.class);
89 assertNotNull(orderedNormalizedNodeWriter.write(mockedContainerNode));
91 final MapEntryNode mockedMapEntryNode = mock(MapEntryNode.class);
92 doReturn(NodeIdentifierWithPredicates.of(myKeyedList, myKeyLeaf, "list-key-value-1"))
93 .when(mockedMapEntryNode).getIdentifier();
94 doReturn(null).when(mockedMapEntryNode).childByArg(any(NodeIdentifier.class));
95 assertNotNull(orderedNormalizedNodeWriter.write(mockedMapEntryNode));
97 final UnkeyedListEntryNode mockedUnkeyedListEntryNode = mock(UnkeyedListEntryNode.class);
98 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListEntryNode));
100 assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
102 assertNotNull(orderedNormalizedNodeWriter.write(mock(AugmentationNode.class)));
104 final UnkeyedListNode mockedUnkeyedListNode = mock(UnkeyedListNode.class);
105 final Set<?> value = Set.of(mockedUnkeyedListEntryNode);
106 doReturn(value).when(mockedUnkeyedListNode).body();
107 assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListNode));
109 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserMapNode.class)));
111 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemMapNode.class)));
113 assertNotNull(orderedNormalizedNodeWriter.write(mock(UserLeafSetNode.class)));
115 assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemLeafSetNode.class)));
117 orderedNormalizedNodeWriter.flush();
118 orderedNormalizedNodeWriter.close();
120 final NormalizedNodeWriter normalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
121 loggingNormalizedNodeStreamWriter, false);
123 assertNotNull(normalizedNodeWriter.write(mockedMapEntryNode));
125 normalizedNodeWriter.flush();
126 normalizedNodeWriter.close();