raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Add missing java api for StreamTestKit (#1186)

Browse source 
* Add missing java api for StreamTestKit

* polish

* chore: Add more methods to streamKit

---------

Co-authored-by: naosense <pingao777@gmail.com>
This commit is contained in:
He-Pin(kerr) 2024-03-20 23:31:25 +08:00 • committed by GitHub
parent b4ff0fcabf
commit 55477ac6eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 290 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

3
docs/src/test/java/jdocs/stream/javadsl/cookbook/RecipeAdhocSourceTest.java
View file

@ -30,7 +30,6 @@ import org.junit.Ignore;
import org.junit.Test;
import scala.concurrent.Await;
import scala.concurrent.Promise;
import scala.concurrent.duration.FiniteDuration;
import java.time.Duration;
import java.util.concurrent.TimeoutException;
@ -235,7 +234,7 @@ public class RecipeAdhocSourceTest extends RecipeTest {
Thread.sleep(500);
assertEquals(BackpressureTimeoutException.class, probe.expectError().getClass());
probe.request(1); // send demand
probe.expectNoMessage(FiniteDuration.create(200, "milliseconds")); // but no more restart
probe.expectNoMessage(Duration.ofMillis(200)); // but no more restart
}
};
}

17
docs/src/test/java/jdocs/stream/javadsl/cookbook/RecipeHold.java
View file

@ -15,11 +15,17 @@ package jdocs.stream.javadsl.cookbook;
import org.apache.pekko.actor.ActorSystem;
import org.apache.pekko.japi.Pair;
import org.apache.pekko.stream.*;
import org.apache.pekko.stream.Attributes;
import org.apache.pekko.stream.FlowShape;
import org.apache.pekko.stream.Inlet;
import org.apache.pekko.stream.Outlet;
import org.apache.pekko.stream.javadsl.Keep;
import org.apache.pekko.stream.javadsl.Sink;
import org.apache.pekko.stream.javadsl.Source;
import org.apache.pekko.stream.stage.*;
import org.apache.pekko.stream.stage.AbstractInHandler;
import org.apache.pekko.stream.stage.AbstractOutHandler;
import org.apache.pekko.stream.stage.GraphStage;
import org.apache.pekko.stream.stage.GraphStageLogic;
import org.apache.pekko.stream.testkit.TestPublisher;
import org.apache.pekko.stream.testkit.TestSubscriber;
import org.apache.pekko.stream.testkit.javadsl.TestSink;
@ -28,9 +34,8 @@ import org.apache.pekko.testkit.javadsl.TestKit;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import scala.concurrent.duration.FiniteDuration;
import java.util.concurrent.TimeUnit;
import java.time.Duration;
public class RecipeHold extends RecipeTest {
static ActorSystem system;
@ -188,10 +193,8 @@ public class RecipeHold extends RecipeTest {
TestPublisher.Probe<Integer> pub = pubSub.first();
TestSubscriber.Probe<Integer> sub = pubSub.second();
FiniteDuration timeout = FiniteDuration.create(200, TimeUnit.MILLISECONDS);
sub.request(1);
sub.expectNoMessage(timeout);
sub.expectNoMessage(Duration.ofMillis(200));
pub.sendNext(1);
sub.expectNext(1);

12
docs/src/test/java/jdocs/stream/javadsl/cookbook/RecipeManualTrigger.java
View file

@ -15,7 +15,10 @@ package jdocs.stream.javadsl.cookbook;
import org.apache.pekko.actor.ActorSystem;
import org.apache.pekko.japi.Pair;
import org.apache.pekko.stream.*;
import org.apache.pekko.stream.ClosedShape;
import org.apache.pekko.stream.FanInShape2;
import org.apache.pekko.stream.FlowShape;
import org.apache.pekko.stream.SourceShape;
import org.apache.pekko.stream.javadsl.*;
import org.apache.pekko.stream.testkit.TestPublisher;
import org.apache.pekko.stream.testkit.TestSubscriber;
@ -25,10 +28,9 @@ import org.apache.pekko.testkit.javadsl.TestKit;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import scala.concurrent.duration.FiniteDuration;
import java.time.Duration;
import java.util.Arrays;
import java.util.concurrent.TimeUnit;
public class RecipeManualTrigger extends RecipeTest {
static ActorSystem system;
@ -85,7 +87,7 @@ public class RecipeManualTrigger extends RecipeTest {
TestPublisher.Probe<Trigger> pub = pubSub.first();
TestSubscriber.Probe<Message> sub = pubSub.second();
FiniteDuration timeout = FiniteDuration.create(100, TimeUnit.MILLISECONDS);
Duration timeout = Duration.ofMillis(100);
sub.expectSubscription().request(1000);
sub.expectNoMessage(timeout);
@ -140,7 +142,7 @@ public class RecipeManualTrigger extends RecipeTest {
TestPublisher.Probe<Trigger> pub = pubSub.first();
TestSubscriber.Probe<Message> sub = pubSub.second();
FiniteDuration timeout = FiniteDuration.create(100, TimeUnit.MILLISECONDS);
Duration timeout = Duration.ofMillis(100);
sub.expectSubscription().request(1000);
sub.expectNoMessage(timeout);

4
docs/src/test/java/jdocs/stream/javadsl/cookbook/RecipeMissedTicks.java
View file

@ -31,6 +31,7 @@ import org.junit.BeforeClass;
import org.junit.Test;
import scala.concurrent.Await;
import java.time.Duration;
import java.util.concurrent.TimeUnit;
public class RecipeMissedTicks extends RecipeTest {
@ -83,8 +84,7 @@ public class RecipeMissedTicks extends RecipeTest {
pub.sendNext(Tick);
pub.sendNext(Tick);
scala.concurrent.duration.FiniteDuration timeout =
scala.concurrent.duration.FiniteDuration.create(200, TimeUnit.MILLISECONDS);
Duration timeout = Duration.ofMillis(200);
Await.ready(latch, scala.concurrent.duration.Duration.create(1, TimeUnit.SECONDS));