2 * Copyright (c) 2014 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.model.util;
10 import com.google.common.collect.Sets;
12 import org.junit.Before;
13 import org.junit.Test;
14 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
15 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
16 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
17 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
19 import java.util.Collections;
20 import java.util.NoSuchElementException;
21 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
23 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
24 import org.opendaylight.yangtools.yang.model.api.Module;
25 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
26 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
28 import static org.junit.Assert.assertEquals;
29 import static org.junit.Assert.assertFalse;
30 import static org.junit.Assert.assertTrue;
31 import static org.mockito.Mockito.doReturn;
32 import static org.mockito.Mockito.mock;
34 public class DataNodeIteratorTest {
36 private DataNodeIterator dataNodeIterator;
40 DataNodeContainer dataNodeContainer = mock(DataNodeContainer.class);
41 this.dataNodeIterator = new DataNodeIterator(dataNodeContainer);
44 @Test(expected=IllegalArgumentException.class)
45 public void createDataNodeIteratorWithNullArgument() {
46 new DataNodeIterator(null);
49 @Test(expected=UnsupportedOperationException.class)
50 public void removeFromEmptyDataNodeContainer() {
51 dataNodeIterator.remove();
54 @Test(expected = NoSuchElementException.class)
55 public void tryNextOnEmptyDataContainer() {
56 dataNodeIterator.next();
60 public void createDataNodeIteratorWith() {
61 assertFalse("Has no next", dataNodeIterator.hasNext());
62 assertEquals("Should be empty list", Collections.EMPTY_LIST, dataNodeIterator.allChoices());
63 assertEquals("Should be empty list", Collections.EMPTY_LIST, dataNodeIterator.allContainers());
64 assertEquals("Should be empty list", Collections.EMPTY_LIST, dataNodeIterator.allTypedefs());
65 assertEquals("Should be empty list", Collections.EMPTY_LIST, dataNodeIterator.allGroupings());
66 assertEquals("Should be empty list", Collections.EMPTY_LIST, dataNodeIterator.allLists());
70 public void testTraversal() {
71 final Module mockedModule = mock(Module.class);
73 final ContainerSchemaNode mockedAugmentingContainer = mock(ContainerSchemaNode.class);
74 doReturn(true).when(mockedAugmentingContainer).isAugmenting();
76 final ContainerSchemaNode mockedContainer = mock(ContainerSchemaNode.class);
78 final ListSchemaNode mockedList = mock(ListSchemaNode.class);
80 final ChoiceSchemaNode mockedChoice = mock(ChoiceSchemaNode.class);
81 final ChoiceCaseNode mockedCase1 = mock(ChoiceCaseNode.class);
82 final ChoiceCaseNode mockedCase2 = mock(ChoiceCaseNode.class);
83 final Set<ChoiceCaseNode> cases = Sets.newHashSet(mockedCase1, mockedCase2);
84 doReturn(cases).when(mockedChoice).getCases();
86 final Set<DataSchemaNode> childNodes = Sets.newHashSet(mockedAugmentingContainer, mockedContainer, mockedList, mockedChoice);
87 doReturn(childNodes).when(mockedModule).getChildNodes();
89 final NotificationDefinition mockedNotification = mock(NotificationDefinition.class);
90 final ContainerSchemaNode mockedContainerInNotification = mock(ContainerSchemaNode.class);
91 final Set<DataSchemaNode> notificationChildNodes = Sets.newHashSet(mockedContainerInNotification);
92 doReturn(notificationChildNodes).when(mockedNotification).getChildNodes();
93 final Set<NotificationDefinition> notifications = Sets.newHashSet(mockedNotification);
95 doReturn(notifications).when(mockedModule).getNotifications();
97 final RpcDefinition mockedRpc = mock(RpcDefinition.class);
98 final ContainerSchemaNode mockedContainerInRpcInput = mock(ContainerSchemaNode.class);
99 final ListSchemaNode mockedListInRpcInputContainer = mock(ListSchemaNode.class);
100 final Set<DataSchemaNode> rpcInputChildNodes = Sets.newHashSet(mockedListInRpcInputContainer);
101 doReturn(rpcInputChildNodes).when(mockedContainerInRpcInput).getChildNodes();
102 doReturn(mockedContainerInRpcInput).when(mockedRpc).getInput();
103 final Set<RpcDefinition> rpcs = Sets.newHashSet(mockedRpc);
105 doReturn(rpcs).when(mockedModule).getRpcs();
107 final GroupingDefinition mockedGrouping = mock(GroupingDefinition.class);
108 final Set<GroupingDefinition> groupings = Sets.newHashSet(mockedGrouping);
110 doReturn(groupings).when(mockedModule).getGroupings();
112 final DataNodeIterator dataNodeIterator = new DataNodeIterator(mockedModule);
113 assertFalse(dataNodeIterator.allContainers().contains(mockedAugmentingContainer));
114 assertTrue(dataNodeIterator.allContainers().contains(mockedContainer));
115 assertTrue(dataNodeIterator.allLists().contains(mockedList));
116 assertTrue(dataNodeIterator.allChoices().contains(mockedChoice));
117 assertTrue(dataNodeIterator.allChoices().get(0).getCases().contains(mockedCase1));
118 assertTrue(dataNodeIterator.allChoices().get(0).getCases().contains(mockedCase2));
119 assertTrue(dataNodeIterator.allContainers().contains(mockedContainerInNotification));
120 assertTrue(dataNodeIterator.allLists().contains(mockedListInRpcInputContainer));
121 assertTrue(dataNodeIterator.allGroupings().contains(mockedGrouping));