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

   1 package org.opendaylight.controller.cluster.datastore;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.mockito.Mockito.doReturn;
   5 import static org.mockito.Mockito.mock;
   6 import org.junit.Before;
   7 import org.junit.Test;
   8 import org.mockito.Mock;
   9 import org.mockito.MockitoAnnotations;
  10 import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier;
  11 import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
  12
  13 public class TransactionContextWrapperTest {
  14
  15     @Mock
  16     TransactionIdentifier identifier;
  17
  18     @Mock
  19     ActorContext actorContext;
  20
  21     @Mock
  22     TransactionContext transactionContext;
  23
  24     TransactionContextWrapper transactionContextWrapper;
  25
  26     @Before
  27     public void setUp(){
  28         MockitoAnnotations.initMocks(this);
  29         doReturn(DatastoreContext.newBuilder().build()).when(actorContext).getDatastoreContext();
  30         transactionContextWrapper = new TransactionContextWrapper(identifier, actorContext);
  31     }
  32
  33     @Test
  34     public void testExecutePriorTransactionOperations(){
  35         for(int i=0;i<100;i++) {
  36             transactionContextWrapper.maybeExecuteTransactionOperation(mock(TransactionOperation.class));
  37         }
  38         assertEquals(901, transactionContextWrapper.getLimiter().availablePermits());
  39
  40         transactionContextWrapper.executePriorTransactionOperations(transactionContext);
  41
  42         assertEquals(1001, transactionContextWrapper.getLimiter().availablePermits());
  43     }
  44 }