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