Examples with HttpExchange com.sun.net.httpserver.HttpExchange used on opensource projects

Example 41 with HttpExchange

use of com.sun.net.httpserver.HttpExchange in project metrics by dropwizard.

the class InstrumentedHttpAsyncClientsTest method registersExpectedExceptionMetrics.

@Test
public void registersExpectedExceptionMetrics() throws Exception {
    client = InstrumentedHttpAsyncClients.custom(metricRegistry, metricNameStrategy).disableAutomaticRetries().build();
    client.start();
    final CountDownLatch countDownLatch = new CountDownLatch(1);
    final SimpleHttpRequest request = SimpleHttpRequests.get("http://localhost:" + httpServer.getAddress().getPort() + "/");
    final String requestMetricName = "request";
    final String exceptionMetricName = "exception";
    httpServer.createContext("/", HttpExchange::close);
    httpServer.start();
    when(metricNameStrategy.getNameFor(any(), any(HttpRequest.class))).thenReturn(requestMetricName);
    when(metricNameStrategy.getNameFor(any(), any(Exception.class))).thenReturn(exceptionMetricName);
    try {
        final Future<SimpleHttpResponse> responseFuture = client.execute(request, new FutureCallback<SimpleHttpResponse>() {

            @Override
            public void completed(SimpleHttpResponse result) {
                fail();
            }

            @Override
            public void failed(Exception ex) {
                countDownLatch.countDown();
            }

            @Override
            public void cancelled() {
                fail();
            }
        });
        countDownLatch.await(5, TimeUnit.SECONDS);
        responseFuture.get(5, TimeUnit.SECONDS);
        fail();
    } catch (ExecutionException e) {
        assertThat(e).hasCauseInstanceOf(ConnectionClosedException.class);
        await().atMost(5, TimeUnit.SECONDS).untilAsserted(() -> assertThat(metricRegistry.getMeters()).containsKey("exception"));
    }
}

Example 42 with HttpExchange

use of com.sun.net.httpserver.HttpExchange in project gradle by gradle.

the class GitHttpRepository method getRefsAction.

private Action<HttpExchange> getRefsAction() {
    return new ErroringAction<HttpExchange>() {

        @Override
        protected void doExecute(HttpExchange httpExchange) throws Exception {
            ProcessBuilder builder = new ProcessBuilder();
            Process process = builder.command("git", "upload-pack", "--advertise-refs", backingRepo.getWorkTree().getAbsolutePath()).redirectErrorStream(true).start();
            ByteArrayOutputStream content = new ByteArrayOutputStream();
            content.write("001e# service=git-upload-pack\n".getBytes());
            content.write("0000".getBytes());
            IOUtils.copy(process.getInputStream(), content);
            process.waitFor();
            int result = process.waitFor();
            if (result != 0) {
                throw new RuntimeException("Failed to run git upload-pack");
            }
            byte[] bytes = content.toByteArray();
            httpExchange.getResponseHeaders().add("content-type", "application/x-git-upload-pack-advertisement");
            httpExchange.sendResponseHeaders(200, bytes.length);
            httpExchange.getResponseBody().write(bytes);
        }
    };
}

Example 43 with HttpExchange

use of com.sun.net.httpserver.HttpExchange in project connect-java-library by urbanairship.

the class StreamConnectionTest method testAuth.

@Test
public void testAuth() throws Exception {
    final AtomicReference<String> authorization = new AtomicReference<>();
    final AtomicReference<String> appKeyHeader = new AtomicReference<>();
    final CountDownLatch received = new CountDownLatch(1);
    Answer httpAnswer = new Answer() {

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            HttpExchange exchange = (HttpExchange) invocation.getArguments()[0];
            authorization.set(exchange.getRequestHeaders().getFirst(HttpHeaders.AUTHORIZATION));
            appKeyHeader.set(exchange.getRequestHeaders().getFirst("X-UA-Appkey"));
            exchange.sendResponseHeaders(200, 0L);
            received.countDown();
            return null;
        }
    };
    doAnswer(httpAnswer).when(serverHandler).handle(Matchers.<HttpExchange>any());
    StreamQueryDescriptor descriptor = descriptor();
    stream = new StreamConnection(descriptor, http, connectionRetryStrategy, consumer, url);
    read(stream, Optional.<StartPosition>absent());
    assertTrue(received.await(10, TimeUnit.SECONDS));
    assertTrue(authorization.get().toLowerCase().startsWith("bearer"));
    String token = authorization.get().substring("bearer ".length());
    assertEquals(descriptor.getCreds().getAppKey(), appKeyHeader.get());
    assertEquals(descriptor.getCreds().getToken(), token);
}

Example 44 with HttpExchange

use of com.sun.net.httpserver.HttpExchange in project connect-java-library by urbanairship.

the class StreamConnectionTest method testCloseKillsConsume.

@Test
public void testCloseKillsConsume() throws Exception {
    Answer httpAnswer = new Answer() {

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            HttpExchange exchange = (HttpExchange) invocation.getArguments()[0];
            exchange.sendResponseHeaders(200, 0L);
            // hoping this is enough to force flushing data down the wire
            exchange.getResponseBody().write(randomAlphabetic(10000).getBytes(UTF_8));
            exchange.getResponseBody().write("\n".getBytes(UTF_8));
            exchange.getResponseBody().flush();
            return null;
        }
    };
    doAnswer(httpAnswer).when(serverHandler).handle(Matchers.<HttpExchange>any());
    final CountDownLatch latch = new CountDownLatch(1);
    Answer consumerAnswer = new Answer() {

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            latch.countDown();
            return null;
        }
    };
    doAnswer(consumerAnswer).when(consumer).accept(anyString());
    stream = new StreamConnection(descriptor(), http, connectionRetryStrategy, consumer, url);
    ExecutorService thread = Executors.newSingleThreadExecutor();
    Callable<Boolean> callable = new Callable<Boolean>() {

        @Override
        public Boolean call() throws Exception {
            stream.read(Optional.<StartPosition>absent());
            return Boolean.TRUE;
        }
    };
    try {
        Future<Boolean> future = thread.submit(callable);
        // Wait till we get something from the server
        latch.await(1, TimeUnit.MINUTES);
        // Now kill the stream
        stream.close();
        // Consume future should return without issue now
        try {
            future.get(1, TimeUnit.SECONDS);
        } catch (TimeoutException e) {
            fail();
        }
    } finally {
        thread.shutdownNow();
    }
}

Example 45 with HttpExchange

use of com.sun.net.httpserver.HttpExchange in project connect-java-library by urbanairship.

the class StreamConnectionTest method testExceptionDuringConsume.

@Test
public void testExceptionDuringConsume() throws Exception {
    Answer httpAnswer = new Answer() {

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            HttpExchange exchange = (HttpExchange) invocation.getArguments()[0];
            exchange.sendResponseHeaders(200, 0L);
            exchange.getResponseBody().write(randomAlphabetic(10).getBytes(UTF_8));
            exchange.getResponseBody().write("\n".getBytes(UTF_8));
            exchange.getResponseBody().flush();
            return null;
        }
    };
    doAnswer(httpAnswer).when(serverHandler).handle(Matchers.<HttpExchange>any());
    doThrow(new RuntimeException("boom")).when(consumer).accept(anyString());
    stream = new StreamConnection(descriptor(), http, connectionRetryStrategy, consumer, url);
    expectedException.expect(RuntimeException.class);
    stream.read(Optional.<StartPosition>absent());
    stream.wait(100);
}