Skip to content

Commit 8ce1de9

Browse files
Vadzim HushchanskouVadzim Hushchanskou
Vadzim Hushchanskou
authored and
Vadzim Hushchanskou
committed
Update a test
1 parent 50f1544 commit 8ce1de9

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

src/test/java/com/epam/reportportal/restassured/ReportPortalRestAssuredLoggingFilterTest.java

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -641,18 +641,22 @@ public void test_rest_assured_logger_text_as_file_multipart() {
641641
@Test
642642
public void test_rest_assured_header_filter() {
643643
FilterableRequestSpecification requestSpecification = mockBasicRequest(HTML_TYPE);
644-
RestAssuredConfig config = RestAssuredConfig.config().logConfig(LogConfig.logConfig().blacklistHeader("Authorization"));
644+
RestAssuredConfig config = RestAssuredConfig.config().logConfig(
645+
LogConfig.logConfig().blacklistHeader("Authorization").blacklistHeader("Set-Cookie"));
645646
when(requestSpecification.getConfig()).thenReturn(config);
646-
Headers headers = new Headers(new Header(HttpHeaders.AUTHORIZATION, "Bearer some_token"));
647-
when(requestSpecification.getHeaders()).thenReturn(headers);
647+
Headers requestHeaders = new Headers(new Header(HttpHeaders.AUTHORIZATION, "Bearer some_token"));
648+
when(requestSpecification.getHeaders()).thenReturn(requestHeaders);
648649
Response responseObject = mockBasicResponse(HTML_TYPE);
650+
Headers responseHeaders = new Headers(new Header("Set-Cookie", "cookie"));
651+
when(responseObject.getHeaders()).thenReturn(responseHeaders);
649652

650653
List<String> logs = runFilterTextMessageCapture(requestSpecification, responseObject);
651654
assertThat(logs, hasSize(2)); // Request + Response
652655

653-
String headerString = "\n\n**Headers**\n" + HttpHeaders.AUTHORIZATION + ": " + REMOVED_TAG;
656+
String requestHeaderString = "\n\n**Headers**\n" + HttpHeaders.AUTHORIZATION + ": " + REMOVED_TAG;
657+
assertThat(logs.get(0), equalTo(EMPTY_REQUEST + requestHeaderString));
654658

655-
assertThat(logs.get(0), equalTo(EMPTY_REQUEST + headerString));
656-
assertThat(logs.get(1), equalTo(EMPTY_RESPONSE));
659+
String responseHeaderString = "\n\n**Headers**\nSet-Cookie: " + REMOVED_TAG;
660+
assertThat(logs.get(1), equalTo(EMPTY_RESPONSE + responseHeaderString));
657661
}
658662
}

0 commit comments

Comments
 (0)