Java 类org.apache.http.message.BasicStatusLine 实例源码

项目：elasticsearch_my 文件：HeapBufferedAsyncResponseConsumerTests.java

public void testResponseProcessing() throws Exception {
    ContentDecoder contentDecoder = mock(ContentDecoder.class);
    IOControl ioControl = mock(IOControl.class);
    HttpContext httpContext = mock(HttpContext.class);

    HeapBufferedAsyncResponseConsumer consumer = spy(new HeapBufferedAsyncResponseConsumer(TEST_BUFFER_LIMIT));

    ProtocolVersion protocolVersion = new ProtocolVersion("HTTP", 1, 1);
    StatusLine statusLine = new BasicStatusLine(protocolVersion, 200, "OK");
    HttpResponse httpResponse = new BasicHttpResponse(statusLine);
    httpResponse.setEntity(new StringEntity("test", ContentType.TEXT_PLAIN));

    //everything goes well
    consumer.responseReceived(httpResponse);
    consumer.consumeContent(contentDecoder, ioControl);
    consumer.responseCompleted(httpContext);

    verify(consumer).releaseResources();
    verify(consumer).buildResult(httpContext);
    assertTrue(consumer.isDone());
    assertSame(httpResponse, consumer.getResult());

    consumer.responseCompleted(httpContext);
    verify(consumer, times(1)).releaseResources();
    verify(consumer, times(1)).buildResult(httpContext);
}

项目：elasticsearch_my 文件：HeapBufferedAsyncResponseConsumerTests.java

private static void bufferLimitTest(HeapBufferedAsyncResponseConsumer consumer, int bufferLimit) throws Exception {
    ProtocolVersion protocolVersion = new ProtocolVersion("HTTP", 1, 1);
    StatusLine statusLine = new BasicStatusLine(protocolVersion, 200, "OK");
    consumer.onResponseReceived(new BasicHttpResponse(statusLine));

    final AtomicReference<Long> contentLength = new AtomicReference<>();
    HttpEntity entity = new StringEntity("", ContentType.APPLICATION_JSON) {
        @Override
        public long getContentLength() {
            return contentLength.get();
        }
    };
    contentLength.set(randomLong(bufferLimit));
    consumer.onEntityEnclosed(entity, ContentType.APPLICATION_JSON);

    contentLength.set(randomLongBetween(bufferLimit + 1, MAX_TEST_BUFFER_SIZE));
    try {
        consumer.onEntityEnclosed(entity, ContentType.APPLICATION_JSON);
    } catch(ContentTooLongException e) {
        assertEquals("entity content is too long [" + entity.getContentLength() +
                "] for the configured buffer limit [" + bufferLimit + "]", e.getMessage());
    }
}

项目：elasticsearch_my 文件：RestClientMultipleHostsTests.java

@Before
@SuppressWarnings("unchecked")
public void createRestClient() throws IOException {
    CloseableHttpAsyncClient httpClient = mock(CloseableHttpAsyncClient.class);
    when(httpClient.<HttpResponse>execute(any(HttpAsyncRequestProducer.class), any(HttpAsyncResponseConsumer.class),
           any(HttpClientContext.class), any(FutureCallback.class))).thenAnswer(new Answer<Future<HttpResponse>>() {
        @Override
        public Future<HttpResponse> answer(InvocationOnMock invocationOnMock) throws Throwable {
            HttpAsyncRequestProducer requestProducer = (HttpAsyncRequestProducer) invocationOnMock.getArguments()[0];
            HttpUriRequest request = (HttpUriRequest)requestProducer.generateRequest();
            HttpHost httpHost = requestProducer.getTarget();
            HttpClientContext context = (HttpClientContext) invocationOnMock.getArguments()[2];
            assertThat(context.getAuthCache().get(httpHost), instanceOf(BasicScheme.class));
            FutureCallback<HttpResponse> futureCallback = (FutureCallback<HttpResponse>) invocationOnMock.getArguments()[3];
            //return the desired status code or exception depending on the path
            if (request.getURI().getPath().equals("/soe")) {
                futureCallback.failed(new SocketTimeoutException(httpHost.toString()));
            } else if (request.getURI().getPath().equals("/coe")) {
                futureCallback.failed(new ConnectTimeoutException(httpHost.toString()));
            } else if (request.getURI().getPath().equals("/ioe")) {
                futureCallback.failed(new IOException(httpHost.toString()));
            } else {
                int statusCode = Integer.parseInt(request.getURI().getPath().substring(1));
                StatusLine statusLine = new BasicStatusLine(new ProtocolVersion("http", 1, 1), statusCode, "");
                futureCallback.completed(new BasicHttpResponse(statusLine));
            }
            return null;
        }
    });
    int numHosts = RandomNumbers.randomIntBetween(getRandom(), 2, 5);
    httpHosts = new HttpHost[numHosts];
    for (int i = 0; i < numHosts; i++) {
        httpHosts[i] = new HttpHost("localhost", 9200 + i);
    }
    failureListener = new HostsTrackingFailureListener();
    restClient = new RestClient(httpClient, 10000, new Header[0], httpHosts, null, failureListener);
}

项目：apigateway-generic-java-sdk 文件：GenericApiGatewayClientTest.java

@Test
public void testExecute_non2xx_exception() throws IOException {
    HttpResponse resp = new BasicHttpResponse(new BasicStatusLine(HttpVersion.HTTP_1_1, 404, "Not found"));
    BasicHttpEntity entity = new BasicHttpEntity();
    entity.setContent(new ByteArrayInputStream("{\"message\" : \"error payload\"}".getBytes()));
    resp.setEntity(entity);
    Mockito.doReturn(resp).when(mockClient).execute(any(HttpUriRequest.class), any(HttpContext.class));

    Map<String, String> headers = new HashMap<>();
    headers.put("Account-Id", "fubar");
    headers.put("Content-Type", "application/json");

    try {
        client.execute(
                new GenericApiGatewayRequestBuilder()
                        .withBody(new ByteArrayInputStream("test request".getBytes()))
                        .withHttpMethod(HttpMethodName.POST)
                        .withHeaders(headers)
                        .withResourcePath("/test/orders").build());

        Assert.fail("Expected exception");
    } catch (GenericApiGatewayException e) {
        assertEquals("Wrong status code", 404, e.getStatusCode());
        assertEquals("Wrong exception message", "{\"message\":\"error payload\"}", e.getErrorMessage());
    }
}

项目：aws-xray-sdk-java 文件：TracedResponseHandlerTest.java

private Segment segmentInResponseToCode(int code) {
    NoOpResponseHandler responseHandler = new NoOpResponseHandler();
    TracedResponseHandler<String> tracedResponseHandler = new TracedResponseHandler<>(responseHandler);
    HttpResponse httpResponse = new BasicHttpResponse(new BasicStatusLine(HttpVersion.HTTP_1_1, code, ""));

    Segment segment = AWSXRay.beginSegment("test");
    AWSXRay.beginSubsegment("someHttpCall");

    try {
        tracedResponseHandler.handleResponse(httpResponse);
    } catch (IOException e) {
        throw new RuntimeException(e);
    }

    AWSXRay.endSubsegment();
    AWSXRay.endSegment();
    return segment;
}

项目：teamcity-msteams-notifier 文件：MsTeamsNotificationTest.java

@Test
public void post_whenResponseIsFailure_logsException() throws IOException {
    ArgumentCaptor<HttpPost> requestCaptor = ArgumentCaptor.forClass(HttpPost.class);
    HttpClient httpClient = mock(HttpClient.class);
    BasicHttpResponse response = new BasicHttpResponse(new BasicStatusLine(new ProtocolVersion("http", 1, 1), 400, ""));
    response.setEntity(new StringEntity("failure reason here"));

    when(httpClient.execute(requestCaptor.capture())).thenReturn(response);

    MsTeamsNotification w = factory.getMsTeamsNotification(httpClient);

    MsTeamsNotificationPayloadContent content = new MsTeamsNotificationPayloadContent();
    content.setBuildDescriptionWithLinkSyntax("http://foo");
    content.setCommits(new ArrayList<Commit>());

    w.setPayload(content);
    w.setEnabled(true);
    w.post();

    assertNotNull(w.getResponse());
    assertFalse(w.getResponse().getOk());
}

项目：java-logging 文件：OutboundRequestLoggingInterceptorTest.java

@Test
public void logsRequestAndResponseFields() {
    HttpContext context = new BasicHttpContext();
    context.setAttribute(HTTP_TARGET_HOST, "http://www.google.com");

    OutboundRequestLoggingInterceptor interceptor = new OutboundRequestLoggingInterceptor(new FakeClock(20));

    interceptor.process(new BasicHttpRequest("GET", "/something"), context);
    interceptor.process(new BasicHttpResponse(new BasicStatusLine(ANY_PROTOCOL, 200, "any")), context);

    Map<String, Object> fields = new ConcurrentHashMap<>();
    fields.put("requestMethod", "GET");
    fields.put("requestURI", "http://www.google.com/something");
    testAppender.assertEvent(0, INFO, "Outbound request start", appendEntries(fields));

    fields.put("responseTime", 20L);
    fields.put("responseCode", 200);
    testAppender.assertEvent(1, INFO, "Outbound request finish", appendEntries(fields));
}

项目：java-logging 文件：OutboundRequestLoggingInterceptorTest.java

@Test
public void allowEmptyConstructorToBuildDefaultClock() {
    testAppender.clearEvents();

    HttpContext context = new BasicHttpContext();
    context.setAttribute(HTTP_TARGET_HOST, "http://www.google.com");

    OutboundRequestLoggingInterceptor interceptor = new OutboundRequestLoggingInterceptor();

    interceptor.process(new BasicHttpRequest("GET", "/something"), context);
    interceptor.process(new BasicHttpResponse(new BasicStatusLine(ANY_PROTOCOL, 200, "any")), context);

    assertThat(testAppender.getEvents()).extracting("message")
        .contains("Outbound request start", Index.atIndex(0))
        .contains("Outbound request finish", Index.atIndex(1));
}

项目：bootstrap 文件：AbstractRestTest.java

/**
 * Wait for the server is ready.
 */
private void waitForServerReady() throws IOException, InterruptedException {
    final HttpGet httpget = new HttpGet(getPingUri());
    HttpResponse response = new BasicHttpResponse(new BasicStatusLine(HttpVersion.HTTP_1_1, HttpStatus.SC_NOT_FOUND, ""));
    int counter = 0;
    while (true) {
        try {
            response = httpclient.execute(httpget);
            final int status = response.getStatusLine().getStatusCode();
            if (status == HttpStatus.SC_OK) {
                break;
            }
            checkRetries(counter);
        } catch (final HttpHostConnectException ex) { // NOSONAR - Wait, and check later
            log.info("Check failed, retrying...");
            checkRetries(counter);
        } finally {
            EntityUtils.consume(response.getEntity());
        }
        counter++;
    }
}

项目：finances 文件：FileDownloadTest.java

@Test
public void downloadNewStatementsClosesEntityStream() throws Exception {
    final Config filesStep = ConfigFactory.parseString("{ method = get, path = /files }");
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    final HttpEntity entity = mock(HttpEntity.class);
    final InputStream stream = mock(InputStream.class);
    doReturn(Collections.singletonList(filesStep)).when(config).getConfigList("fileList");
    when(client.execute(any(HttpGet.class))).thenReturn(response);
    when(response.getStatusLine()).thenReturn(new BasicStatusLine(HttpVersion.HTTP_1_1, HttpStatus.SC_OK, null));
    when(response.getEntity()).thenReturn(entity);
    when(entity.isStreaming()).thenReturn(true);
    when(entity.getContent()).thenReturn(stream);

    download.downloadNewStatements();

    verify(response).getEntity();
    verify(entity).getContent();
    verify(stream).close();
}

项目：finances 文件：FileDownloadTest.java

@Test
public void downloadNewStatementsCapturesResult() throws Exception {
    final Config filesStep = ConfigFactory.parseString("{ method = get, path = /files, " +
            "result = { format = json, extract = [{ name = var, selector = x }] } }");
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    final HttpEntity entity = mock(HttpEntity.class);
    final InputStream stream = new ByteArrayInputStream("{\"x\":\"value\"}".getBytes("UTF-8"));
    doReturn(Collections.singletonList(filesStep)).when(config).getConfigList("fileList");
    when(client.execute(any(HttpGet.class))).thenReturn(response);
    when(response.getStatusLine()).thenReturn(new BasicStatusLine(HttpVersion.HTTP_1_1, HttpStatus.SC_OK, null));
    when(response.getEntity()).thenReturn(entity);
    when(entity.getContent()).thenReturn(stream);

    download.downloadNewStatements();

    verify(response).getEntity();
    verify(context).putValue("var", "value");
}

项目：finances 文件：FileDownloadTest.java

@Test
public void downloadNewStatementsCapturesFileList() throws Exception {
    final Config filesStep = ConfigFactory.parseString("{ method = get, path = /files, " +
            "result = { format = html, files.selector = { path = li, fields = [\"text()\"] } } }");
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    final HttpEntity entity = mock(HttpEntity.class);
    final InputStream stream = new ByteArrayInputStream("<html><body><li>file url</li></body></html>".getBytes("UTF-8"));
    doReturn(Collections.singletonList(filesStep)).when(config).getConfigList("fileList");
    when(context.getBaseUrl()).thenReturn("http://example.com");
    when(client.execute(any(HttpGet.class))).thenReturn(response);
    when(response.getStatusLine()).thenReturn(new BasicStatusLine(HttpVersion.HTTP_1_1, HttpStatus.SC_OK, null));
    when(response.getEntity()).thenReturn(entity);
    when(entity.getContent()).thenReturn(stream);

    download.downloadNewStatements();

    verify(response).getEntity();
    verify(context).addFile(Collections.singletonList("file url"));
}

项目：finances 文件：FileDownloadTest.java

@Test
public void downloadNewStatementsSavesNewFile() throws Exception {
    final Config downloadStep = ConfigFactory.parseString("{ method = get, path = /files, save = true }");
    final CloseableHttpResponse response = mock(CloseableHttpResponse.class);
    final HttpEntity entity = mock(HttpEntity.class);
    final InputStream stream = new ByteArrayInputStream("statement content".getBytes("UTF-8"));
    final List<Object> fileKeys = Collections.singletonList("file key");
    final File statement = new File("./test.txt");
    if (statement.exists()) statement.delete();
    doReturn(Collections.emptyList()).when(config).getConfigList("fileList");
    doReturn(Collections.singletonList(downloadStep)).when(config).getConfigList("download");
    when(context.getFileList()).thenReturn(Collections.singletonList(fileKeys));
    when(context.getFile(fileKeys)).thenReturn(statement);
    when(client.execute(any(HttpGet.class))).thenReturn(response);
    when(response.getStatusLine()).thenReturn(new BasicStatusLine(HttpVersion.HTTP_1_1, HttpStatus.SC_OK, null));
    when(response.getEntity()).thenReturn(entity);
    when(entity.getContent()).thenReturn(stream);

    download.downloadNewStatements();

    verify(response).getEntity();
    verify(context).addStatement(statement);
}