private void ensureNoWarnings() throws IOException { //Check that there are no unaccounted warning headers. These should be checked with {@link #assertWarnings(String...)} in the //appropriate test try { final List<String> warnings = threadContext.getResponseHeaders().get("Warning"); assertNull("unexpected warning headers", warnings); } finally { resetDeprecationLogger(false); } }
@Override public void handleException(TransportException exp) { try { assertTrue(threadPool.getThreadContext().getResponseHeaders().containsKey("foo.bar")); assertEquals(1, threadPool.getThreadContext().getResponseHeaders().get("foo.bar").size()); assertEquals("baz", threadPool.getThreadContext().getResponseHeaders().get("foo.bar").get(0)); assertNull(threadPool.getThreadContext().getTransient("boom")); } finally { latch.countDown(); } }
@Override public void handleResponse(TransportResponse response) { try { assertSame(response, TransportResponse.Empty.INSTANCE); assertTrue(threadPool.getThreadContext().getResponseHeaders().containsKey("foo.bar")); assertEquals(1, threadPool.getThreadContext().getResponseHeaders().get("foo.bar").size()); assertEquals("baz", threadPool.getThreadContext().getResponseHeaders().get("foo.bar").get(0)); assertNull(threadPool.getThreadContext().getTransient("boom")); } finally { latch.countDown(); } }
protected final void assertWarnings(String... expectedWarnings) { if (enableWarningsCheck() == false) { throw new IllegalStateException("unable to check warning headers if the test is not set to do so"); } try { final List<String> actualWarnings = threadContext.getResponseHeaders().get("Warning"); assertNotNull(actualWarnings); final Set<String> actualWarningValues = actualWarnings.stream().map(DeprecationLogger::extractWarningValueFromWarningHeader).collect(Collectors.toSet()); for (String msg : expectedWarnings) { assertThat(actualWarningValues, hasItem(DeprecationLogger.escapeAndEncode(msg))); } assertEquals("Expected " + expectedWarnings.length + " warnings but found " + actualWarnings.size() + "\nExpected: " + Arrays.asList(expectedWarnings) + "\nActual: " + actualWarnings, expectedWarnings.length, actualWarnings.size()); } finally { resetDeprecationLogger(true); } }
addCustomHeaders(resp, threadContext.getResponseHeaders());
addCustomHeaders(resp, threadContext.getResponseHeaders());
addCustomHeaders(resp, threadContext.getResponseHeaders());
addCustomHeaders(resp, threadContext.getResponseHeaders());
addCustomHeaders(resp, threadContext.getResponseHeaders());