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