TaskServiceTest.java

package com.sungardas.enhancedsnapshots.service.impl;

import com.amazonaws.services.ec2.model.VolumeType;
import com.sungardas.enhancedsnapshots.aws.dynamodb.model.BackupEntry;
import com.sungardas.enhancedsnapshots.aws.dynamodb.model.SnapshotEntry;
import com.sungardas.enhancedsnapshots.aws.dynamodb.model.TaskEntry;
import com.sungardas.enhancedsnapshots.aws.dynamodb.repository.BackupRepository;
import com.sungardas.enhancedsnapshots.aws.dynamodb.repository.SnapshotRepository;
import com.sungardas.enhancedsnapshots.aws.dynamodb.repository.TaskRepository;
import com.sungardas.enhancedsnapshots.components.ConfigurationMediator;
import com.sungardas.enhancedsnapshots.dto.TaskDto;
import com.sungardas.enhancedsnapshots.service.NotificationService;
import com.sungardas.enhancedsnapshots.service.SchedulerService;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import java.util.ArrayList;
import java.util.Arrays;

import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.*;

//TODO
@Ignore
@RunWith(MockitoJUnitRunner.class)
public class TaskServiceTest {

    @InjectMocks
    private TaskServiceImpl taskService;
    private TaskDto taskDto;
    private int iopsPerGb = 30;

    @Mock
    private TaskRepository taskRepository;
    @Mock
    private BackupRepository backupRepository;
    @Mock
    private SnapshotRepository snapshotRepository;
    @Mock
    private ConfigurationMediator configuration;
    @Mock
    private SchedulerService schedulerService;
    @Mock
    private NotificationService notificationService;


    @Before
    public void setUp(){

        taskDto = new TaskDto();
        taskDto.setVolumes(Arrays.asList("volId-1"));

        when(configuration.getTempVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getTempVolumeType()).thenReturn(VolumeType.Gp2.toString());
        when(configuration.getRestoreVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getRestoreVolumeType()).thenReturn(VolumeType.Gp2.toString());
        when(configuration.getMaxQueueSize()).thenReturn(5);

        when(backupRepository.findByVolumeId(anyObject())).thenReturn(Arrays.asList(new BackupEntry()));
//        when(backupRepository.findOne(anyObject())).thenReturn(new BackupEntry());
        when(snapshotRepository.findOne((anyString()))).thenReturn(new SnapshotEntry());
    }

    @Test
    public void shouldSetIO1TempVolumeTypeForBackupTask(){
        taskDto.setType("backup");

        when(configuration.getTempVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getTempVolumeType()).thenReturn(VolumeType.Io1.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        // should set temp volume info
        Assert.assertTrue(taskEntry.getTempVolumeIopsPerGb() == iopsPerGb);
        Assert.assertTrue(taskEntry.getTempVolumeType().equals(VolumeType.Io1.toString()));
    }

    @Test
    public void shouldSetGP2TempVolumeTypeForBackupTask(){
        taskDto.setType("backup");

        when(configuration.getTempVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getTempVolumeType()).thenReturn(VolumeType.Gp2.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        // should set temp volume info
        Assert.assertTrue(taskEntry.getTempVolumeIopsPerGb() == 0);
        Assert.assertTrue(taskEntry.getTempVolumeType().equals(VolumeType.Gp2.toString()));
    }

    @Test
    public void shouldSetGP2TempVolumeTypeForRestoreTask(){
        taskDto.setType("restore");

        when(configuration.getTempVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getTempVolumeType()).thenReturn(VolumeType.Gp2.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        // should set temp volume info
        Assert.assertTrue(taskEntry.getTempVolumeIopsPerGb() == 0);
        Assert.assertTrue(taskEntry.getTempVolumeType().equals(VolumeType.Gp2.toString()));
    }


    @Test
    public void shouldSetOP1TempVolumeTypeForRestoreTask(){
        taskDto.setType("restore");

        when(configuration.getTempVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getTempVolumeType()).thenReturn(VolumeType.Io1.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        // should set temp volume info
        Assert.assertTrue(taskEntry.getTempVolumeIopsPerGb() == iopsPerGb);
        Assert.assertTrue(taskEntry.getTempVolumeType().equals(VolumeType.Io1.toString()));

    }

    @Test
    public void shouldSetOP1RestoreVolumeTypeForRestoreTask(){
        taskDto.setType("restore");

        when(configuration.getRestoreVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getRestoreVolumeType()).thenReturn(VolumeType.Io1.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        //should set restore volume info
        Assert.assertTrue(taskEntry.getRestoreVolumeIopsPerGb() == iopsPerGb);
        Assert.assertTrue(taskEntry.getRestoreVolumeType() == VolumeType.Io1.toString());
    }

    @Test
    public void shouldSetGP2RestoreVolumeTypeForRestoreTask(){
        taskDto.setType("restore");

        when(configuration.getRestoreVolumeIopsPerGb()).thenReturn(iopsPerGb);
        when(configuration.getRestoreVolumeType()).thenReturn(VolumeType.Gp2.toString());

        taskService.createTask(taskDto);

        ArgumentCaptor<ArrayList> validTasks = ArgumentCaptor.forClass(ArrayList.class);
        verify(taskRepository).save(validTasks.capture());
        TaskEntry taskEntry = (TaskEntry) validTasks.getAllValues().get(0).get(0);

        //should set restore volume info
        Assert.assertTrue(taskEntry.getRestoreVolumeIopsPerGb() == 0);
        Assert.assertTrue(taskEntry.getRestoreVolumeType() == VolumeType.Gp2.toString());
    }
}