e5c5b20070cef56a0367e58bc0fceb52c6c61b43
[yangtools.git] / data / yang-data-api / src / test / java / org / opendaylight / yangtools / yang / data / api / schema / stream / NormalizedNodeWriterTest.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.yangtools.yang.data.api.schema.stream;
9
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;
18
19 import java.io.IOException;
20 import java.util.Set;
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;
45
46 public class NormalizedNodeWriterTest {
47     private QNameModule bazModule;
48     private QName myKeyedList;
49     private QName myKeyLeaf;
50     private QName myLeafList;
51
52     @BeforeEach
53     public void setUp() {
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");
58     }
59
60     @Test
61     public void testNormalizedNodeWriter() throws IOException {
62         final NormalizedNodeStreamWriter loggingNormalizedNodeStreamWriter = new LoggingNormalizedNodeStreamWriter();
63         final NormalizedNodeWriter orderedNormalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
64                 loggingNormalizedNodeStreamWriter);
65
66         assertEquals(loggingNormalizedNodeStreamWriter, orderedNormalizedNodeWriter.getWriter());
67
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"));
71
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));
76
77         final LeafNode<?> mockedLeafNode = mock(LeafNode.class);
78         doReturn("leaf-value-1").when(mockedLeafNode).body();
79         assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafNode));
80
81         final DOMSourceAnyxmlNode mockedAnyXmlNode = mock(DOMSourceAnyxmlNode.class);
82         doCallRealMethod().when(mockedAnyXmlNode).bodyObjectModel();
83         doReturn(new DOMSource()).when(mockedAnyXmlNode).body();
84         assertNotNull(orderedNormalizedNodeWriter.write(mockedAnyXmlNode));
85
86         final NormalizedNode mockedContainerNode = mock(ContainerNode.class);
87         assertNotNull(orderedNormalizedNodeWriter.write(mockedContainerNode));
88
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));
94
95         final UnkeyedListEntryNode mockedUnkeyedListEntryNode = mock(UnkeyedListEntryNode.class);
96         assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListEntryNode));
97
98         assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
99
100         assertNotNull(orderedNormalizedNodeWriter.write(mock(AugmentationNode.class)));
101
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));
106
107         assertNotNull(orderedNormalizedNodeWriter.write(mock(UserMapNode.class)));
108
109         assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemMapNode.class)));
110
111         assertNotNull(orderedNormalizedNodeWriter.write(mock(UserLeafSetNode.class)));
112
113         assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemLeafSetNode.class)));
114
115         orderedNormalizedNodeWriter.flush();
116         orderedNormalizedNodeWriter.close();
117
118         final NormalizedNodeWriter normalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
119                 loggingNormalizedNodeStreamWriter, false);
120
121         assertNotNull(normalizedNodeWriter.write(mockedMapEntryNode));
122
123         normalizedNodeWriter.flush();
124         normalizedNodeWriter.close();
125     }
126 }