bdc90052aa
* Add CopyrightHeader support for sbt-boilerplate plugin. * Add CopyrightHeader support for `*.proto` files. * Add regex match for both `–` and `-` for CopyrightHeader. * Add CopyrightHeader support for sbt build files. * Update copyright from 2018 to 2019.
95 lines
2.6 KiB
Java
95 lines
2.6 KiB
Java
/*
|
|
* Copyright (C) 2009-2019 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package jdocs.testkit;
|
|
|
|
//#fullsample
|
|
import jdocs.AbstractJavaTest;
|
|
import akka.testkit.javadsl.TestKit;
|
|
import org.junit.AfterClass;
|
|
import org.junit.Assert;
|
|
import org.junit.BeforeClass;
|
|
import org.junit.Test;
|
|
|
|
import akka.actor.ActorRef;
|
|
import akka.actor.ActorSystem;
|
|
import akka.actor.Props;
|
|
import akka.actor.AbstractActor;
|
|
|
|
import java.time.Duration;
|
|
|
|
public class TestKitSampleTest extends AbstractJavaTest {
|
|
|
|
public static class SomeActor extends AbstractActor {
|
|
ActorRef target = null;
|
|
|
|
@Override
|
|
public Receive createReceive() {
|
|
return receiveBuilder()
|
|
.matchEquals("hello", message -> {
|
|
getSender().tell("world", getSelf());
|
|
if (target != null) target.forward(message, getContext());
|
|
})
|
|
.match(ActorRef.class, actorRef -> {
|
|
target = actorRef;
|
|
getSender().tell("done", getSelf());
|
|
})
|
|
.build();
|
|
}
|
|
}
|
|
|
|
static ActorSystem system;
|
|
|
|
@BeforeClass
|
|
public static void setup() {
|
|
system = ActorSystem.create();
|
|
}
|
|
|
|
@AfterClass
|
|
public static void teardown() {
|
|
TestKit.shutdownActorSystem(system);
|
|
system = null;
|
|
}
|
|
|
|
@Test
|
|
public void testIt() {
|
|
/*
|
|
* Wrap the whole test procedure within a testkit constructor
|
|
* if you want to receive actor replies or use Within(), etc.
|
|
*/
|
|
new TestKit(system) {{
|
|
final Props props = Props.create(SomeActor.class);
|
|
final ActorRef subject = system.actorOf(props);
|
|
|
|
// can also use JavaTestKit “from the outside”
|
|
final TestKit probe = new TestKit(system);
|
|
// “inject” the probe by passing it to the test subject
|
|
// like a real resource would be passed in production
|
|
subject.tell(probe.getRef(), getRef());
|
|
// await the correct response
|
|
expectMsg(Duration.ofSeconds(1), "done");
|
|
|
|
// the run() method needs to finish within 3 seconds
|
|
within(Duration.ofSeconds(3), () -> {
|
|
subject.tell("hello", getRef());
|
|
|
|
// This is a demo: would normally use expectMsgEquals().
|
|
// Wait time is bounded by 3-second deadline above.
|
|
awaitCond(probe::msgAvailable);
|
|
|
|
// response must have been enqueued to us before probe
|
|
expectMsg(Duration.ZERO, "world");
|
|
// check that the probe we injected earlier got the msg
|
|
probe.expectMsg(Duration.ZERO, "hello");
|
|
Assert.assertEquals(getRef(), probe.getLastSender());
|
|
|
|
// Will wait for the rest of the 3 seconds
|
|
expectNoMessage();
|
|
return null;
|
|
});
|
|
}};
|
|
}
|
|
|
|
}
|
|
//#fullsample
|