2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019 Nokia Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
20 package org.onap.vid.controller;
22 import com.google.common.collect.ImmutableMap;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.junit.runner.RunWith;
26 import org.mockito.InjectMocks;
27 import org.mockito.Mock;
28 import org.mockito.junit.MockitoJUnitRunner;
29 import org.onap.vid.model.errorReport.ReportCreationParameters;
30 import org.onap.vid.reports.BasicReportGenerator;
31 import org.onap.vid.reports.DeploymentReportGenerator;
33 import javax.servlet.http.HttpServletRequest;
35 import java.util.Arrays;
36 import java.util.HashMap;
39 import static org.assertj.core.api.Assertions.assertThat;
40 import static org.mockito.Mockito.mock;
41 import static org.mockito.Mockito.when;
43 @RunWith(MockitoJUnitRunner.class)
44 public class ErrorReportControllerTest {
46 private BasicReportGenerator basicReportGenerator;
47 private DeploymentReportGenerator deploymentReportGenerator;
48 private HttpServletRequest httpServletRequest;
50 private ErrorReportController errorReportController;
54 basicReportGenerator = mock(BasicReportGenerator.class);
55 deploymentReportGenerator = mock(DeploymentReportGenerator.class);
56 httpServletRequest = mock(HttpServletRequest.class);
59 = new ErrorReportController(Arrays.asList(basicReportGenerator, deploymentReportGenerator));
63 public void shouldGenerateBasicReportDataWhenNoParameters() {
65 ReportCreationParameters parameters = new ReportCreationParameters();
67 ImmutableMap<String, Object> expectedReport = ImmutableMap.<String, Object>builder()
68 .put("X-ECOMP-RequestID", "request_id")
69 .put("X-ECOMP-RequestID1", "request_id1")
70 .put("X-ECOMP-RequestID2", "request_id2")
71 .put("X-ECOMP-RequestID3", "request_id3")
74 when(basicReportGenerator.apply(httpServletRequest, parameters)).thenReturn(expectedReport);
75 when(basicReportGenerator.canGenerate(parameters)).thenReturn(true);
76 when(deploymentReportGenerator.canGenerate(parameters)).thenReturn(false);
79 Map<String, Object> actualReport = errorReportController.generateReportsData(httpServletRequest, parameters);
82 assertThat(actualReport).isEqualTo(expectedReport);
86 public void shouldGenerateDeploymentReportDataWhenSpecificParameters() {
88 ReportCreationParameters parameters =
89 new ReportCreationParameters("request_id", "service_uuid");
91 ImmutableMap<String, Object> basicReport = ImmutableMap.<String, Object>builder()
92 .put("X-ECOMP-RequestID", "request_id")
93 .put("X-ECOMP-RequestID1", "request_id1")
96 ImmutableMap<String, Object> extendedReport = ImmutableMap.<String, Object>builder()
97 .put("serviceInstanceInfo", "serviceInstanceInfoVal")
98 .put("serviceInstanceInfo1", "serviceInstanceInfoVal1")
101 HashMap<String, Object> expectedReport = new HashMap<>(basicReport);
102 expectedReport.putAll(extendedReport);
104 when(basicReportGenerator.apply(httpServletRequest, parameters)).thenReturn(basicReport);
105 when(deploymentReportGenerator.apply(httpServletRequest, parameters)).thenReturn(extendedReport);
106 when(basicReportGenerator.canGenerate(parameters)).thenReturn(true);
107 when(deploymentReportGenerator.canGenerate(parameters)).thenReturn(true);
110 Map<String, Object> actualReport = errorReportController.generateReportsData(httpServletRequest, parameters);
113 assertThat(actualReport).isEqualTo(expectedReport);