Add some timed grace for assertion on log-lines
[vid.git] / vid-automation / src / test / java / org / onap / vid / more / AuditLoggerTest.java
1 package org.onap.vid.more;
2
3 import static org.hamcrest.CoreMatchers.containsString;
4 import static org.hamcrest.MatcherAssert.assertThat;
5 import static org.hamcrest.Matchers.allOf;
6 import static org.hamcrest.Matchers.contains;
7 import static org.hamcrest.Matchers.equalToIgnoringCase;
8 import static org.hamcrest.Matchers.hasSize;
9 import static org.hamcrest.collection.IsMapContaining.hasKey;
10 import static org.onap.vid.api.TestUtils.assertAndRetryIfNeeded;
11 import static vid.automation.test.services.SimulatorApi.RegistrationStrategy.APPEND;
12 import static vid.automation.test.services.SimulatorApi.RegistrationStrategy.CLEAR_THEN_SET;
13 import static vid.automation.test.services.SimulatorApi.registerExpectationFromPreset;
14
15 import java.util.List;
16 import java.util.function.Supplier;
17 import org.onap.simulator.presetGenerator.presets.aai.PresetAAIGetVpnsByType;
18 import org.onap.vid.api.BaseApiTest;
19 import org.onap.vid.more.LoggerFormatTest.LogName;
20 import org.springframework.http.ResponseEntity;
21 import org.testng.annotations.BeforeClass;
22 import org.testng.annotations.BeforeMethod;
23 import org.testng.annotations.Test;
24 import vid.automation.test.services.SimulatorApi;
25
26 public class AuditLoggerTest extends BaseApiTest {
27
28     private final String ECOMP_REQUEST_ID_ECHO = "x-ecomp-requestid-echo";
29
30     @BeforeClass
31     public void login() {
32         super.login();
33     }
34
35     @BeforeMethod
36     public void resetPreset() {
37         SimulatorApi.registerExpectation("create_new_instance/aai_get_full_subscribers.json", CLEAR_THEN_SET);
38     }
39
40     @Test
41     public void aaiController2$GetVpnList_requestIdIsAuditedInEntryAndInExit() {
42         registerExpectationFromPreset(new PresetAAIGetVpnsByType(), APPEND);
43         String requestId = getRequestId(() -> restTemplate.getForEntity(buildUri("aai_get_vpn_list"), String.class));
44
45         assertAndRetryIfNeeded(5, () -> assertThat("request id must be found in exactly two rows", getRequestLogLines(requestId),
46             contains(
47                 allOf(containsString(requestId), containsString("Entering")),
48                 allOf(containsString(requestId), containsString("Exiting"))
49             )));
50     }
51
52     private List<String> getRequestLogLines(String requestId) {
53         return LoggerFormatTest.getRequestLogLines(requestId, LogName.audit2019, restTemplate, uri);
54     }
55
56     private String getRequestId(Supplier<ResponseEntity<?>> request) {
57         ResponseEntity<?> response = request.get();
58
59         assertThat(response.getHeaders(), hasKey(equalToIgnoringCase(ECOMP_REQUEST_ID_ECHO)));
60         List<String> requestIds = response.getHeaders().get(ECOMP_REQUEST_ID_ECHO);
61
62         assertThat(requestIds, hasSize(1));
63         return requestIds.get(0);
64     }
65
66 }