asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/service/SdcRepoServiceTest.java

   1 package org.openecomp.sdc.asdctool.migration.service;
   2
   3 import org.mockito.InjectMocks;
   4 import org.mockito.Mock;
   5 import org.mockito.MockitoAnnotations;
   6 import org.mockito.internal.verification.Times;
   7 import org.openecomp.sdc.asdctool.migration.core.DBVersion;
   8 import org.openecomp.sdc.be.dao.cassandra.MigrationTasksDao;
   9 import org.openecomp.sdc.be.resources.data.MigrationTaskEntry;
  10 import org.testng.annotations.BeforeMethod;
  11 import org.testng.annotations.Test;
  12
  13 import java.math.BigInteger;
  14
  15 import static org.mockito.Mockito.verify;
  16 import static org.mockito.Mockito.when;
  17 import static org.testng.Assert.assertEquals;
  18
  19 public class SdcRepoServiceTest {
  20
  21     @InjectMocks
  22     private SdcRepoService testInstance;
  23
  24     @Mock
  25     private MigrationTasksDao migrationTasksDaoMock;
  26
  27     @BeforeMethod
  28     public void setUp() throws Exception {
  29         MockitoAnnotations.initMocks(this);
  30     }
  31
  32     @Test
  33     public void testGetLatestVersion_noMinorVersionForCurrentVersion() throws Exception {
  34         when(migrationTasksDaoMock.getLatestMinorVersion(DBVersion.CURRENT_VERSION.getMajor())).thenReturn(null);
  35         DBVersion latestDBVersion = testInstance.getLatestDBVersion();
  36         assertEquals(latestDBVersion.getMajor(), DBVersion.CURRENT_VERSION.getMajor());
  37         assertEquals(latestDBVersion.getMinor(), BigInteger.valueOf(Integer.MIN_VALUE));
  38     }
  39
  40     @Test
  41     public void testGetLatestVersion() throws Exception {
  42         when(migrationTasksDaoMock.getLatestMinorVersion(DBVersion.CURRENT_VERSION.getMajor())).thenReturn(BigInteger.TEN);
  43         DBVersion latestDBVersion = testInstance.getLatestDBVersion();
  44         assertEquals(latestDBVersion.getMajor(), DBVersion.CURRENT_VERSION.getMajor());
  45         assertEquals(latestDBVersion.getMinor(), BigInteger.TEN);
  46     }
  47
  48     @Test
  49     public void testCreateMigrationTask() throws Exception {
  50         MigrationTaskEntry taskEntry =  new MigrationTaskEntry();
  51         testInstance.createMigrationTask(taskEntry);
  52         verify(migrationTasksDaoMock, new Times(1)).createMigrationTask(taskEntry);
  53     }
  54
  55 }