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

   1 package org.opendaylight.controller.cluster.datastore;
   2
   3 import static org.mockito.Mockito.doReturn;
   4 import static org.mockito.Mockito.mock;
   5 import static org.mockito.Mockito.verify;
   6 import com.google.common.base.Optional;
   7 import com.google.common.util.concurrent.Futures;
   8 import com.google.common.util.concurrent.SettableFuture;
   9 import java.util.concurrent.Semaphore;
  10 import org.junit.Before;
  11 import org.junit.Test;
  12 import org.mockito.Mock;
  13 import org.mockito.MockitoAnnotations;
  14 import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier;
  15 import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransaction;
  16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  17 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  18
  19 public class LocalTransactionContextTest {
  20
  21     @Mock
  22     Semaphore limiter;
  23
  24     @Mock
  25     TransactionIdentifier identifier;
  26
  27     @Mock
  28     DOMStoreReadWriteTransaction readWriteTransaction;
  29
  30     LocalTransactionContext localTransactionContext;
  31
  32     @Before
  33     public void setUp(){
  34         MockitoAnnotations.initMocks(this);
  35         localTransactionContext = new LocalTransactionContext(identifier, readWriteTransaction, new OperationCompleter(limiter));
  36     }
  37
  38     @Test
  39     public void testWrite(){
  40         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  41         NormalizedNode normalizedNode = mock(NormalizedNode.class);
  42         localTransactionContext.writeData(yangInstanceIdentifier, normalizedNode);
  43         verify(limiter).release();
  44         verify(readWriteTransaction).write(yangInstanceIdentifier, normalizedNode);
  45     }
  46
  47     @Test
  48     public void testMerge(){
  49         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  50         NormalizedNode normalizedNode = mock(NormalizedNode.class);
  51         localTransactionContext.mergeData(yangInstanceIdentifier, normalizedNode);
  52         verify(limiter).release();
  53         verify(readWriteTransaction).merge(yangInstanceIdentifier, normalizedNode);
  54     }
  55
  56     @Test
  57     public void testDelete(){
  58         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  59         localTransactionContext.deleteData(yangInstanceIdentifier);
  60         verify(limiter).release();
  61         verify(readWriteTransaction).delete(yangInstanceIdentifier);
  62     }
  63
  64
  65     @Test
  66     public void testRead(){
  67         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  68         NormalizedNode normalizedNode = mock(NormalizedNode.class);
  69         doReturn(Futures.immediateCheckedFuture(Optional.of(normalizedNode))).when(readWriteTransaction).read(yangInstanceIdentifier);
  70         localTransactionContext.readData(yangInstanceIdentifier, SettableFuture.<Optional<NormalizedNode<?,?>>>create());
  71         verify(limiter).release();
  72         verify(readWriteTransaction).read(yangInstanceIdentifier);
  73     }
  74
  75     @Test
  76     public void testExists(){
  77         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  78         doReturn(Futures.immediateCheckedFuture(true)).when(readWriteTransaction).exists(yangInstanceIdentifier);
  79         localTransactionContext.dataExists(yangInstanceIdentifier, SettableFuture.<Boolean> create());
  80         verify(limiter).release();
  81         verify(readWriteTransaction).exists(yangInstanceIdentifier);
  82     }
  83
  84     @Test
  85     public void testReady(){
  86         doReturn(mock(LocalThreePhaseCommitCohort.class)).when(readWriteTransaction).ready();
  87         localTransactionContext.readyTransaction();
  88         verify(limiter).release();
  89         verify(readWriteTransaction).ready();
  90     }
  91
  92
  93 }