+ public void testReady() {
+ final LocalThreePhaseCommitCohort mockCohort = mock(LocalThreePhaseCommitCohort.class);
+ doReturn(akka.dispatch.Futures.successful(null)).when(mockCohort).initiateCoordinatedCommit(Optional.empty());
+ doReturn(mockCohort).when(mockReadySupport).onTransactionReady(readWriteTransaction, null);
+
+ Future<ActorSelection> future = localTransactionContext.readyTransaction(null, 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.executeWrite(yangInstanceIdentifier, normalizedNode, null);
+ localTransactionContext.executeWrite(yangInstanceIdentifier, normalizedNode, null);
+
+ verify(readWriteTransaction).write(yangInstanceIdentifier, normalizedNode);
+
+ doReadyWithExpectedError(error);
+ }
+
+ @Test
+ public void testReadyWithMergeError() {
+ YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.empty();
+ NormalizedNode normalizedNode = mock(NormalizedNode.class);
+ RuntimeException error = new RuntimeException("mock");
+ doThrow(error).when(readWriteTransaction).merge(yangInstanceIdentifier, normalizedNode);
+
+ localTransactionContext.executeMerge(yangInstanceIdentifier, normalizedNode, null);
+ localTransactionContext.executeMerge(yangInstanceIdentifier, normalizedNode, null);
+
+ verify(readWriteTransaction).merge(yangInstanceIdentifier, normalizedNode);
+
+ doReadyWithExpectedError(error);