89 lines
2.4 KiB
Java
89 lines
2.4 KiB
Java
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
* license agreements; and to You under the Apache License, version 2.0:
|
|
*
|
|
* https://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* This file is part of the Apache Pekko project, which was derived from Akka.
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2009-2022 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package jdocs.actor.fsm;
|
|
|
|
import static jdocs.actor.fsm.Events.Batch;
|
|
import static jdocs.actor.fsm.Events.Flush.Flush;
|
|
import static jdocs.actor.fsm.Events.Queue;
|
|
import static jdocs.actor.fsm.Events.SetTarget;
|
|
|
|
import java.util.LinkedList;
|
|
import jdocs.AbstractJavaTest;
|
|
import org.apache.pekko.actor.ActorRef;
|
|
import org.apache.pekko.actor.ActorSystem;
|
|
import org.apache.pekko.actor.Props;
|
|
import org.apache.pekko.testkit.javadsl.TestKit;
|
|
import org.junit.AfterClass;
|
|
import org.junit.BeforeClass;
|
|
import org.junit.Test;
|
|
|
|
// #test-code
|
|
public class BuncherTest extends AbstractJavaTest {
|
|
|
|
static ActorSystem system;
|
|
|
|
@BeforeClass
|
|
public static void setup() {
|
|
system = ActorSystem.create("BuncherTest");
|
|
}
|
|
|
|
@AfterClass
|
|
public static void tearDown() {
|
|
TestKit.shutdownActorSystem(system);
|
|
system = null;
|
|
}
|
|
|
|
@Test
|
|
public void testBuncherActorBatchesCorrectly() {
|
|
new TestKit(system) {
|
|
{
|
|
final ActorRef buncher = system.actorOf(Props.create(Buncher.class));
|
|
final ActorRef probe = getRef();
|
|
|
|
buncher.tell(new SetTarget(probe), probe);
|
|
buncher.tell(new Queue(42), probe);
|
|
buncher.tell(new Queue(43), probe);
|
|
LinkedList<Object> list1 = new LinkedList<>();
|
|
list1.add(42);
|
|
list1.add(43);
|
|
expectMsgEquals(new Batch(list1));
|
|
buncher.tell(new Queue(44), probe);
|
|
buncher.tell(Flush, probe);
|
|
buncher.tell(new Queue(45), probe);
|
|
LinkedList<Object> list2 = new LinkedList<>();
|
|
list2.add(44);
|
|
expectMsgEquals(new Batch(list2));
|
|
LinkedList<Object> list3 = new LinkedList<>();
|
|
list3.add(45);
|
|
expectMsgEquals(new Batch(list3));
|
|
system.stop(buncher);
|
|
}
|
|
};
|
|
}
|
|
|
|
@Test
|
|
public void testBuncherActorDoesntBatchUninitialized() {
|
|
new TestKit(system) {
|
|
{
|
|
final ActorRef buncher = system.actorOf(Props.create(Buncher.class));
|
|
final ActorRef probe = getRef();
|
|
|
|
buncher.tell(new Queue(42), probe);
|
|
expectNoMessage();
|
|
system.stop(buncher);
|
|
}
|
|
};
|
|
}
|
|
}
|
|
// #test-code
|