asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/task/MigrationTasksTest.java

   1 package org.openecomp.sdc.asdctool.migration.task;
   2
   3 import org.apache.commons.lang.StringUtils;
   4 import org.openecomp.sdc.asdctool.migration.core.DBVersion;
   5 import org.openecomp.sdc.asdctool.migration.core.task.Migration;
   6 import org.openecomp.sdc.asdctool.migration.scanner.ClassScanner;
   7 import org.testng.Assert;
   8 import org.testng.annotations.BeforeMethod;
   9 import org.testng.annotations.Test;
  10
  11
  12 import java.util.Collection;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.Set;
  16 import java.util.stream.Collectors;
  17
  18
  19 public class MigrationTasksTest  {
  20
  21     public static final String MIGRATIONS_BASE_PACKAGE = "org.openecomp.sdc.asdctool.migration.tasks";
  22     private List<Migration> migrations;
  23
  24     @BeforeMethod
  25     public void setUp() throws Exception {
  26         ClassScanner classScanner = new ClassScanner();
  27         migrations = classScanner.getAllClassesOfType(MIGRATIONS_BASE_PACKAGE, Migration.class);
  28     }
  29
  30     @Test
  31     public void testNoTasksWithSameVersion() throws Exception {
  32         Map<DBVersion, List<Migration>> migrationsByVersion = migrations.stream().collect(Collectors.groupingBy(Migration::getVersion));
  33         migrationsByVersion.forEach((version, migrations) -> {
  34             if (migrations.size() > 1) {
  35                 System.out.println(String.format("the following migration tasks have the same version %s. versions must be unique", version.toString()));
  36                 Assert.fail(String.format("migration tasks %s has same version %s. migration tasks versions must be unique.", getMigrationsNameAsString(migrations), version.toString()));
  37             }
  38         });
  39     }
  40
  41     @Test
  42     public void testNoTaskWithVersionGreaterThanCurrentVersion() throws Exception {
  43         Set<Migration> migrationsWithVersionsGreaterThanCurrent = migrations.stream().filter(mig -> mig.getVersion().compareTo(DBVersion.CURRENT_VERSION) > 0)
  44                 .collect(Collectors.toSet());
  45
  46         if (!migrationsWithVersionsGreaterThanCurrent.isEmpty()) {
  47             Assert.fail(String.format("migrations tasks %s have version which is greater than DBVersion.CURRENT_VERSION %s. did you forget to update current version?",
  48                                getMigrationsNameAsString(migrationsWithVersionsGreaterThanCurrent),
  49                                DBVersion.CURRENT_VERSION.toString()));
  50         }
  51     }
  52
  53     private String getMigrationsNameAsString(Collection<Migration> migrations) {
  54         return StringUtils.join(migrations.stream().map(mig -> mig.getClass().getName()).collect(Collectors.toList()), ",");
  55     }
  56 }