+ public void testReady() {
+ final LocalThreePhaseCommitCohort mockCohort = mock(LocalThreePhaseCommitCohort.class);
+ doReturn(akka.dispatch.Futures.successful(null)).when(mockCohort).initiateCoordinatedCommit();
+ doReturn(mockCohort).when(mockReadySupport).onTransactionReady(readWriteTransaction, null);
+
+ Future<ActorSelection> future = localTransactionContext.readyTransaction();
+ 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));
+ localTransactionContext.executeModification(new WriteModification(yangInstanceIdentifier, normalizedNode));
+
+ 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.executeModification(new MergeModification(yangInstanceIdentifier, normalizedNode));
+ localTransactionContext.executeModification(new MergeModification(yangInstanceIdentifier, normalizedNode));
+
+ verify(readWriteTransaction).merge(yangInstanceIdentifier, normalizedNode);
+
+ doReadyWithExpectedError(error);
+ }
+
+ @Test
+ public void testReadyWithDeleteError() {
+ YangInstanceIdentifier yangInstanceIdentifier = YangInstanceIdentifier.EMPTY;
+ RuntimeException error = new RuntimeException("mock");
+ doThrow(error).when(readWriteTransaction).delete(yangInstanceIdentifier);
+
+ localTransactionContext.executeModification(new DeleteModification(yangInstanceIdentifier));
+ localTransactionContext.executeModification(new DeleteModification(yangInstanceIdentifier));
+
+ verify(readWriteTransaction).delete(yangInstanceIdentifier);
+
+ doReadyWithExpectedError(error);