X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FOperationLimiterTest.java;h=d06a742c146d9b4e060096173a4deae0921e5fb8;hp=40ce84b234d033d2cda630105c631459bbf91be5;hb=7204c455a1636a7fc89bcd28fe9e9000eaa81b3b;hpb=c7e1ddeaf842ebb696c8dd38c0ca14c925ee31a1

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/OperationLimiterTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/OperationLimiterTest.java
index 40ce84b234..d06a742c14 100644
--- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/OperationLimiterTest.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/OperationLimiterTest.java
@@ -8,9 +8,9 @@
 package org.opendaylight.controller.cluster.datastore;
 
 import static org.junit.Assert.assertEquals;
-import java.util.concurrent.Semaphore;
+
 import org.junit.Test;
-import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
 import org.opendaylight.controller.cluster.datastore.messages.BatchedModificationsReply;
 import org.opendaylight.controller.cluster.datastore.messages.DataExistsReply;
 
@@ -20,26 +20,28 @@ import org.opendaylight.controller.cluster.datastore.messages.DataExistsReply;
  * @author Thomas Pantelis
  */
 public class OperationLimiterTest {
+    private final TransactionIdentifier transactionId = MockIdentifiers.transactionIdentifier(
+        OperationLimiterTest.class, "mock");
 
     @Test
     public void testOnComplete() throws Exception {
         int permits = 10;
-        OperationLimiter limiter = new OperationLimiter(new TransactionIdentifier("foo", 1), permits, 1);
-        Semaphore semaphore = limiter.getSemaphore();
-        semaphore.acquire(permits);
+        OperationLimiter limiter = new OperationLimiter(transactionId, permits, 1);
+        limiter.acquire(permits);
         int availablePermits = 0;
 
-        limiter.onComplete(null, DataExistsReply.create(true));
-        assertEquals("availablePermits", ++availablePermits, semaphore.availablePermits());
+        limiter.onComplete(null, new DataExistsReply(true, DataStoreVersions.CURRENT_VERSION));
+        assertEquals("availablePermits", ++availablePermits, limiter.availablePermits());
 
-        limiter.onComplete(null, DataExistsReply.create(true));
-        assertEquals("availablePermits", ++availablePermits, semaphore.availablePermits());
+        limiter.onComplete(null, new DataExistsReply(true, DataStoreVersions.CURRENT_VERSION));
+        assertEquals("availablePermits", ++availablePermits, limiter.availablePermits());
 
         limiter.onComplete(null, new IllegalArgumentException());
-        assertEquals("availablePermits", ++availablePermits, semaphore.availablePermits());
+        assertEquals("availablePermits", ++availablePermits, limiter.availablePermits());
 
         limiter.onComplete(null, new BatchedModificationsReply(4));
         availablePermits += 4;
-        assertEquals("availablePermits", availablePermits, semaphore.availablePermits());
+        assertEquals("availablePermits", availablePermits, limiter.availablePermits());
     }
+
 }