catalog-be/src/test/java/org/openecomp/sdc/be/components/scheduledtasks/RecoveryThreadManagerTest.java

   1 package org.openecomp.sdc.be.components.scheduledtasks;
   2
   3 import fj.data.Either;
   4 import org.apache.commons.lang.math.NumberUtils;
   5 import org.junit.Before;
   6 import org.junit.Test;
   7 import org.mockito.InjectMocks;
   8 import org.mockito.Mockito;
   9 import org.mockito.MockitoAnnotations;
  10 import org.openecomp.sdc.be.components.BeConfDependentTest;
  11 import org.openecomp.sdc.be.components.distribution.engine.EnvironmentsEngine;
  12 import org.openecomp.sdc.be.components.scheduledtasks.RecoveryThreadManager.FixEnvironmentTask;
  13 import org.openecomp.sdc.be.dao.cassandra.CassandraOperationStatus;
  14 import org.openecomp.sdc.be.dao.cassandra.OperationalEnvironmentDao;
  15 import org.openecomp.sdc.be.datatypes.enums.EnvironmentStatusEnum;
  16 import org.openecomp.sdc.be.resources.data.OperationalEnvironmentEntry;
  17 import org.openecomp.sdc.common.datastructure.Wrapper;
  18
  19 import java.util.Arrays;
  20 import java.util.Date;
  21 import java.util.List;
  22 import java.util.concurrent.ScheduledExecutorService;
  23 import java.util.concurrent.TimeUnit;
  24
  25 import static org.mockito.Mockito.*;
  26
  27 public class RecoveryThreadManagerTest extends BeConfDependentTest {
  28
  29     @InjectMocks
  30     RecoveryThreadManager recoveryThreadManager = new RecoveryThreadManager();
  31
  32     private OperationalEnvironmentDao operationalEnvironmentDao = mock(OperationalEnvironmentDao.class);
  33     private EnvironmentsEngine environmentsEngine = mock(EnvironmentsEngine.class);
  34     private ScheduledExecutorService scheduledService = mock(ScheduledExecutorService.class);
  35
  36     @Before
  37     public void initMocks() {
  38         MockitoAnnotations.initMocks(this);
  39         reset(operationalEnvironmentDao, environmentsEngine, scheduledService);
  40         recoveryThreadManager.init();
  41     }
  42
  43     @Test
  44     public void testInit() {
  45         verify(scheduledService, Mockito.times(1)).scheduleAtFixedRate(recoveryThreadManager.task,
  46                 NumberUtils.LONG_ZERO, 180L, TimeUnit.SECONDS);
  47
  48     }
  49     @SuppressWarnings("unchecked")
  50     @Test
  51     public void testTaskNoRecords() {
  52         FixEnvironmentTask fixEnvironmentTask = recoveryThreadManager.task;
  53         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> emptyList = Either.left(Arrays.asList());
  54         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.FAILED)).thenReturn(emptyList);
  55         fixEnvironmentTask.run();
  56         verify(environmentsEngine, Mockito.times(0)).buildOpEnv(Mockito.any(Wrapper.class), Mockito.any(OperationalEnvironmentEntry.class));
  57
  58     }
  59
  60     @SuppressWarnings("unchecked")
  61     @Test
  62     public void testTaskCassandraCrash() {
  63         FixEnvironmentTask fixEnvironmentTask = recoveryThreadManager.task;
  64         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> eitherResult = Either.right(CassandraOperationStatus.GENERAL_ERROR);
  65         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.FAILED)).thenReturn(eitherResult);
  66         fixEnvironmentTask.run();
  67         verify(environmentsEngine, Mockito.times(0)).buildOpEnv(Mockito.any(Wrapper.class), Mockito.any(OperationalEnvironmentEntry.class));
  68
  69     }
  70
  71     @SuppressWarnings("unchecked")
  72     @Test
  73     public void testTaskMultipleFailedRecords() {
  74         FixEnvironmentTask fixEnvironmentTask = recoveryThreadManager.task;
  75         OperationalEnvironmentEntry mockEntry = Mockito.mock(OperationalEnvironmentEntry.class);
  76         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> nonEmptyList = Either.left(Arrays.asList(mockEntry, mockEntry, mockEntry));
  77         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> emptyList = Either.left(Arrays.asList());
  78         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.FAILED)).thenReturn(nonEmptyList);
  79         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.IN_PROGRESS)).thenReturn(emptyList);
  80         fixEnvironmentTask.run();
  81         verify(environmentsEngine, Mockito.times(3)).buildOpEnv(Mockito.any(Wrapper.class), Mockito.any(OperationalEnvironmentEntry.class));
  82
  83     }
  84
  85     @SuppressWarnings("unchecked")
  86     @Test
  87     public void testTaskStaleRecords() {
  88         FixEnvironmentTask fixEnvironmentTask = recoveryThreadManager.task;
  89         OperationalEnvironmentEntry mockFailedEntryFirst = Mockito.mock(OperationalEnvironmentEntry.class);
  90         OperationalEnvironmentEntry mockFailedEntrySecond = Mockito.mock(OperationalEnvironmentEntry.class);
  91         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> failedRecordsList = Either.left(Arrays.asList(mockFailedEntryFirst, mockFailedEntrySecond));
  92         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.FAILED)).thenReturn(failedRecordsList);
  93
  94         OperationalEnvironmentEntry mockInProgressNonStaleEntry = Mockito.mock(OperationalEnvironmentEntry.class);
  95         OperationalEnvironmentEntry mockInProgressStaleEntry = Mockito.mock(OperationalEnvironmentEntry.class);
  96         doReturn(new Date(System.currentTimeMillis() - recoveryThreadManager.allowedTimeBeforeStaleSec * 1000 /2)).when(mockInProgressNonStaleEntry).getLastModified();
  97         doReturn(new Date(System.currentTimeMillis() - recoveryThreadManager.allowedTimeBeforeStaleSec * 1000 * 2)).when(mockInProgressStaleEntry).getLastModified();
  98         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> inProgressList = Either.left(Arrays.asList(mockInProgressNonStaleEntry, mockInProgressStaleEntry));
  99         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.IN_PROGRESS)).thenReturn(inProgressList);
 100         fixEnvironmentTask.run();
 101
 102         verify(environmentsEngine).buildOpEnv(Mockito.any(Wrapper.class), Mockito.eq(mockFailedEntryFirst));
 103         verify(environmentsEngine).buildOpEnv(Mockito.any(Wrapper.class), Mockito.eq(mockFailedEntrySecond));
 104         verify(environmentsEngine).buildOpEnv(Mockito.any(Wrapper.class), Mockito.eq(mockInProgressStaleEntry));
 105         verify(environmentsEngine, times(0)).buildOpEnv(Mockito.any(Wrapper.class), Mockito.eq(mockInProgressNonStaleEntry));
 106
 107         verify(environmentsEngine, times(3)).buildOpEnv(Mockito.any(Wrapper.class), Mockito.any(OperationalEnvironmentEntry.class));
 108
 109     }
 110
 111
 112
 113     @Test
 114     public void testUnconnectedRecords() {
 115         FixEnvironmentTask fixEnvironmentTask = recoveryThreadManager.task;
 116         OperationalEnvironmentEntry mockCompleteConnected = Mockito.mock(OperationalEnvironmentEntry.class);
 117         OperationalEnvironmentEntry mockCompleteUnconnected = Mockito.mock(OperationalEnvironmentEntry.class);
 118         Either<List<OperationalEnvironmentEntry>, CassandraOperationStatus> completeRecordsList = Either.left(Arrays.asList(mockCompleteConnected, mockCompleteUnconnected));
 119         when(environmentsEngine.isInMap(mockCompleteConnected)).thenReturn(true);
 120         when(environmentsEngine.isInMap(mockCompleteUnconnected)).thenReturn(false);
 121         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.COMPLETED)).thenReturn(completeRecordsList);
 122         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.FAILED)).thenReturn(Either.left(Arrays.asList()));
 123         when(operationalEnvironmentDao.getByEnvironmentsStatus(EnvironmentStatusEnum.IN_PROGRESS)).thenReturn(Either.left(Arrays.asList()));
 124         fixEnvironmentTask.run();
 125
 126         verify(environmentsEngine).createUebTopicsForEnvironment(mockCompleteUnconnected);
 127         verify(environmentsEngine, times(0)).createUebTopicsForEnvironment(mockCompleteConnected);
 128         verify(environmentsEngine).addToMap(mockCompleteUnconnected);
 129         verify(environmentsEngine, times(0)).addToMap(mockCompleteConnected);
 130         verify(environmentsEngine, times(1)).createUebTopicsForEnvironment(Mockito.any(OperationalEnvironmentEntry.class));
 131
 132     }
 133 }