opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/PruningShardDataTreeSnapshotTest.java

   1 package org.opendaylight.controller.cluster.datastore;
   2
   3 import static org.junit.Assert.assertTrue;
   4 import static org.mockito.Mockito.verify;
   5 import java.net.URI;
   6 import java.util.Set;
   7 import org.junit.Before;
   8 import org.junit.Test;
   9 import org.mockito.Mock;
  10 import org.mockito.MockitoAnnotations;
  11 import org.opendaylight.controller.cluster.datastore.utils.PruningDataTreeModification;
  12 import org.opendaylight.controller.md.cluster.datastore.model.CarsModel;
  13 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
  14 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeSnapshot;
  15
  16 public class PruningShardDataTreeSnapshotTest {
  17
  18     @Mock
  19     DataTreeSnapshot dataTreeSnapshot;
  20
  21     @Mock
  22     Set<URI> validNamespaces;
  23
  24     @Before
  25     public void setUp(){
  26         MockitoAnnotations.initMocks(this);
  27     }
  28
  29     @Test
  30     public void testNewModification(){
  31         PruningShardDataTreeSnapshot snapshot1
  32                 = new PruningShardDataTreeSnapshot(dataTreeSnapshot, validNamespaces);
  33
  34         DataTreeModification dataTreeModification1 = snapshot1.newModification();
  35
  36         assertTrue(dataTreeModification1 instanceof PruningDataTreeModification);
  37     }
  38
  39     @Test
  40     public void testReadNode(){
  41         PruningShardDataTreeSnapshot snapshot
  42                 = new PruningShardDataTreeSnapshot(dataTreeSnapshot, validNamespaces);
  43
  44         snapshot.readNode(CarsModel.BASE_PATH);
  45
  46         verify(dataTreeSnapshot).readNode(CarsModel.BASE_PATH);
  47     }
  48 }