Add example of testing actors with subclassing (#24539)

This commit is contained in:
Arnout Engelen 2018-02-13 19:29:30 +01:00 committed by Johan Andrén
parent 3bd3f00c41
commit f2e39c7534
3 changed files with 97 additions and 1 deletions

View file

@ -21,6 +21,7 @@ import org.junit.Test;
import com.typesafe.config.ConfigFactory;
import akka.actor.AbstractActorWithTimers;
import akka.actor.ActorKilledException;
import akka.actor.ActorRef;
import akka.actor.ActorSystem;
@ -29,6 +30,7 @@ import akka.actor.PoisonPill;
import akka.actor.Props;
import akka.actor.Terminated;
import akka.actor.AbstractActor;
import akka.actor.AbstractActor.Receive;
import akka.testkit.TestActor.AutoPilot;
import scala.concurrent.duration.Duration;
import scala.concurrent.duration.FiniteDuration;
@ -37,6 +39,7 @@ import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;
public class TestKitDocTest extends AbstractJavaTest {
@ -72,6 +75,29 @@ public class TestKitDocTest extends AbstractJavaTest {
}
//#test-actor-ref
//#timer
static class TestTimerActor extends AbstractActorWithTimers {
private static Object SCHED_KEY = "SchedKey";
static final class TriggerScheduling {
}
static final class ScheduledMessage {
}
@Override
public Receive createReceive() {
return receiveBuilder()
.match(TriggerScheduling.class, msg -> triggerScheduling())
.build();
}
void triggerScheduling() {
getTimers().startSingleTimer(
SCHED_KEY,
new ScheduledMessage(),
Duration.create(500, TimeUnit.MILLISECONDS)
);
}
}
//#timer
@Test
public void demonstrateAsk() throws Exception {
//#test-behavior
@ -258,7 +284,7 @@ public class TestKitDocTest extends AbstractJavaTest {
.build();
}
}
// create a test probe
final TestKit probe = new TestKit(system);
@ -349,6 +375,23 @@ public class TestKitDocTest extends AbstractJavaTest {
//#test-probe-forward
}
@Test
public void demonstrateUsingInheritenceToTestTimers() {
//#timer-test
new TestKit(system) {{
final TestKit probe = new TestKit(system);
final ActorRef target = system.actorOf(Props.create(TestTimerActor.class, () -> new TestTimerActor() {
@Override
void triggerScheduling() {
probe.getRef().tell(new ScheduledMessage(), getSelf());
}
}));
target.tell(new TestTimerActor.TriggerScheduling(), ActorRef.noSender());
probe.expectMsgClass(TestTimerActor.ScheduledMessage.class);
}};
//#timer-test
}
@Test
public void demonstrateWithinProbe() {
try {