1 package org.onap.vid.controller;
3 import com.google.common.collect.ImmutableList;
4 import com.google.common.collect.ImmutableMap;
5 import org.mockito.InjectMocks;
6 import org.mockito.Mock;
7 import org.mockito.Mockito;
8 import org.mockito.MockitoAnnotations;
9 import org.onap.vid.aai.AaiResponseTranslator;
10 import org.onap.vid.aai.AaiResponseTranslator.PortMirroringConfigDataError;
11 import org.onap.vid.aai.AaiResponseTranslator.PortMirroringConfigDataOk;
12 import org.onap.vid.controllers.AaiController;
13 import org.onap.vid.services.AaiService;
14 import org.testng.annotations.BeforeMethod;
15 import org.testng.annotations.Test;
19 import static org.hamcrest.MatcherAssert.assertThat;
20 import static org.hamcrest.core.Is.is;
23 public class AaiControllerTest {
26 AaiController aaiController = new AaiController();
29 AaiService aaiService;
32 public void initMocks(){
33 MockitoAnnotations.initMocks(this);
37 public void getPortMirroringConfigData_givenThreeIds_ReturnsThreeResults() {
39 final PortMirroringConfigDataOk toBeReturnedForA = new PortMirroringConfigDataOk("foobar");
40 final PortMirroringConfigDataError toBeReturnedForB = new PortMirroringConfigDataError("foo", "{ baz: qux }");
41 final PortMirroringConfigDataOk toBeReturnedForC = new PortMirroringConfigDataOk("corge");
44 .doReturn(toBeReturnedForA)
45 .doReturn(toBeReturnedForB)
46 .doReturn(toBeReturnedForC)
47 .when(aaiService).getPortMirroringConfigData(Mockito.anyString());
49 final Map<String, AaiResponseTranslator.PortMirroringConfigData> result = aaiController.getPortMirroringConfigsData(ImmutableList.of("a", "b", "c"));
51 assertThat(result, is(ImmutableMap.of(
52 "a", toBeReturnedForA,
53 "b", toBeReturnedForB,