@Test
public void testProcess_WithContext() throws IOException, HttpException {

  HttpContext httpContext = new BasicHttpContext();
  httpContext.setAttribute("TRACE-CONTEXT", testContext);
  responseInterceptor.process(response, httpContext);
  ArgumentCaptor<HttpResponse> responseCaptor = ArgumentCaptor.forClass(HttpResponse.class);
  ArgumentCaptor<TraceContext> contextCaptor = ArgumentCaptor.forClass(TraceContext.class);
  verify(mockDelegate).process(responseCaptor.capture(), contextCaptor.capture());
  HttpResponse response = responseCaptor.getValue();
  TraceContext traceContext = contextCaptor.getValue();
  assertThat(traceContext).isEqualTo(testContext);
  assertThat(response.getHeader("Content-Length")).isEqualTo("123");
  assertThat(response.getStatus()).isEqualTo(200);
}