.build(); String ticket = httpClient.execute(request, StringResponseHandler.createStringResponseHandler()).getBody();
@Test public void testTraceToken() { Request request = prepareGet().setUri(server.getBaseUrl().resolve("/testing/echo_token")).build(); StringResponse response = httpClient.execute(request, createStringResponseHandler()); assertEquals(response.getStatusCode(), SC_OK); assertEquals(response.getBody(), filter.getLastToken()); }
assertEquals(response.getStatusCode(), responseCode); assertEquals(response.getBody(), responseBody); assertEquals(response.getHeader("Content-Type"), responseContentType);
if (join.getStatusCode() != 200) { if (join.getBody().startsWith("Code: 44") || join.getBody().startsWith("Code: 57")) { if (tryCount > 0) { return getOrCreateCollectionFields(project, collection, fields, tryCount - 1); throw new IllegalStateException(join.getBody()); if (join.getStatusCode() != 200) { if (!getCollection(project, collection).stream().anyMatch(e -> e.getName().equals(f.getName()))) { throw new IllegalStateException(join.getBody());
assertEquals(response.getStatusCode(), responseCode); assertEquals(response.getBody(), responseBody); assertEquals(response.getHeader("Content-Type"), responseContentType);
.setHeader("User-Agent", nodeInfo.getNodeId()); StringResponse stringResponse = httpClient.execute(requestBuilder.build(), createStringResponseHandler()); if (stringResponse.getStatusCode() != 200) { logServerError(resourceUp, "Error loading %s file from %s: statusCode=%s statusMessage=%s", name, uri, stringResponse.getStatusCode(), stringResponse.getStatusMessage()); return ImmutableList.of(); contents = stringResponse.getBody();
private void executeRequest(ProjectCollection collection, List<SchemaField> schema, List<Event> events, CompletableFuture<Void> future, boolean tried) { HttpResponseFuture<StringResponse> f = HTTP_CLIENT.executeAsync(Request.builder() .setUri(buildInsertUri(collection, schema)) .setMethod("POST") .setBodyGenerator(new BinaryRawGenerator(events, schema)) .build(), createStringResponseHandler()); f.addListener(() -> { try { StringResponse stringResponse = f.get(1L, MINUTES); if (stringResponse.getStatusCode() == 200) { future.complete(null); } else { RuntimeException ex = new RuntimeException(stringResponse.getStatusMessage() + " : " + stringResponse.getBody().split("\n", 2)[0]); future.completeExceptionally(ex); } } catch (InterruptedException|ExecutionException|TimeoutException e) { if(!tried) { executeRequest(collection, schema, events, future, true); } else { future.completeExceptionally(e); LOGGER.error(e); } } }, Runnable::run); }
@Test(enabled = false, description = "This test requires a working socks proxy to run") public void testGetMethod() throws Exception { // To run this test: // // 1) create a ssh socks proxy with: // // ssh -N -D 1080 any.server.running.sshd // // 2) replace the uri below with a location only accessible from that host // Alternatively, run with the proxy up and once with the proxy down // URI uri = URI.create("http://example.com"); Request request = prepareGet() .setUri(uri) .build(); HttpClientConfig config = new HttpClientConfig(); config.setSocksProxy(fromParts("localhost", 1080)); try (NettyAsyncHttpClient client = new NettyAsyncHttpClient(config)) { StringResponse json = client.execute(request, createStringResponseHandler()); System.out.println(json.getBody()); } } }
@Test public void testClientCertificateJava() throws Exception { config.setHttpEnabled(false) .setAdminEnabled(false) .setHttpsEnabled(true) .setHttpsPort(0) .setKeystorePath(getResource("clientcert-java/server.keystore").getPath()) .setKeystorePassword("airlift"); createAndStartServer(createCertTestServlet()); HttpClientConfig clientConfig = new HttpClientConfig() .setKeyStorePath(getResource("clientcert-java/client.keystore").getPath()) .setKeyStorePassword("airlift") .setTrustStorePath(getResource("clientcert-java/client.truststore").getPath()) .setTrustStorePassword("airlift"); try (JettyHttpClient httpClient = new JettyHttpClient(clientConfig)) { StringResponse response = httpClient.execute( prepareGet().setUri(httpServerInfo.getHttpsUri()).build(), createStringResponseHandler()); assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); assertEquals(response.getBody(), "CN=testing,OU=Client,O=Airlift,L=Palo Alto,ST=CA,C=US"); } }
@Test public void testAuth() throws Exception { File file = File.createTempFile("auth", ".properties", tempDir); Files.write("user: password", file, Charsets.UTF_8); config.setUserAuthFile(file.getAbsolutePath()); createServer(); server.start(); try (HttpClient client = new JettyHttpClient()) { StringResponse response = client.execute( prepareGet() .setUri(httpServerInfo.getHttpUri()) .addHeader("Authorization", "Basic " + Base64.getEncoder().encodeToString("user:password".getBytes()).trim()) .build(), createStringResponseHandler()); assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); assertEquals(response.getBody(), "user"); } }
@Test public void testAuth() throws Exception { File file = File.createTempFile("auth", ".properties", tempDir); asCharSink(file, UTF_8).write("user: password"); config.setUserAuthFile(file.getAbsolutePath()); createServer(); server.start(); try (HttpClient client = new JettyHttpClient()) { StringResponse response = client.execute( prepareGet() .setUri(httpServerInfo.getHttpUri()) .addHeader("Authorization", "Basic " + Base64.getEncoder().encodeToString("user:password".getBytes()).trim()) .build(), createStringResponseHandler()); assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); assertEquals(response.getBody(), "user"); } }
@Test public void testPostAndGet() throws Throwable { String expected = "FooBarBaz"; StringResponse responsePost = client.execute( preparePost() .setUri(server.getBaseUrl().resolve("/temp-store")) .setBodyGenerator(createStaticBodyGenerator(expected.getBytes(Charsets.UTF_8))) .build(), createStringResponseHandler()); assertEquals(responsePost.getBody(), ""); assertEquals(responsePost.getStatusCode(), 201); StringResponse responseGet = client.execute( prepareGet().setUri(server.getBaseUrl().resolve("/temp-store")).build(), createStringResponseHandler()); assertEquals(responseGet.getBody(), expected); assertEquals(responseGet.getStatusCode(), 200); }
@Test public void testCompressionIsDisabled() throws Exception { Request request = prepareGet() .setUri(baseURI) .build(); String body = executeRequest(request, createStringResponseHandler()).getBody(); assertEquals(body, ""); assertFalse(servlet.getRequestHeaders().containsKey(HeaderName.of(ACCEPT_ENCODING))); String json = "{\"foo\":\"bar\",\"hello\":\"world\"}"; assertGreaterThanOrEqual(json.length(), GzipHandler.DEFAULT_MIN_GZIP_SIZE); servlet.setResponseBody(json); servlet.addResponseHeader(CONTENT_TYPE, "application/json"); StringResponse response = executeRequest(request, createStringResponseHandler()); assertEquals(response.getHeader(CONTENT_TYPE), "application/json"); assertEquals(response.getBody(), json); }
@Test public void testCompressionIsDisabled() throws Exception { Request request = prepareGet() .setUri(baseURI) .build(); String body = executeRequest(request, createStringResponseHandler()).getBody(); assertEquals(body, ""); assertFalse(servlet.getRequestHeaders().containsKey(HeaderName.of(ACCEPT_ENCODING))); String json = "{\"foo\":\"bar\",\"hello\":\"world\"}"; assertGreaterThanOrEqual(json.length(), GzipHandler.DEFAULT_MIN_GZIP_SIZE); servlet.setResponseBody(json); servlet.addResponseHeader(CONTENT_TYPE, "application/json"); StringResponse response = executeRequest(request, createStringResponseHandler()); assertEquals(response.getHeader(CONTENT_TYPE), "application/json"); assertEquals(response.getBody(), json); }
@Test public void testCompressionIsDisabled() throws Exception { Request request = prepareGet() .setUri(baseURI) .build(); String body = executeRequest(request, createStringResponseHandler()).getBody(); assertEquals(body, ""); assertFalse(servlet.getRequestHeaders().containsKey(HeaderName.of(ACCEPT_ENCODING))); String json = "{\"foo\":\"bar\",\"hello\":\"world\"}"; assertGreaterThanOrEqual(json.length(), GzipHandler.DEFAULT_MIN_GZIP_SIZE); servlet.setResponseBody(json); servlet.addResponseHeader(CONTENT_TYPE, "application/json"); StringResponse response = executeRequest(request, createStringResponseHandler()); assertEquals(response.getHeader(CONTENT_TYPE), "application/json"); assertEquals(response.getBody(), json); }
@Test public void testExecute() throws Exception { String expected = "{\"columns\":[" + "{\"name\":\"foo\",\"type\":\"bigint\",\"typeSignature\":{\"rawType\":\"bigint\",\"arguments\":[],\"typeArguments\":[],\"literalArguments\":[]}}," + "{\"name\":\"bar\",\"type\":\"varchar(3)\",\"typeSignature\":" + "{\"rawType\":\"varchar\",\"arguments\":[{\"kind\":\"LONG_LITERAL\",\"value\":3}],\"typeArguments\":[],\"literalArguments\":[]}}," + "{\"name\":\"baz\",\"type\":\"array(bigint)\",\"typeSignature\":" + "{\"rawType\":\"array\",\"arguments\":[{\"kind\":\"TYPE_SIGNATURE\",\"value\":{\"rawType\":\"bigint\",\"arguments\":[],\"typeArguments\":[],\"literalArguments\":[]}}],\"typeArguments\":[{\"rawType\":\"bigint\",\"arguments\":[],\"typeArguments\":[],\"literalArguments\":[]}],\"literalArguments\":[]}}]," + "\"data\":[[123,\"abc\",[42,44]]]}\n"; StringResponse response = executeQuery("SELECT 123 foo, 'abc' bar, CAST(JSON_PARSE('[42,44]') AS ARRAY<BIGINT>) baz"); assertEquals(response.getStatusCode(), HttpStatus.OK.code()); assertEquals(response.getHeader(HttpHeaders.CONTENT_TYPE), "application/json"); ObjectMapper mapper = new ObjectMapper(); assertEquals(mapper.readTree(response.getBody()), mapper.readTree(expected)); }
@Test public void testResponseIsClosed() throws Throwable { StringResponse response = client.execute( prepareGet().setUri(server.getBaseUrl().resolve("/closable-response")).build(), createStringResponseHandler()); assertEquals(response.getBody(), "hello"); assertEquals(response.getStatusCode(), 200); response = client.execute( prepareGet().setUri(server.getBaseUrl().resolve("/close-called")).build(), createStringResponseHandler()); assertEquals(response.getBody(), "true"); assertEquals(response.getStatusCode(), 200); } }
private JsonNode jsonpRequest(URI uri) throws IOException { uri = uriBuilderFrom(uri) .addParameter("jsonp", "test") .build(); Request request = prepareGet().setUri(uri).build(); StringResponse response = client.execute(request, createStringResponseHandler()); assertEquals(response.getStatusCode(), 200, response.getBody()); assertContentType(response, JSON_UTF_8); String jsonp = response.getBody().trim(); assertTrue(jsonp.startsWith("test("), jsonp); assertTrue(jsonp.endsWith(")"), jsonp); jsonp = jsonp.substring(5, jsonp.length() - 1); return new ObjectMapperProvider().get().readTree(jsonp); }
private JsonNode jsonpRequest(URI uri) throws IOException { uri = uriBuilderFrom(uri) .addParameter("jsonp", "test") .build(); Request request = prepareGet().setUri(uri).build(); StringResponse response = client.execute(request, createStringResponseHandler()); assertEquals(response.getStatusCode(), 200, response.getBody()); assertContentType(response, JSON_UTF_8); String jsonp = response.getBody().trim(); assertTrue(jsonp.startsWith("test("), jsonp); assertTrue(jsonp.endsWith(")"), jsonp); jsonp = jsonp.substring(5, jsonp.length() - 1); return new ObjectMapperProvider().get().readTree(jsonp); }
@Test public void testShowStackTraceDisabled() throws Exception { config.setShowStackTrace(false); createServer(new ErrorServlet()); server.start(); try (HttpClient client = new JettyHttpClient()) { StringResponse response = client.execute(prepareGet().setUri(httpServerInfo.getHttpUri()).build(), createStringResponseHandler()); assertEquals(response.getStatusCode(), 500); assertTrue(!response.getBody().contains("ErrorServlet.java")); } }