137 lines
5.1 KiB
Java
137 lines
5.1 KiB
Java
/**
|
|
* Copyright (C) 2009-2017 Lightbend Inc. <http://www.lightbend.com>
|
|
*/
|
|
package jdocs.tutorial_3;
|
|
|
|
import java.util.stream.Stream;
|
|
import java.util.stream.Collectors;
|
|
|
|
import akka.actor.ActorRef;
|
|
import akka.actor.ActorSystem;
|
|
import akka.actor.PoisonPill;
|
|
import akka.testkit.javadsl.TestKit;
|
|
|
|
import org.junit.AfterClass;
|
|
import org.junit.BeforeClass;
|
|
import org.junit.Test;
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertNotEquals;
|
|
|
|
import org.scalatest.junit.JUnitSuite;
|
|
|
|
public class DeviceGroupTest extends JUnitSuite {
|
|
|
|
static ActorSystem system;
|
|
|
|
@BeforeClass
|
|
public static void setup() {
|
|
system = ActorSystem.create();
|
|
}
|
|
|
|
@AfterClass
|
|
public static void teardown() {
|
|
TestKit.shutdownActorSystem(system);
|
|
system = null;
|
|
}
|
|
|
|
//#device-group-test-registration
|
|
@Test
|
|
public void testRegisterDeviceActor() {
|
|
TestKit probe = new TestKit(system);
|
|
ActorRef groupActor = system.actorOf(DeviceGroup.props("group"));
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device1"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
ActorRef deviceActor1 = probe.getLastSender();
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device2"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
ActorRef deviceActor2 = probe.getLastSender();
|
|
assertNotEquals(deviceActor1, deviceActor2);
|
|
|
|
// Check that the device actors are workingl
|
|
deviceActor1.tell(new Device.RecordTemperature(0L, 1.0), probe.getRef());
|
|
assertEquals(0L, probe.expectMsgClass(Device.TemperatureRecorded.class).requestId);
|
|
deviceActor2.tell(new Device.RecordTemperature(1L, 2.0), probe.getRef());
|
|
assertEquals(1L, probe.expectMsgClass(Device.TemperatureRecorded.class).requestId);
|
|
}
|
|
|
|
@Test
|
|
public void testIgnoreRequestsForWrongGroupId() {
|
|
TestKit probe = new TestKit(system);
|
|
ActorRef groupActor = system.actorOf(DeviceGroup.props("group"));
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("wrongGroup", "device1"), probe.getRef());
|
|
probe.expectNoMsg();
|
|
}
|
|
//#device-group-test-registration
|
|
|
|
//#device-group-test3
|
|
@Test
|
|
public void testReturnSameActorForSameDeviceId() {
|
|
TestKit probe = new TestKit(system);
|
|
ActorRef groupActor = system.actorOf(DeviceGroup.props("group"));
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device1"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
ActorRef deviceActor1 = probe.getLastSender();
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device1"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
ActorRef deviceActor2 = probe.getLastSender();
|
|
assertEquals(deviceActor1, deviceActor2);
|
|
}
|
|
//#device-group-test3
|
|
|
|
//#device-group-list-terminate-test
|
|
@Test
|
|
public void testListActiveDevices() {
|
|
TestKit probe = new TestKit(system);
|
|
ActorRef groupActor = system.actorOf(DeviceGroup.props("group"));
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device1"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device2"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
|
|
groupActor.tell(new DeviceGroup.RequestDeviceList(0L), probe.getRef());
|
|
DeviceGroup.ReplyDeviceList reply = probe.expectMsgClass(DeviceGroup.ReplyDeviceList.class);
|
|
assertEquals(0L, reply.requestId);
|
|
assertEquals(Stream.of("device1", "device2").collect(Collectors.toSet()), reply.ids);
|
|
}
|
|
|
|
@Test
|
|
public void testListActiveDevicesAfterOneShutsDown() {
|
|
TestKit probe = new TestKit(system);
|
|
ActorRef groupActor = system.actorOf(DeviceGroup.props("group"));
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device1"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
ActorRef toShutDown = probe.getLastSender();
|
|
|
|
groupActor.tell(new DeviceManager.RequestTrackDevice("group", "device2"), probe.getRef());
|
|
probe.expectMsgClass(DeviceManager.DeviceRegistered.class);
|
|
|
|
groupActor.tell(new DeviceGroup.RequestDeviceList(0L), probe.getRef());
|
|
DeviceGroup.ReplyDeviceList reply = probe.expectMsgClass(DeviceGroup.ReplyDeviceList.class);
|
|
assertEquals(0L, reply.requestId);
|
|
assertEquals(Stream.of("device1", "device2").collect(Collectors.toSet()), reply.ids);
|
|
|
|
probe.watch(toShutDown);
|
|
toShutDown.tell(PoisonPill.getInstance(), ActorRef.noSender());
|
|
probe.expectTerminated(toShutDown);
|
|
|
|
// using awaitAssert to retry because it might take longer for the groupActor
|
|
// to see the Terminated, that order is undefined
|
|
probe.awaitAssert(() -> {
|
|
groupActor.tell(new DeviceGroup.RequestDeviceList(1L), probe.getRef());
|
|
DeviceGroup.ReplyDeviceList r =
|
|
probe.expectMsgClass(DeviceGroup.ReplyDeviceList.class);
|
|
assertEquals(1L, r.requestId);
|
|
assertEquals(Stream.of("device2").collect(Collectors.toSet()), r.ids);
|
|
return null;
|
|
});
|
|
}
|
|
//#device-group-list-terminate-test
|
|
}
|