import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.inOrder;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.inOrder;
import org.mockito.invocation.InvocationOnMock;
import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
import org.opendaylight.controller.cluster.datastore.persisted.CommitTransactionPayload;
import org.mockito.invocation.InvocationOnMock;
import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
import org.opendaylight.controller.cluster.datastore.persisted.CommitTransactionPayload;
-import org.opendaylight.controller.cluster.raft.protobuff.client.messages.Payload;
-import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
+import org.opendaylight.yangtools.yang.common.Empty;
+import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidate;
- final FutureCallback<Void> callback = mockCallback();
- doNothing().when(callback).onSuccess(null);
+ final FutureCallback<Empty> callback = mockCallback();
+ doNothing().when(callback).onSuccess(Empty.value());
- public static FutureCallback<Void> coordinatedCanCommit(final ShardDataTreeCohort cohort) {
- final FutureCallback<Void> callback = mockCallback();
- doNothing().when(callback).onSuccess(null);
+ public static FutureCallback<Empty> coordinatedCanCommit(final ShardDataTreeCohort cohort) {
+ final FutureCallback<Empty> callback = mockCallback();
+ doNothing().when(callback).onSuccess(Empty.value());
}).when(preCommitCallback).onSuccess(any(DataTreeCandidate.class));
doNothing().when(preCommitCallback).onFailure(any(Throwable.class));
}).when(preCommitCallback).onSuccess(any(DataTreeCandidate.class));
doNothing().when(preCommitCallback).onFailure(any(Throwable.class));
return null;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedCanCommit(final ShardDataTreeCohort mock) {
return null;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedCanCommit(final ShardDataTreeCohort mock) {
- doAnswer(invocation -> {
- return invokeFailure(invocation);
- }).when(mock).canCommit(any(FutureCallback.class));
+ doAnswer(ShardDataTreeMocking::invokeFailure).when(mock).canCommit(any(FutureCallback.class));
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedPreCommit(final ShardDataTreeCohort mock) {
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedPreCommit(final ShardDataTreeCohort mock) {
- doAnswer(invocation -> {
- return invokeFailure(invocation);
- }).when(mock).preCommit(any(FutureCallback.class));
+ doAnswer(ShardDataTreeMocking::invokeFailure).when(mock).preCommit(any(FutureCallback.class));
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedCommit(final ShardDataTreeCohort mock) {
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort failedCommit(final ShardDataTreeCohort mock) {
- doAnswer(invocation -> {
- return invokeFailure(invocation);
- }).when(mock).commit(any(FutureCallback.class));
+ doAnswer(ShardDataTreeMocking::invokeFailure).when(mock).commit(any(FutureCallback.class));
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulCanCommit(final ShardDataTreeCohort mock) {
return mock;
}
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulCanCommit(final ShardDataTreeCohort mock) {
- doAnswer(invocation -> {
- return invokeSuccess(invocation, null);
- }).when(mock).canCommit(any(FutureCallback.class));
+ doAnswer(invocation -> invokeSuccess(invocation, null)).when(mock).canCommit(any(FutureCallback.class));
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulPreCommit(final ShardDataTreeCohort mock,
final DataTreeCandidate candidate) {
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulPreCommit(final ShardDataTreeCohort mock,
final DataTreeCandidate candidate) {
- doAnswer(invocation -> {
- return invokeSuccess(invocation, candidate);
- }).when(mock).preCommit(any(FutureCallback.class));
+ doAnswer(invocation -> invokeSuccess(invocation, candidate)).when(mock).preCommit(any(FutureCallback.class));
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulCommit(final ShardDataTreeCohort mock, final UnsignedLong index) {
@SuppressWarnings("unchecked")
public static ShardDataTreeCohort successfulCommit(final ShardDataTreeCohort mock, final UnsignedLong index) {
- doAnswer(invocation -> {
- return invokeSuccess(invocation, index);
- }).when(mock).commit(any(FutureCallback.class));
+ doAnswer(invocation -> invokeSuccess(invocation, index)).when(mock).commit(any(FutureCallback.class));
- shardDataTree.applyReplicatedPayload(invocation.getArgumentAt(0, TransactionIdentifier.class),
- invocation.getArgumentAt(1, Payload.class));
+ shardDataTree.applyReplicatedPayload(invocation.getArgument(0), invocation.getArgument(1));
return null;
}).when(mockShard).persistPayload(any(TransactionIdentifier.class), any(CommitTransactionPayload.class),
anyBoolean());
return null;
}).when(mockShard).persistPayload(any(TransactionIdentifier.class), any(CommitTransactionPayload.class),
anyBoolean());