コード例 #1
0
 @Before
 public void setup() throws Exception {
   mockClientTracer = mock(ClientTracer.class);
   mockExecutionContext = mock(ClientExecutionContext.class);
   mockClientRequest = buildClientRequest();
   when(mockExecutionContext.getRequest()).thenReturn(mockClientRequest);
   mockClientResponse = buildClientResponse();
   when(mockExecutionContext.proceed()).thenReturn(mockClientResponse);
   interceptor = new BraveClientExecutionInterceptor(mockClientTracer);
 }
コード例 #2
0
  @Override
  public ClientResponse execute(ClientExecutionContext context) throws Exception {
    if (context.getRequest().getBody() != null) {
      return context.proceed(); // let the MBW handle this
    }

    MultivaluedMap<String, Object> headers = context.getRequest().getHeadersAsObjects();
    List<DKIMSignature> list = getHeaders(headers);

    for (DKIMSignature dosetaSignature : list) {
      KeyRepository repository =
          (KeyRepository) context.getRequest().getAttributes().get(KeyRepository.class.getName());
      sign(repository, headers, null, dosetaSignature);
    }

    return context.proceed();
  }
コード例 #3
0
 protected ClientResponse handleExpired(
     ClientExecutionContext ctx, ClientRequest request, BrowserCache.Entry entry)
     throws Exception {
   ClientResponse response = ctx.proceed();
   if (response.getStatus() == Response.Status.NOT_MODIFIED.getStatusCode()) {
     return updateOnNotModified(request, entry, (BaseClientResponse) response);
   }
   return cache(request, response);
 }
コード例 #4
0
  public ClientResponse execute(ClientExecutionContext ctx) throws Exception {
    ClientRequest request = ctx.getRequest();
    if (!request.getHttpMethod().equals("GET")) {
      return ctx.proceed();
    }

    BrowserCache.Entry entry = getEntry(request);
    if (entry == null) {
      return cache(request, ctx.proceed());
    }

    if (entry.expired()) {
      cache.remove(request.getUri(), entry.getMediaType());
      BrowserCache.Header[] headers = entry.getValidationHeaders();
      for (BrowserCache.Header header : headers) {
        request.header(header.getName(), header.getValue());
      }
      return handleExpired(ctx, request, entry);
    }

    return createClientResponse(request, entry);
  }
コード例 #5
0
  @Test
  public void testExecuteProceedThrowsException() throws Exception {

    final SpanId mockSpanId = mock(SpanId.class);
    when(mockSpanId.getTraceId()).thenReturn(TRACE_ID);
    when(mockSpanId.getSpanId()).thenReturn(SPAN_ID);
    when(mockSpanId.getParentSpanId()).thenReturn(PARENT_SPAN_ID);

    when(mockClientTracer.startNewSpan(PATH)).thenReturn(mockSpanId);
    final IllegalStateException exception = new IllegalStateException("Test exception");
    when(mockExecutionContext.proceed()).thenThrow(exception);

    try {
      interceptor.execute(mockExecutionContext);
      fail("Expected exception.");
    } catch (final IllegalStateException e) {
      assertSame(exception, e);
    }

    final InOrder inOrder = inOrder(mockClientTracer, mockClientRequest, mockExecutionContext);

    inOrder.verify(mockClientTracer).startNewSpan(PATH);
    inOrder.verify(mockClientRequest).header(BraveHttpHeaders.Sampled.getName(), "true");
    inOrder.verify(mockClientRequest).header(BraveHttpHeaders.TraceId.getName(), TRACE_ID_HEX);
    inOrder.verify(mockClientRequest).header(BraveHttpHeaders.SpanId.getName(), SPAN_ID_HEX);
    inOrder
        .verify(mockClientRequest)
        .header(BraveHttpHeaders.ParentSpanId.getName(), Long.toHexString(PARENT_SPAN_ID));
    inOrder.verify(mockClientTracer).setCurrentClientServiceName(CONTEXT_PATH);
    inOrder
        .verify(mockClientTracer)
        .submitBinaryAnnotation(REQUEST_ANNOTATION, HTTP_METHOD + " " + URI);
    inOrder.verify(mockClientTracer).setClientSent();
    inOrder.verify(mockExecutionContext).proceed();
    inOrder.verify(mockClientTracer).submitBinaryAnnotation(HTTP_RESPONSE_CODE_ANNOTATION, 0);
    inOrder.verify(mockClientTracer).submitAnnotation("failure");
    inOrder.verify(mockClientTracer).setClientReceived();
    verifyNoMoreInteractions(mockClientTracer);
  }