import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
+import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import org.junit.Before;
import org.junit.Test;
+import org.junit.runner.RunWith;
import org.mockito.InOrder;
import org.mockito.Mock;
import org.mockito.Mockito;
-import org.mockito.MockitoAnnotations;
+import org.mockito.junit.MockitoJUnitRunner;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
/**
* Unit test for {@link ParameterAwareNormalizedNodeWriter} used with fields parameter.
*/
+@RunWith(MockitoJUnitRunner.StrictStubs.class)
public class ParameterAwareNormalizedNodeWriterFieldsTest {
@Mock
private NodeIdentifier mapNodeIdentifier;
private NodeIdentifierWithPredicates mapEntryNodeIdentifier;
private NodeIdentifier leafSetNodeIdentifier;
- private NodeWithValue<?> leafSetEntryNodeIdentifier;
+ private NodeWithValue<String> leafSetEntryNodeIdentifier;
private NodeIdentifier keyLeafNodeIdentifier;
- private Collection<DataContainerChild<?, ?>> containerNodeValue;
+ private Collection<DataContainerChild> containerNodeValue;
private Collection<MapEntryNode> mapNodeValue;
- private Collection<DataContainerChild<?, ?>> mapEntryNodeValue;
+ private Collection<DataContainerChild> mapEntryNodeValue;
private Collection<LeafSetEntryNode<String>> leafSetNodeValue;
private String leafSetEntryNodeValue;
private String keyLeafNodeValue;
@Before
public void setUp() {
- MockitoAnnotations.initMocks(this);
-
// identifiers
containerNodeIdentifier = NodeIdentifier.create(QName.create("namespace", "container"));
Mockito.when(containerNodeData.getIdentifier()).thenReturn(containerNodeIdentifier);
- Mockito.when(containerNodeData.getNodeType()).thenReturn(containerNodeIdentifier.getNodeType());
mapNodeIdentifier = NodeIdentifier.create(QName.create("namespace", "list"));
Mockito.when(mapNodeData.getIdentifier()).thenReturn(mapNodeIdentifier);
- Mockito.when(mapNodeData.getNodeType()).thenReturn(mapNodeIdentifier.getNodeType());
final QName leafSetEntryNodeQName = QName.create("namespace", "leaf-set-entry");
leafSetEntryNodeValue = "leaf-set-value";
leafSetEntryNodeIdentifier = new NodeWithValue<>(leafSetEntryNodeQName, leafSetEntryNodeValue);
Mockito.when(leafSetEntryNodeData.getIdentifier()).thenReturn(leafSetEntryNodeIdentifier);
- Mockito.when(leafSetEntryNodeData.getNodeType()).thenReturn(leafSetEntryNodeIdentifier.getNodeType());
- Mockito.when(leafSetEntryNodeData.getNodeType()).thenReturn(leafSetEntryNodeQName);
leafSetNodeIdentifier = NodeIdentifier.create(QName.create("namespace", "leaf-set"));
Mockito.when(leafSetNodeData.getIdentifier()).thenReturn(leafSetNodeIdentifier);
- Mockito.when(leafSetNodeData.getNodeType()).thenReturn(leafSetNodeIdentifier.getNodeType());
final QName mapEntryNodeKey = QName.create("namespace", "key-field");
keyLeafNodeIdentifier = NodeIdentifier.create(mapEntryNodeKey);
keyLeafNodeValue = "key-value";
- mapEntryNodeIdentifier = new NodeIdentifierWithPredicates(
- QName.create("namespace", "list-entry"), Collections.singletonMap(mapEntryNodeKey, keyLeafNodeValue));
+ mapEntryNodeIdentifier = NodeIdentifierWithPredicates.of(
+ QName.create("namespace", "list-entry"), mapEntryNodeKey, keyLeafNodeValue);
Mockito.when(mapEntryNodeData.getIdentifier()).thenReturn(mapEntryNodeIdentifier);
- Mockito.when(mapEntryNodeData.getNodeType()).thenReturn(mapEntryNodeIdentifier.getNodeType());
- Mockito.when(mapEntryNodeData.getChild(keyLeafNodeIdentifier)).thenReturn(Optional.of(keyLeafNodeData));
+ Mockito.when(mapEntryNodeData.findChildByArg(keyLeafNodeIdentifier)).thenReturn(Optional.of(keyLeafNodeData));
- Mockito.when(keyLeafNodeData.getValue()).thenReturn(keyLeafNodeValue);
+ Mockito.when(keyLeafNodeData.body()).thenReturn(keyLeafNodeValue);
Mockito.when(keyLeafNodeData.getIdentifier()).thenReturn(keyLeafNodeIdentifier);
- Mockito.when(keyLeafNodeData.getNodeType()).thenReturn(keyLeafNodeIdentifier.getNodeType());
// values
- Mockito.when(leafSetEntryNodeData.getValue()).thenReturn(leafSetEntryNodeValue);
+ Mockito.when(leafSetEntryNodeData.body()).thenReturn(leafSetEntryNodeValue);
leafSetNodeValue = Collections.singletonList(leafSetEntryNodeData);
- Mockito.when(leafSetNodeData.getValue()).thenReturn(leafSetNodeValue);
+ Mockito.when(leafSetNodeData.body()).thenReturn(leafSetNodeValue);
containerNodeValue = Collections.singleton(leafSetNodeData);
- Mockito.when(containerNodeData.getValue()).thenReturn(containerNodeValue);
+ Mockito.when(containerNodeData.body()).thenReturn(containerNodeValue);
mapEntryNodeValue = Sets.newHashSet(keyLeafNodeData);
- Mockito.when(mapEntryNodeData.getValue()).thenReturn(mapEntryNodeValue);
+ Mockito.when(mapEntryNodeData.body()).thenReturn(mapEntryNodeValue);
mapNodeValue = Collections.singleton(mapEntryNodeData);
- Mockito.when(mapNodeData.getValue()).thenReturn(mapNodeValue);
+ Mockito.when(mapNodeData.body()).thenReturn(mapNodeValue);
}
/**
@Test
public void writeContainerWithLimitedFieldsTest() throws Exception {
final List<Set<QName>> limitedFields = new ArrayList<>();
- limitedFields.add(Sets.newHashSet());
+ limitedFields.add(new HashSet<>());
final ParameterAwareNormalizedNodeWriter parameterWriter = ParameterAwareNormalizedNodeWriter.forStreamWriter(
writer, null, limitedFields);
final InOrder inOrder = Mockito.inOrder(writer);
inOrder.verify(writer, Mockito.times(1)).startContainerNode(containerNodeIdentifier, containerNodeValue.size());
inOrder.verify(writer, Mockito.times(1)).startLeafSet(leafSetNodeIdentifier, leafSetNodeValue.size());
- inOrder.verify(writer, Mockito.times(1)).leafSetEntryNode(
- leafSetEntryNodeIdentifier.getNodeType(), leafSetEntryNodeValue);
- inOrder.verify(writer, Mockito.times(2)).endNode();
+ inOrder.verify(writer, Mockito.times(1)).startLeafSetEntryNode(leafSetEntryNodeIdentifier);
+ inOrder.verify(writer, Mockito.times(1)).scalarValue(leafSetEntryNodeValue);
+ inOrder.verify(writer, Mockito.times(3)).endNode();
Mockito.verifyNoMoreInteractions(writer);
}
@Test
public void writeMapEntryNodeWithLimitedFieldsTest() throws Exception {
final List<Set<QName>> limitedFields = new ArrayList<>();
- limitedFields.add(Sets.newHashSet());
+ limitedFields.add(new HashSet<>());
final ParameterAwareNormalizedNodeWriter parameterWriter = ParameterAwareNormalizedNodeWriter.forStreamWriter(
writer, null, limitedFields);
@Test
public void writeMapNodeAllFieldsTest() throws Exception {
final List<Set<QName>> limitedFields = new ArrayList<>();
- limitedFields.add(Sets.newHashSet(keyLeafNodeData.getNodeType()));
+ limitedFields.add(Sets.newHashSet(keyLeafNodeData.getIdentifier().getNodeType()));
final ParameterAwareNormalizedNodeWriter parameterWriter = ParameterAwareNormalizedNodeWriter.forStreamWriter(
writer, null, limitedFields);
final InOrder inOrder = Mockito.inOrder(writer);
inOrder.verify(writer, Mockito.times(1)).startMapNode(mapNodeIdentifier, mapNodeValue.size());
inOrder.verify(writer, Mockito.times(1)).startMapEntryNode(mapEntryNodeIdentifier, mapEntryNodeValue.size());
- inOrder.verify(writer, Mockito.times(1)).leafNode(keyLeafNodeIdentifier, keyLeafNodeValue);
- inOrder.verify(writer, Mockito.times(2)).endNode();
+ inOrder.verify(writer, Mockito.times(1)).startLeafNode(keyLeafNodeIdentifier);
+ inOrder.verify(writer, Mockito.times(1)).scalarValue(keyLeafNodeValue);
+ inOrder.verify(writer, Mockito.times(3)).endNode();
Mockito.verifyNoMoreInteractions(writer);
}
}
\ No newline at end of file