import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
+import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.opendaylight.controller.cluster.datastore.DataStoreVersions.CURRENT_VERSION;
import akka.actor.ActorRef;
MutableCompositeModification modification,
boolean doCommitOnReady) {
if(remoteReadWriteTransaction){
- return new ForwardedReadyTransaction(transactionID, CURRENT_VERSION,
- cohort, modification, true, doCommitOnReady);
+ return prepareForwardedReadyTransaction(cohort, transactionID, CURRENT_VERSION,
+ doCommitOnReady);
} else {
setupCohortDecorator(shard, cohort);
return prepareBatchedModifications(transactionID, modification, doCommitOnReady);
}
}
+ static ShardDataTreeTransactionParent newShardDataTreeTransactionParent(ShardDataTreeCohort cohort) {
+ ShardDataTreeTransactionParent mockParent = mock(ShardDataTreeTransactionParent.class);
+ doReturn(cohort).when(mockParent).finishTransaction(any(ReadWriteShardDataTreeTransaction.class));
+ doNothing().when(mockParent).abortTransaction(any(AbstractShardDataTreeTransaction.class));
+ return mockParent;
+ }
+
+ protected ForwardedReadyTransaction prepareForwardedReadyTransaction(ShardDataTreeCohort cohort,
+ String transactionID, short version, boolean doCommitOnReady) {
+ return new ForwardedReadyTransaction(transactionID, version,
+ new ReadWriteShardDataTreeTransaction(newShardDataTreeTransactionParent(cohort), transactionID,
+ mock(DataTreeModification.class)), true, doCommitOnReady);
+ }
+
protected Object prepareReadyTransactionMessage(boolean remoteReadWriteTransaction, Shard shard, ShardDataTreeCohort cohort,
String transactionID,
MutableCompositeModification modification) {
return prepareBatchedModifications(transactionID, modification, false);
}
- private BatchedModifications prepareBatchedModifications(String transactionID,
+ private static BatchedModifications prepareBatchedModifications(String transactionID,
MutableCompositeModification modification,
boolean doCommitOnReady) {
final BatchedModifications batchedModifications = new BatchedModifications(transactionID, CURRENT_VERSION, null);