test: add test tools for repair migration e2e tests.

SandPod · Nov 17, 2023 · 438622a · 438622a
1 parent d2b40d7
commit 438622a
Showing 1 changed file with 74 additions and 1 deletion.
diff --git a/tests/serverpod_test_server/lib/test_util/migration_test_utils.dart b/tests/serverpod_test_server/lib/test_util/migration_test_utils.dart
@@ -13,8 +13,11 @@ abstract class MigrationTestUtils {
     String tag = 'test',
   }) async {
     for (var protocols in migrationProtocols) {
+      var exitCode =
+          await createMigrationFromProtocols(protocols: protocols, tag: tag);
+
       assert(
-        await createMigrationFromProtocols(protocols: protocols, tag: tag) == 0,
+        exitCode == 0,
         'Failed to create migration.',
       );
     }
@@ -50,6 +53,7 @@ abstract class MigrationTestUtils {
         '--tag',
         suffixedTag,
         if (force) '--force',
+        '--verbose',
       ],
     );
   }
@@ -113,10 +117,79 @@ abstract class MigrationTestUtils {
         'maintenance',
         '--mode',
         'production',
+        '--logging',
+        'verbose',
+      ],
+    );
+  }
+
+  static Future<int> runApplyRepairMigration() async {
+    return await _runProcess(
+      'dart',
+      arguments: [
+        'run',
+        'bin/main.dart',
+        '--apply-repair-migration',
+        '--role',
+        'maintenance',
+        '--mode',
+        'production',
+        '--logging',
+        'verbose',
       ],
     );
   }
 
+  static Future<int> runApplyBothRepairMigrationAndMigrations() async {
+    return await _runProcess(
+      'dart',
+      arguments: [
+        'run',
+        'bin/main.dart',
+        '--apply-repair-migration',
+        '--apply-migrations',
+        '--role',
+        'maintenance',
+        '--mode',
+        'production',
+        '--logging',
+        'verbose',
+      ],
+    );
+  }
+
+  static Future<int> runCreateRepairMigration({
+    String tag = 'test',
+    bool force = false,
+    String? targetVersion,
+  }) async {
+    return await _runProcess(
+      'serverpod',
+      arguments: [
+        'create-repair-migration',
+        '--tag',
+        tag,
+        if (targetVersion != null) ...['--version', targetVersion],
+        if (force) '--force',
+        '--verbose',
+      ],
+    );
+  }
+
+  static File? tryLoadRepairMigrationFile() {
+    var repairMigrationDirectory = _repairMigrationDirectory();
+    if (!repairMigrationDirectory.existsSync()) {
+      return null;
+    }
+
+    var repairMigrationFiles = repairMigrationDirectory.listSync();
+    if (repairMigrationFiles.isEmpty) {
+      return null;
+    }
+
+    return repairMigrationFiles.first as File;
+  }
+
   static Directory _migrationProtocolTestDirectory() => Directory(path.join(
         Directory.current.path,
         'lib',