Bug 4823: Use tx commit timeout for BatchedModifications
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / compat / PreBoronTransactionProxyTest.java
index 46f55f1760a3d0f1fe85e33d959475ae1638be47..0bf4f48cead2308e62482e7f0dc4c233e31e3646 100644 (file)
@@ -140,7 +140,7 @@ public class PreBoronTransactionProxyTest extends AbstractTransactionProxyTest {
 
         NormalizedNode<?, ?> expectedNode = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
         doReturn(readDataReply(expectedNode)).when(mockActorContext).executeOperationAsync(
-                eq(actorSelection(actorRef)), eqLegacySerializedReadData(TestModel.TEST_PATH));
+                eq(actorSelection(actorRef)), eqLegacySerializedReadData(TestModel.TEST_PATH), any(Timeout.class));
 
         TransactionProxy transactionProxy = new TransactionProxy(mockComponentFactory, READ_WRITE);
 
@@ -156,7 +156,7 @@ public class PreBoronTransactionProxyTest extends AbstractTransactionProxyTest {
         ActorRef actorRef = setupPreBoronActorContextWithInitialCreateTransaction(getSystem(), READ_WRITE);
 
         doReturn(dataExistsReply(true)).when(mockActorContext).executeOperationAsync(
-                eq(actorSelection(actorRef)), eqLegacySerializedDataExists());
+                eq(actorSelection(actorRef)), eqLegacySerializedDataExists(), any(Timeout.class));
 
         TransactionProxy transactionProxy = new TransactionProxy(mockComponentFactory, READ_WRITE);