Remove Augmentation{Identifier,Node}
[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.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;
44
45 public class NormalizedNodeWriterTest {
46     private QNameModule bazModule;
47     private QName myKeyedList;
48     private QName myKeyLeaf;
49     private QName myLeafList;
50
51     @BeforeEach
52     public void setUp() {
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");
57     }
58
59     @Test
60     public void testNormalizedNodeWriter() throws IOException {
61         final NormalizedNodeStreamWriter loggingNormalizedNodeStreamWriter = new LoggingNormalizedNodeStreamWriter();
62         final NormalizedNodeWriter orderedNormalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
63                 loggingNormalizedNodeStreamWriter);
64
65         assertEquals(loggingNormalizedNodeStreamWriter, orderedNormalizedNodeWriter.getWriter());
66
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"));
70
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));
75
76         final LeafNode<?> mockedLeafNode = mock(LeafNode.class);
77         doReturn("leaf-value-1").when(mockedLeafNode).body();
78         assertNotNull(orderedNormalizedNodeWriter.write(mockedLeafNode));
79
80         final DOMSourceAnyxmlNode mockedAnyXmlNode = mock(DOMSourceAnyxmlNode.class);
81         doCallRealMethod().when(mockedAnyXmlNode).bodyObjectModel();
82         doReturn(new DOMSource()).when(mockedAnyXmlNode).body();
83         assertNotNull(orderedNormalizedNodeWriter.write(mockedAnyXmlNode));
84
85         final NormalizedNode mockedContainerNode = mock(ContainerNode.class);
86         assertNotNull(orderedNormalizedNodeWriter.write(mockedContainerNode));
87
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));
93
94         final UnkeyedListEntryNode mockedUnkeyedListEntryNode = mock(UnkeyedListEntryNode.class);
95         assertNotNull(orderedNormalizedNodeWriter.write(mockedUnkeyedListEntryNode));
96
97         assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
98
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));
103
104         assertNotNull(orderedNormalizedNodeWriter.write(mock(UserMapNode.class)));
105
106         assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemMapNode.class)));
107
108         assertNotNull(orderedNormalizedNodeWriter.write(mock(UserLeafSetNode.class)));
109
110         assertNotNull(orderedNormalizedNodeWriter.write(mock(SystemLeafSetNode.class)));
111
112         orderedNormalizedNodeWriter.flush();
113         orderedNormalizedNodeWriter.close();
114
115         try (var nnWriter = NormalizedNodeWriter.forStreamWriter(loggingNormalizedNodeStreamWriter, false)) {
116             assertNotNull(nnWriter.write(mockedMapEntryNode));
117         }
118     }
119 }