1 package org.onap.vid.more;
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;
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;
26 public class AuditLoggerTest extends BaseApiTest {
28 private final String ECOMP_REQUEST_ID_ECHO = "x-ecomp-requestid-echo";
36 public void resetPreset() {
37 SimulatorApi.registerExpectation("create_new_instance/aai_get_full_subscribers.json", CLEAR_THEN_SET);
41 public void aaiController2$GetVpnList_requestIdIsAuditedInEntryAndInExit() {
42 registerExpectationFromPreset(new PresetAAIGetVpnsByType(), APPEND);
43 String requestId = getRequestId(() -> restTemplate.getForEntity(buildUri("aai_get_vpn_list"), String.class));
45 assertAndRetryIfNeeded(5, () -> assertThat("request id must be found in exactly two rows", getRequestLogLines(requestId),
47 allOf(containsString(requestId), containsString("Entering")),
48 allOf(containsString(requestId), containsString("Exiting"))
52 private List<String> getRequestLogLines(String requestId) {
53 return LoggerFormatTest.getRequestLogLines(requestId, LogName.audit2019, restTemplate, uri);
56 private String getRequestId(Supplier<ResponseEntity<?>> request) {
57 ResponseEntity<?> response = request.get();
59 assertThat(response.getHeaders(), hasKey(equalToIgnoringCase(ECOMP_REQUEST_ID_ECHO)));
60 List<String> requestIds = response.getHeaders().get(ECOMP_REQUEST_ID_ECHO);
62 assertThat(requestIds, hasSize(1));
63 return requestIds.get(0);