private Object executeShardOperationResponse;
private Object executeRemoteOperationResponse;
private Object executeLocalOperationResponse;
+ private Object executeLocalShardOperationResponse;
public MockActorContext(ActorSystem actorSystem) {
- super(actorSystem, null);
+ super(actorSystem, null, new MockClusterWrapper(), new MockConfiguration());
}
public MockActorContext(ActorSystem actorSystem, ActorRef shardManager) {
- super(actorSystem, shardManager);
+ super(actorSystem, shardManager, new MockClusterWrapper(), new MockConfiguration());
}
this.executeLocalOperationResponse = executeLocalOperationResponse;
}
+ public void setExecuteLocalShardOperationResponse(
+ Object executeLocalShardOperationResponse) {
+ this.executeLocalShardOperationResponse = executeLocalShardOperationResponse;
+ }
+
+ @Override public Object executeLocalOperation(ActorRef actor,
+ Object message, FiniteDuration duration) {
+ return this.executeLocalOperationResponse;
+ }
+ @Override public Object executeLocalShardOperation(String shardName,
+ Object message, FiniteDuration duration) {
+ return this.executeLocalShardOperationResponse;
+ }
}