re base code
[sdc.git] / asdctool / src / test / java / org / openecomp / sdc / asdctool / migration / task / MigrationTasksTest.java
index 52dc7c9..8d48a5e 100644 (file)
@@ -1,11 +1,5 @@
 package org.openecomp.sdc.asdctool.migration.task;
 
-import java.util.Collection;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
-import java.util.stream.Collectors;
-
 import org.apache.commons.lang.StringUtils;
 import org.openecomp.sdc.asdctool.migration.core.DBVersion;
 import org.openecomp.sdc.asdctool.migration.core.task.Migration;
@@ -14,6 +8,12 @@ import org.testng.Assert;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.Test;
 
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
+
 
 public class MigrationTasksTest  {
 
@@ -21,13 +21,13 @@ public class MigrationTasksTest  {
     private List<Migration> migrations;
 
     @BeforeMethod
-    public void setUp() {
+    public void setUp() throws Exception {
         ClassScanner classScanner = new ClassScanner();
         migrations = classScanner.getAllClassesOfType(MIGRATIONS_BASE_PACKAGE, Migration.class);
     }
 
     @Test
-    public void testNoTasksWithSameVersion() {
+    public void testNoTasksWithSameVersion() throws Exception {
         Map<DBVersion, List<Migration>> migrationsByVersion = migrations.stream().collect(Collectors.groupingBy(Migration::getVersion));
         migrationsByVersion.forEach((version, migrations) -> {
             if (migrations.size() > 1) {
@@ -38,14 +38,14 @@ public class MigrationTasksTest  {
     }
 
     @Test
-    public void testNoTaskWithVersionGreaterThanCurrentVersion() {
+    public void testNoTaskWithVersionGreaterThanCurrentVersion() throws Exception {
         Set<Migration> migrationsWithVersionsGreaterThanCurrent = migrations.stream().filter(mig -> mig.getVersion().compareTo(DBVersion.CURRENT_VERSION) > 0)
                 .collect(Collectors.toSet());
 
         if (!migrationsWithVersionsGreaterThanCurrent.isEmpty()) {
             Assert.fail(String.format("migrations tasks %s have version which is greater than DBVersion.CURRENT_VERSION %s. did you forget to update current version?",
-                               getMigrationsNameAsString(migrationsWithVersionsGreaterThanCurrent),
-                               DBVersion.CURRENT_VERSION.toString()));
+                    getMigrationsNameAsString(migrationsWithVersionsGreaterThanCurrent),
+                    DBVersion.CURRENT_VERSION.toString()));
         }
     }