Skip to content

Commit d4e106e

Browse files
committed
Fix flaky tests
1 parent 5fed8ca commit d4e106e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

test/codegen-generated-classes-test/src/test/java/software/amazon/awssdk/services/BlockingAsyncRequestResponseBodyResourceManagementTest.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import io.netty.channel.ChannelDuplexHandler;
2828
import io.netty.channel.ChannelHandlerContext;
2929
import io.netty.channel.ChannelInitializer;
30+
import io.netty.channel.ChannelOption;
3031
import io.netty.channel.ChannelPipeline;
3132
import io.netty.channel.nio.NioEventLoopGroup;
3233
import io.netty.channel.socket.ServerSocketChannel;
@@ -59,6 +60,7 @@
5960
import software.amazon.awssdk.core.ResponseInputStream;
6061
import software.amazon.awssdk.core.async.AsyncResponseTransformer;
6162
import software.amazon.awssdk.core.retry.RetryPolicy;
63+
import software.amazon.awssdk.http.nio.netty.NettyNioAsyncHttpClient;
6264
import software.amazon.awssdk.regions.Region;
6365
import software.amazon.awssdk.services.protocolrestjson.ProtocolRestJsonAsyncClient;
6466
import software.amazon.awssdk.services.protocolrestjson.model.StreamingOutputOperationRequest;
@@ -88,6 +90,7 @@ void setUpPerTest() throws Exception {
8890
.credentialsProvider(AnonymousCredentialsProvider.create())
8991
.endpointOverride(URI.create("http://localhost:" + server.port()))
9092
.overrideConfiguration(o -> o.retryPolicy(RetryPolicy.none()))
93+
.httpClientBuilder(NettyNioAsyncHttpClient.builder().putChannelOption(ChannelOption.SO_RCVBUF, 8))
9194
.build();
9295
}
9396

@@ -138,7 +141,7 @@ void verifyConnection(Consumer<ResponseInputStream<StreamingOutputOperationRespo
138141

139142
private static class Server extends ChannelInitializer<Channel> {
140143
private static final byte[] CONTENT = ("{ "
141-
+ "\"foo\": " + RandomStringUtils.randomAscii(1024)
144+
+ "\"foo\": " + RandomStringUtils.randomAscii(1024 * 1024 * 10)
142145
+ "}").getBytes(StandardCharsets.UTF_8);
143146
private ServerBootstrap bootstrap;
144147
private ServerSocketChannel serverSock;

0 commit comments

Comments
 (0)