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 org.junit.Before;
  10 import org.junit.Test;
  11 import org.mockito.Mock;
  12 import org.mockito.MockitoAnnotations;
  13 import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier;
  14 import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
  15 import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransaction;
  16 import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
  17 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  18 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  19
  20 public class LocalTransactionContextTest {
  21
  22     @Mock
  23     OperationLimiter limiter;
  24
  25     @Mock
  26     TransactionIdentifier identifier;
  27
  28     @Mock
  29     DOMStoreReadWriteTransaction readWriteTransaction;
  30
  31     LocalTransactionContext localTransactionContext;
  32
  33     @Before
  34     public void setUp(){
  35         MockitoAnnotations.initMocks(this);
  36         localTransactionContext = new LocalTransactionContext(identifier, readWriteTransaction, limiter) {
  37             @Override
  38             protected DOMStoreWriteTransaction getWriteDelegate() {
  39                 return readWriteTransaction;
  40             }
  41
  42             @Override
  43             protected DOMStoreReadTransaction getReadDelegate() {
  44                 return readWriteTransaction;
  45             }
  46         };
  47     }
  48
  49     @Test
  50     public void testWrite(){
  51         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  52         NormalizedNode<?, ?> normalizedNode = mock(NormalizedNode.class);
  53         localTransactionContext.writeData(yangInstanceIdentifier, normalizedNode);
  54         verify(limiter).release();
  55         verify(readWriteTransaction).write(yangInstanceIdentifier, normalizedNode);
  56     }
  57
  58     @Test
  59     public void testMerge(){
  60         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  61         NormalizedNode<?, ?> normalizedNode = mock(NormalizedNode.class);
  62         localTransactionContext.mergeData(yangInstanceIdentifier, normalizedNode);
  63         verify(limiter).release();
  64         verify(readWriteTransaction).merge(yangInstanceIdentifier, normalizedNode);
  65     }
  66
  67     @Test
  68     public void testDelete(){
  69         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  70         localTransactionContext.deleteData(yangInstanceIdentifier);
  71         verify(limiter).release();
  72         verify(readWriteTransaction).delete(yangInstanceIdentifier);
  73     }
  74
  75
  76     @Test
  77     public void testRead(){
  78         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  79         NormalizedNode<?, ?> normalizedNode = mock(NormalizedNode.class);
  80         doReturn(Futures.immediateCheckedFuture(Optional.of(normalizedNode))).when(readWriteTransaction).read(yangInstanceIdentifier);
  81         localTransactionContext.readData(yangInstanceIdentifier, SettableFuture.<Optional<NormalizedNode<?,?>>>create());
  82         verify(limiter).release();
  83         verify(readWriteTransaction).read(yangInstanceIdentifier);
  84     }
  85
  86     @Test
  87     public void testExists(){
  88         YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.builder().build();
  89         doReturn(Futures.immediateCheckedFuture(true)).when(readWriteTransaction).exists(yangInstanceIdentifier);
  90         localTransactionContext.dataExists(yangInstanceIdentifier, SettableFuture.<Boolean> create());
  91         verify(limiter).release();
  92         verify(readWriteTransaction).exists(yangInstanceIdentifier);
  93     }
  94
  95     @Test
  96     public void testReady(){
  97         doReturn(mock(LocalThreePhaseCommitCohort.class)).when(readWriteTransaction).ready();
  98         localTransactionContext.readyTransaction();
  99         verify(limiter).release();
 100         verify(readWriteTransaction).ready();
 101     }
 102
 103
 104 }