Split out sal-distributed-eos
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / TransactionRateLimitingCallbackTest.java
index 6e282e770564506d9fff03c59bf5d9ee85d8cca0..de3a78073f26cf5d72ee87998ed28a74229a28a0 100644 (file)
@@ -8,11 +8,12 @@
 package org.opendaylight.controller.cluster.datastore;
 
 import static org.junit.Assert.fail;
-import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyLong;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyLong;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.verify;
+
 import com.codahale.metrics.Timer;
 import com.google.common.base.Ticker;
 import java.util.concurrent.TimeUnit;
@@ -20,7 +21,7 @@ import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
-import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
+import org.opendaylight.controller.cluster.datastore.utils.ActorUtils;
 
 /**
  * Unit tests for TransactionRateLimitingCallback.
@@ -30,7 +31,7 @@ import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
 public class TransactionRateLimitingCallbackTest {
 
     @Mock
-    ActorContext mockContext;
+    ActorUtils mockContext;
 
     @Mock
     Timer mockTimer;
@@ -41,9 +42,9 @@ public class TransactionRateLimitingCallbackTest {
     TransactionRateLimitingCallback callback;
 
     @Before
-    public void setUp(){
+    public void setUp() {
         MockitoAnnotations.initMocks(this);
-        doReturn(mockTimer).when(mockContext).getOperationTimer(ActorContext.COMMIT);
+        doReturn(mockTimer).when(mockContext).getOperationTimer(ActorUtils.COMMIT);
         callback = new TransactionRateLimitingCallback(mockContext);
         TransactionRateLimitingCallback.setTicker(mockTicker);
     }
@@ -83,12 +84,12 @@ public class TransactionRateLimitingCallbackTest {
     }
 
     @Test
-    public void testSuccessWithoutRun(){
+    public void testSuccessWithoutRun() {
         try {
             callback.success();
             fail("Expected IllegalStateException");
-        } catch(IllegalStateException e){
-
+        } catch (IllegalStateException e) {
+            // expected
         }
 
         verify(mockTimer, never()).update(anyLong(), any(TimeUnit.class));