1 package org.opendaylight.controller.cluster.datastore;
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;
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;
13 public class TransactionContextWrapperTest {
16 TransactionIdentifier identifier;
19 ActorContext actorContext;
22 TransactionContext transactionContext;
24 TransactionContextWrapper transactionContextWrapper;
28 MockitoAnnotations.initMocks(this);
29 doReturn(DatastoreContext.newBuilder().build()).when(actorContext).getDatastoreContext();
30 transactionContextWrapper = new TransactionContextWrapper(identifier, actorContext);
34 public void testExecutePriorTransactionOperations(){
35 for(int i=0;i<100;i++) {
36 transactionContextWrapper.maybeExecuteTransactionOperation(mock(TransactionOperation.class));
38 assertEquals(901, transactionContextWrapper.getLimiter().availablePermits());
40 transactionContextWrapper.executePriorTransactionOperations(transactionContext);
42 assertEquals(1001, transactionContextWrapper.getLimiter().availablePermits());