yang/yang-model-util/src/test/java/org/opendaylight/yangtools/yang/model/util/DataNodeIteratorTest.java

   1 /*
   2  * Copyright (c) 2014 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.model.util;
   9
  10 import com.google.common.collect.Sets;
  11 import java.util.Set;
  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;
  18
  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;
  27
  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;
  33
  34 public class DataNodeIteratorTest {
  35
  36     private DataNodeIterator dataNodeIterator;
  37
  38     @Before
  39     public void setUp() {
  40         DataNodeContainer dataNodeContainer = mock(DataNodeContainer.class);
  41         this.dataNodeIterator = new DataNodeIterator(dataNodeContainer);
  42     }
  43
  44     @Test(expected=IllegalArgumentException.class)
  45     public void createDataNodeIteratorWithNullArgument() {
  46         new DataNodeIterator(null);
  47     }
  48
  49     @Test(expected=UnsupportedOperationException.class)
  50     public void removeFromEmptyDataNodeContainer() {
  51         dataNodeIterator.remove();
  52     }
  53
  54     @Test(expected = NoSuchElementException.class)
  55     public void tryNextOnEmptyDataContainer() {
  56         dataNodeIterator.next();
  57     }
  58
  59     @Test
  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());
  67     }
  68
  69     @Test
  70     public void testTraversal() {
  71         final Module mockedModule = mock(Module.class);
  72
  73         final ContainerSchemaNode mockedAugmentingContainer = mock(ContainerSchemaNode.class);
  74         doReturn(true).when(mockedAugmentingContainer).isAugmenting();
  75
  76         final ContainerSchemaNode mockedContainer = mock(ContainerSchemaNode.class);
  77
  78         final ListSchemaNode mockedList = mock(ListSchemaNode.class);
  79
  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();
  85
  86         final Set<DataSchemaNode> childNodes = Sets.newHashSet(mockedAugmentingContainer, mockedContainer, mockedList, mockedChoice);
  87         doReturn(childNodes).when(mockedModule).getChildNodes();
  88
  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);
  94
  95         doReturn(notifications).when(mockedModule).getNotifications();
  96
  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);
 104
 105         doReturn(rpcs).when(mockedModule).getRpcs();
 106
 107         final GroupingDefinition mockedGrouping = mock(GroupingDefinition.class);
 108         final Set<GroupingDefinition> groupings = Sets.newHashSet(mockedGrouping);
 109
 110         doReturn(groupings).when(mockedModule).getGroupings();
 111
 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));
 122     }
 123 }