Signal/Signal-Server
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Server synced 2026-04-21 15:48:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

avoid baos::writeTo on virtual threads

Browse Source
This commit is contained in:
ravi-signal
2024-03-27 16:58:38 -05:00
committed by GitHub
parent a733f5c615
commit 37b657cbbd
3 changed files with 142 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
service/src/test/java/org/whispersystems/textsecuregcm/BufferingInterceptorIntegrationTest.java Normal file
View File

@@ -0,0 +1,77 @@
package org.whispersystems.textsecuregcm;
import static org.assertj.core.api.Assertions.assertThat;
import io.dropwizard.core.Application;
import io.dropwizard.core.Configuration;
import io.dropwizard.core.setup.Environment;
import io.dropwizard.testing.junit5.DropwizardAppExtension;
import io.dropwizard.testing.junit5.DropwizardExtensionsSupport;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import org.apache.commons.lang3.RandomStringUtils;
import org.eclipse.jetty.websocket.server.config.JettyWebSocketServletContainerInitializer;
import org.glassfish.jersey.server.ManagedAsync;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.whispersystems.textsecuregcm.util.BufferingInterceptor;
import org.whispersystems.textsecuregcm.util.VirtualExecutorServiceProvider;
@ExtendWith(DropwizardExtensionsSupport.class)
public class BufferingInterceptorIntegrationTest {
private static final DropwizardAppExtension<Configuration> DROPWIZARD_APP_EXTENSION =
new DropwizardAppExtension<>(TestApplication.class);
public static class TestApplication extends Application<Configuration> {
@Override
public void run(final Configuration configuration, final Environment environment) throws Exception {
final TestController testController = new TestController();
environment.jersey().register(testController);
environment.jersey().register(new BufferingInterceptor());
environment.jersey().register(new VirtualExecutorServiceProvider("virtual-thread-"));
JettyWebSocketServletContainerInitializer.configure(environment.getApplicationContext(), null);
}
}
@Test
public void testVirtual() {
final Response response = DROPWIZARD_APP_EXTENSION.client()
.target("http://127.0.0.1:%d/test/virtual/8".formatted(DROPWIZARD_APP_EXTENSION.getLocalPort()))
.request().get();
assertThat(response.getHeaders().getFirst(HttpHeaders.CONTENT_LENGTH)).isEqualTo("8");
}
@Test
public void testPlatform() {
final Response response = DROPWIZARD_APP_EXTENSION.client()
.target("http://127.0.0.1:%d/test/platform/8".formatted(DROPWIZARD_APP_EXTENSION.getLocalPort()))
.request().get();
assertThat(response.getHeaders().getFirst(HttpHeaders.CONTENT_LENGTH)).isEqualTo("8");
}
@Path("/test")
public static class TestController {
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("/virtual/{size}")
@ManagedAsync
public String getVirtual(@PathParam("size") int size) {
return RandomStringUtils.randomAscii(size);
}
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("/platform/{size}")
public String getPlatform(@PathParam("size") int size) {
return RandomStringUtils.randomAscii(size);
}
}
}