vid-app-common/src/test/java/org/onap/vid/controller/AaiControllerTest.java

   1 package org.onap.vid.controller;
   2
   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.services.AaiService;
  11 import org.testng.annotations.BeforeMethod;
  12 import org.testng.annotations.Test;
  13
  14 import java.util.Map;
  15
  16 import static org.hamcrest.MatcherAssert.assertThat;
  17 import static org.hamcrest.core.Is.is;
  18
  19 public class AaiControllerTest {
  20
  21     @InjectMocks
  22     AaiController aaiController = new AaiController();
  23
  24     @Mock
  25     AaiService aaiService;
  26
  27     @BeforeMethod
  28     public void initMocks(){
  29         MockitoAnnotations.initMocks(this);
  30     }
  31
  32     @Test
  33     public void getPortMirroringConfigData_givenThreeIds_ReturnsThreeResults() {
  34
  35         final AaiResponseTranslator.PortMirroringConfigDataOk toBeReturnedForA = new AaiResponseTranslator.PortMirroringConfigDataOk("foobar");
  36         final AaiResponseTranslator.PortMirroringConfigDataError toBeReturnedForB = new AaiResponseTranslator.PortMirroringConfigDataError("foo", "{ baz: qux }");
  37         final AaiResponseTranslator.PortMirroringConfigDataOk toBeReturnedForC = new AaiResponseTranslator.PortMirroringConfigDataOk("corge");
  38
  39         Mockito
  40                 .doReturn(toBeReturnedForA)
  41                 .doReturn(toBeReturnedForB)
  42                 .doReturn(toBeReturnedForC)
  43                 .when(aaiService).getPortMirroringConfigData(Mockito.anyString());
  44
  45         final Map<String, AaiResponseTranslator.PortMirroringConfigData> result = aaiController.getPortMirroringConfigsData(ImmutableList.of("a", "b", "c"));
  46
  47         assertThat(result, is(ImmutableMap.of(
  48                 "a", toBeReturnedForA,
  49                 "b", toBeReturnedForB,
  50                 "c", toBeReturnedForC
  51         )));
  52     }
  53
  54
  55
  56 }