root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Improve cancel in SubscriberInputStreamTests

This commit is contained in:
rstoyanchev 2024-10-31 10:05:18 +00:00
parent 38c88e769d
commit bc65f93467
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
spring-web/src/test/java/org/springframework/http/client/SubscriberInputStreamTests.java
View File

@ -133,21 +133,24 @@ class SubscriberInputStreamTests {
}
@Test
void cancel() throws InterruptedException, IOException {
CountDownLatch latch = new CountDownLatch(1);
void cancel() throws Exception {
CountDownLatch latch1 = new CountDownLatch(1);
CountDownLatch latch2 = new CountDownLatch(1);
Flow.Publisher<byte[]> publisher = new OutputStreamPublisher<>(
out -> {
assertThatIOException().isThrownBy(() -> {
out.write(FOO);
out.flush();
out.write(BAR);
out.flush();
out.write(BAZ);
out.flush();
}).withMessage("Subscription has been terminated");
latch.countDown();
assertThatIOException()
.isThrownBy(() -> {
out.write(FOO);
out.flush();
out.write(BAR);
out.flush();
latch1.countDown();
out.write(BAZ);
out.flush();
})
.withMessage("Subscription has been terminated");
latch2.countDown();
}, this.byteMapper, this.executor, null);
List<byte[]> discarded = new ArrayList<>();
@ -158,10 +161,11 @@ class SubscriberInputStreamTests {
assertThat(is.read(chunk)).isEqualTo(3);
assertThat(chunk).containsExactly(FOO);
latch1.await();
}
latch.await();
latch2.await();
assertThat(discarded).containsExactly("bar".getBytes(UTF_8));
}