Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polishing in SubscriberInputStreamTests

Browse Source
This commit is contained in:
rstoyanchev
2024-10-31 10:03:36 +00:00
parent 4a0edc59cc
commit 38c88e769d
1 changed files with 23 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -26,8 +26,6 @@ import java.util.concurrent.Executors;
import java.util.concurrent.Flow;
import org.junit.jupiter.api.Test;
import org.reactivestreams.FlowAdapters;
import reactor.test.StepVerifier;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.assertj.core.api.Assertions.assertThat;
@@ -68,22 +66,7 @@ class SubscriberInputStreamTests {
@Test
void basic() {
Flow.Publisher<byte[]> publisher = new OutputStreamPublisher<>(
out -> {
out.write(FOO);
out.write(BAR);
out.write(BAZ);
},
this.byteMapper, this.executor, null);
StepVerifier.create(FlowAdapters.toPublisher(publisher))
.assertNext(s -> assertThat(s).containsExactly("foobarbaz".getBytes(UTF_8)))
.verifyComplete();
}
@Test
void flush() throws IOException {
void basic() throws IOException {
Flow.Publisher<byte[]> publisher = new OutputStreamPublisher<>(
out -> {
out.write(FOO);
@@ -115,7 +98,7 @@ class SubscriberInputStreamTests {
}
@Test
void chunkSize() {
void chunkSize() throws Exception {
Flow.Publisher<byte[]> publisher = new OutputStreamPublisher<>(
out -> {
out.write(FOO);
@@ -127,24 +110,25 @@ class SubscriberInputStreamTests {
try (SubscriberInputStream<byte[]> is = new SubscriberInputStream<>(s -> s, s -> {}, 1)) {
publisher.subscribe(is);
StringBuilder stringBuilder = new StringBuilder();
StringBuilder sb = new StringBuilder();
byte[] chunk = new byte[3];
stringBuilder.append(new String(new byte[]{(byte)is.read()}, UTF_8));
assertThat(is.read(chunk)).isEqualTo(3);
sb.append((char) is.read());
assertThat(sb).matches("f");
stringBuilder.append(new String(chunk, UTF_8));
assertThat(is.read(chunk)).isEqualTo(3);
sb.append(new String(chunk, UTF_8));
assertThat(sb).matches("foob");
assertThat(is.read(chunk)).isEqualTo(3);
sb.append(new String(chunk, UTF_8));
assertThat(sb).matches("foobarb");
stringBuilder.append(new String(chunk, UTF_8));
assertThat(is.read(chunk)).isEqualTo(2);
sb.append(new String(chunk,0, 2, UTF_8));
assertThat(sb).matches("foobarbaz");
stringBuilder.append(new String(chunk,0, 2, UTF_8));
assertThat(is.read()).isEqualTo(-1);
assertThat(stringBuilder.toString()).isEqualTo("foobarbaz");
}
catch (IOException e) {
throw new RuntimeException(e);
}
}
@@ -155,13 +139,13 @@ class SubscriberInputStreamTests {
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");
out.write(FOO);
out.flush();
out.write(BAR);
out.flush();
out.write(BAZ);
out.flush();
}).withMessage("Subscription has been terminated");
latch.countDown();
}, this.byteMapper, this.executor, null);
@@ -201,6 +185,7 @@ class SubscriberInputStreamTests {
assertThat(is.read(chunk)).isEqualTo(3);
assertThat(chunk).containsExactly(FOO);
assertThat(is.read(chunk)).isEqualTo(-1);
}
@@ -215,7 +200,8 @@ class SubscriberInputStreamTests {
out -> {
out.write(FOO);
out.flush();
assertThatIOException().isThrownBy(() -> {
assertThatIOException()
.isThrownBy(() -> {
out.write(BAR);
out.flush();
})