org.glassfish.jersey.server.spi.ContainerResponseWriter Java Examples

/**
 * Creates a new {@link ContainerRequest} for the given input.
 *
 * @param request
 * @param containerResponseWriter
 * @param securityContext
 * @return
 */
@Nonnull
protected ContainerRequest createContainerRequest(@Nonnull RequestT request,
		@Nonnull ContainerResponseWriter containerResponseWriter, @Nonnull SecurityContext securityContext) {
	requireNonNull(request, "request may not be null");
	URI baseUri = request.getBaseUri();
	URI requestUri = requireNonNull(request.getRequestUri(), "request.getRequestUri() may not be null");
	String httpMethod = requireNonNull(request.getHttpMethod(), "request.getHttpMethod() may not be null");
	InputStream entityStream = requireNonNull(request.getEntityStream(),
			"request.getEntityStream() may not be null");
	Map<String, List<String>> headers = requireNonNull(request.getHeaders(),
			"request.getHeaderParams() may not be null");
	requireNonNull(containerResponseWriter, "containerResponseWriter may not be null");
	requireNonNull(securityContext, "securityContext may not be null");

	ContainerRequest requestContext = new ContainerRequest(baseUri, requestUri, httpMethod, securityContext,
			new MapPropertiesDelegate());
	requestContext.setEntityStream(entityStream);
	requestContext.getHeaders().putAll(headers);
	requestContext.setWriter(containerResponseWriter);

	return requestContext;
}
 

@Test
public void createContainerRequest_RequestAndContainerResponseWriterAndSecurityContextGiven_ShouldCreateContainerRequestUsingValues() {
	URI baseUri = URI.create("/");
	URI requestUri = URI.create("/entity");
	String httpMethod = "POST";
	ByteArrayInputStream entityStream = new ByteArrayInputStream(new byte[0]);
	MultivaluedMap<String, String> headers = new MultivaluedHashMap<>();
	headers.add("hk0", "hv0_0");
	headers.add("hk0", "hv0_1");
	headers.add("hk1", "hv1_0");
	JRestlessContainerRequest request = createRequest(baseUri, requestUri, httpMethod, entityStream, headers);

	ContainerResponseWriter containerResponseWriter = mock(ContainerResponseWriter.class);
	SecurityContext securityContext = mock(SecurityContext.class);

	ContainerRequest containerRequest = container.createContainerRequest(request, containerResponseWriter, securityContext);

	assertEquals(baseUri, containerRequest.getBaseUri());
	assertEquals(requestUri, containerRequest.getRequestUri());
	assertEquals(httpMethod, containerRequest.getMethod());
	assertSame(entityStream, containerRequest.getEntityStream());
	assertEquals(headers, containerRequest.getHeaders());
	assertSame(containerResponseWriter, containerRequest.getResponseWriter());
	assertSame(securityContext, containerRequest.getSecurityContext());
}
 

@Override
public boolean suspend(long timeOut, TimeUnit timeUnit, final ContainerResponseWriter.TimeoutHandler timeoutHandler) {
    suspendTimeoutHandler = () -> timeoutHandler.onTimeout(this);
    if (timeOut <= 0) {
        return true;
    }
    suspendTimeoutFuture = container.getScheduledExecutorService().schedule(suspendTimeoutHandler, timeOut, timeUnit);
    return true;
}
 

@Override
public boolean suspend(long timeOut, TimeUnit timeUnit, final ContainerResponseWriter.TimeoutHandler timeoutHandler) {
    suspendTimeoutHandler = () -> timeoutHandler.onTimeout(this);
    if (timeOut <= 0) {
        return true;
    }
    suspendTimeoutFuture = getScheduledExecutorService().schedule(suspendTimeoutHandler, timeOut, timeUnit);
    return true;
}
 

@Test
public void createContainerRequest_NoRequestGiven_ShouldThrowNpe() {
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(null, mock(ContainerResponseWriter.class), mock(SecurityContext.class)));
}
 

@Test
public void createContainerRequest_NoSecurityContext_ShouldThrowNpe() {
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(createAnyRequest(), mock(ContainerResponseWriter.class), null));
}
 

public void createContainerRequest_NoRequestBaseUriGiven_ShouldPass() {
	JRestlessContainerRequest request = createAnyRequest();
	when(request.getBaseUri()).thenReturn(null);
	container.createContainerRequest(request, mock(ContainerResponseWriter.class), mock(SecurityContext.class));
}
 

@Test
public void createContainerRequest_NoRequestRequestUriGiven_ShouldThrowNpe() {
	JRestlessContainerRequest request = createAnyRequest();
	when(request.getRequestUri()).thenReturn(null);
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(request, mock(ContainerResponseWriter.class), mock(SecurityContext.class)));
}
 

@Test
public void createContainerRequest_NoRequestHttpMethodGiven_ShouldThrowNpe() {
	JRestlessContainerRequest request = createAnyRequest();
	when(request.getHttpMethod()).thenReturn(null);
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(request, mock(ContainerResponseWriter.class), mock(SecurityContext.class)));
}
 

@Test
public void createContainerRequest_NoRequestEntityStreamGiven_ShouldThrowNpe() {
	JRestlessContainerRequest request = createAnyRequest();
	when(request.getEntityStream()).thenReturn(null);
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(request, mock(ContainerResponseWriter.class), mock(SecurityContext.class)));
}
 

@Test
public void createContainerRequest_NoRequestHeadersGiven_ShouldThrowNpe() {
	JRestlessContainerRequest request = createAnyRequest();
	when(request.getHeaders()).thenReturn(null);
	assertThrows(NullPointerException.class, () -> container.createContainerRequest(request, mock(ContainerResponseWriter.class), mock(SecurityContext.class)));
}
 

/**
 * <p>setWriter.</p>
 *
 * @param responseWriter a {@link org.glassfish.jersey.server.spi.ContainerResponseWriter} object.
 */
public static void setWriter(ContainerResponseWriter responseWriter) {
    getRequest().setWriter(responseWriter);
}
 

/**
 * <p>getResponseWriter.</p>
 *
 * @return a {@link org.glassfish.jersey.server.spi.ContainerResponseWriter} object.
 */
public static ContainerResponseWriter getResponseWriter() {
    return getRequest().getResponseWriter();
}