- final ActorContext mockContext = mock(ActorContext.class);
- doReturn(mockContext).when(mockCohort).getActorContext();
- doReturn(ExecutionContexts.fromExecutor(MoreExecutors.directExecutor())).when(mockContext).getClientDispatcher();
- doReturn(akka.dispatch.Futures.successful(null)).when(mockCohort).initiateCoordinatedCommit();
- doReturn(mockCohort).when(readWriteTransaction).ready();
- localTransactionContext.readyTransaction();
- verify(readWriteTransaction).ready();
+ doReturn(akka.dispatch.Futures.successful(null)).when(mockCohort).initiateCoordinatedCommit(
+ java.util.Optional.empty());
+ doReturn(mockCohort).when(mockReadySupport).onTransactionReady(readWriteTransaction, null);
+
+ Future<ActorSelection> future = localTransactionContext.readyTransaction(null, java.util.Optional.empty());
+ assertTrue(future.isCompleted());
+
+ verify(mockReadySupport).onTransactionReady(readWriteTransaction, null);
+ }
+
+ @Test
+ public void testReadyWithWriteError() {
+ YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.EMPTY;
+ NormalizedNode<?, ?> normalizedNode = mock(NormalizedNode.class);
+ RuntimeException error = new RuntimeException("mock");
+ doThrow(error).when(readWriteTransaction).write(yangInstanceIdentifier, normalizedNode);
+
+ localTransactionContext.executeModification(new WriteModification(yangInstanceIdentifier, normalizedNode),
+ null);
+ localTransactionContext.executeModification(new WriteModification(yangInstanceIdentifier, normalizedNode),
+ null);
+
+ verify(readWriteTransaction).write(yangInstanceIdentifier, normalizedNode);
+
+ doReadyWithExpectedError(error);