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

Fix mustBeAbleToUseMapWithAutoCloseableResource broken test (#1241)

Browse source 
Closes #1229.
This commit is contained in:
injae kim 2024-03-30 21:20:31 +09:00 • committed by GitHub
parent ecdea09145
commit 9e54e4695b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
stream-tests/src/test/java/org/apache/pekko/stream/javadsl/FlowTest.java
View file

@ -239,7 +239,7 @@ public class FlowTest extends StreamTest {
}
@Test
public void mustBeAbleToUseMapWithAutoCloseableResource() {
public void mustBeAbleToUseMapWithAutoCloseableResource() throws Exception {
final TestKit probe = new TestKit(system);
final AtomicInteger closed = new AtomicInteger();
Source.from(Arrays.asList("1", "2", "3"))
@ -247,7 +247,9 @@ public class FlowTest extends StreamTest {
Flow.of(String.class)
.mapWithResource(
() -> (AutoCloseable) closed::incrementAndGet, (resource, elem) -> elem))
.runWith(Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), system);
.runWith(Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), system)
.toCompletableFuture()
.get(3, TimeUnit.SECONDS);
probe.expectMsgAllOf("1", "2", "3");
Assert.assertEquals(closed.get(), 1);

6
stream-tests/src/test/java/org/apache/pekko/stream/javadsl/SourceTest.java
View file

@ -817,12 +817,14 @@ public class SourceTest extends StreamTest {
}
@Test
public void mustBeAbleToUseMapWithAutoCloseableResource() {
public void mustBeAbleToUseMapWithAutoCloseableResource() throws Exception {
final TestKit probe = new TestKit(system);
final AtomicInteger closed = new AtomicInteger();
Source.from(Arrays.asList("1", "2", "3"))
.mapWithResource(() -> (AutoCloseable) closed::incrementAndGet, (resource, elem) -> elem)
.runWith(Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), system);
.runWith(Sink.foreach(elem -> probe.getRef().tell(elem, ActorRef.noSender())), system)
.toCompletableFuture()
.get(3, TimeUnit.SECONDS);
probe.expectMsgAllOf("1", "2", "3");
Assert.assertEquals(closed.get(), 1);