pekko/akka-docs/rst/java/code/docs/jrouting/RouterViaProgramDocTestBase.java

/**
 * Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
 */
package docs.jrouting;

import akka.actor.*;
import akka.remote.routing.RemoteRouterConfig;
import akka.routing.Broadcast;
import akka.routing.RoundRobinRouter;
import akka.testkit.JavaTestKit;
import docs.jrouting.RouterViaProgramExample.ExampleActor;
import docs.routing.RouterViaProgramDocSpec.Echo;
import java.io.Serializable;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

public class RouterViaProgramDocTestBase {

  static ActorSystem system;
  
  @BeforeClass
  public static void setup() {
    system = ActorSystem.create();
  }
  
  @AfterClass
  public static void teardown() {
    system.shutdown();
  }

  private static class JavaTestKitWithSelf extends JavaTestKit {
    public JavaTestKitWithSelf(ActorSystem system) {
      super(system);
    }
    /**
     * Wrap `getRef()` so our examples look like they're within a normal actor.
     */
    public ActorRef getSelf() {
      return getRef();
    }
  }

  @Test
  public void demonstrateRouteesFromPaths() {
    new JavaTestKit(system) {{
      //#programmaticRoutingRouteePaths      
      ActorRef actor1 = system.actorOf(new Props(ExampleActor.class), "actor1");
      ActorRef actor2 = system.actorOf(new Props(ExampleActor.class), "actor2");
      ActorRef actor3 = system.actorOf(new Props(ExampleActor.class), "actor3");
      Iterable<String> routees = Arrays.asList(
        new String[] { "/user/actor1", "/user/actor2", "/user/actor3" });
      ActorRef router = system.actorOf(
        new Props().withRouter(new RoundRobinRouter(routees)));
      //#programmaticRoutingRouteePaths
      for (int i = 1; i <= 6; i++) {
        router.tell(new ExampleActor.Message(i), null);
      }
    }};
  }

  @Test
  public void demonstrateBroadcast() {
    new JavaTestKitWithSelf(system) {{
      ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
      //#broadcastDavyJonesWarning
      router.tell(new Broadcast("Watch out for Davy Jones' locker"), getSelf());
      //#broadcastDavyJonesWarning
      receiveN(5, duration("5 seconds"));
    }};
  }

  @Test
  public void demonstratePoisonPill() {
    new JavaTestKitWithSelf(system) {{
      ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
      watch(router);
      //#poisonPill
      router.tell(PoisonPill.getInstance(), getSelf());
      //#poisonPill
      expectMsgClass(Terminated.class);
    }};
  }

  @Test
  public void demonstrateBroadcastOfPoisonPill() {
    new JavaTestKitWithSelf(system) {{
      ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
      watch(router);
      //#broadcastPoisonPill
      router.tell(new Broadcast(PoisonPill.getInstance()), getSelf());
      //#broadcastPoisonPill
        expectMsgClass(Terminated.class);
    }};
  }

  @Test
  public void demonstrateKill() {
    new JavaTestKitWithSelf(system) {{
      ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
      watch(router);
      //#kill
      router.tell(Kill.getInstance(), getSelf());
      //#kill
      expectMsgClass(Terminated.class);
    }};
  }

  @Test
  public void demonstrateBroadcastOfKill() {
    new JavaTestKitWithSelf(system) {{
      ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
      watch(router);
      //#broadcastKill
      router.tell(new Broadcast(Kill.getInstance()), getSelf());
      //#broadcastKill
      expectMsgClass(Terminated.class);
    }};
  }

}