asdctool/src/test/java/org/openecomp/sdc/asdctool/impl/VrfObjectFixHandlerTest.java

   1 package org.openecomp.sdc.asdctool.impl;
   2
   3 import fj.data.Either;
   4 import org.junit.Before;
   5 import org.junit.Test;
   6 import org.junit.runner.RunWith;
   7 import org.mockito.Mockito;
   8 import org.mockito.junit.MockitoJUnitRunner;
   9 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
  10 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  11 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
  12
  13 import static org.assertj.core.api.Assertions.assertThat;
  14 import static org.mockito.ArgumentMatchers.anyMap;
  15 import static org.mockito.ArgumentMatchers.eq;
  16 import static org.mockito.Mockito.when;
  17
  18 @RunWith(MockitoJUnitRunner.class)
  19 public class VrfObjectFixHandlerTest {
  20
  21     private TitanDao titanDao;
  22
  23     private VrfObjectFixHandler vrfObjectFixHandler;
  24
  25     @Before
  26     public void init(){
  27         titanDao = Mockito.mock(TitanDao.class);
  28         vrfObjectFixHandler = new VrfObjectFixHandler(titanDao);
  29     }
  30
  31     @Test
  32     public void handleInvalidModeTest(){
  33         assertThat(vrfObjectFixHandler.handle("invalid mode", null)).isFalse();
  34     }
  35
  36     @Test
  37     public void handleDetectNotFoundTest(){
  38         when(titanDao.getByCriteria(eq(VertexTypeEnum.NODE_TYPE), anyMap())).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND));
  39         assertThat(vrfObjectFixHandler.handle("detect", null)).isTrue();
  40     }
  41
  42     @Test
  43     public void handleDetectTitanNotConnectedTest(){
  44         when(titanDao.getByCriteria(eq(VertexTypeEnum.NODE_TYPE), anyMap())).thenReturn(Either.right(TitanOperationStatus.NOT_CONNECTED));
  45         assertThat(vrfObjectFixHandler.handle("detect", null)).isFalse();
  46     }
  47
  48     @Test
  49     public void handleFixNotFoundTest(){
  50         when(titanDao.getByCriteria(eq(VertexTypeEnum.NODE_TYPE), anyMap())).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND));
  51         assertThat(vrfObjectFixHandler.handle("fix", null)).isTrue();
  52     }
  53
  54     @Test
  55     public void handleFixNotCreatedTest(){
  56         when(titanDao.getByCriteria(eq(VertexTypeEnum.NODE_TYPE), anyMap())).thenReturn(Either.right(TitanOperationStatus.NOT_CREATED));
  57         assertThat(vrfObjectFixHandler.handle("fix", null)).isFalse();
  58     }
  59
  60 }