+ return message;
+ }
+ }
+
+ static final class SendImmediateHeartBeat implements ControlMessage {
+ static final SendImmediateHeartBeat INSTANCE = new SendImmediateHeartBeat();
+
+ private SendImmediateHeartBeat() {
+ }
+ }
+
+ static final class GetBehaviorState implements ControlMessage {
+ static final GetBehaviorState INSTANCE = new GetBehaviorState();
+
+ private GetBehaviorState() {
+ }
+ }
+
+ static class SetBehavior implements ControlMessage {
+ RaftActorBehavior behavior;
+ MockRaftActorContext context;
+
+ SetBehavior(final RaftActorBehavior behavior, final MockRaftActorContext context) {
+ this.behavior = behavior;
+ this.context = context;