migration: Test migration failures

alvasw · Nov 4, 2024 · b397b76 · b397b76
1 parent da6178d
commit b397b76
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/evolution/src/main/java/bisq/evolution/migration/migrations/MigrationFailedException.java b/evolution/src/main/java/bisq/evolution/migration/migrations/MigrationFailedException.java
@@ -1,6 +1,10 @@
 package bisq.evolution.migration.migrations;
 
 public class MigrationFailedException extends RuntimeException {
+    public MigrationFailedException(String message) {
+        super(message);
+    }
+
     public MigrationFailedException(Throwable cause) {
         super(cause);
     }

diff --git a/evolution/src/test/java/bisq/evolution/migration/MigratorTest.java b/evolution/src/test/java/bisq/evolution/migration/MigratorTest.java
@@ -2,13 +2,16 @@
 
 import bisq.common.application.ApplicationVersion;
 import bisq.common.platform.Version;
+import bisq.evolution.migration.migrations.Migration;
+import bisq.evolution.migration.migrations.MigrationFailedException;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.io.TempDir;
 
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.Collections;
+import java.util.List;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -27,4 +30,30 @@ void migrationSuccess(@TempDir Path dataDir) throws IOException {
         String readVersion = Files.readString(dataDir.resolve("version"));
         assertThat(readVersion).isEqualTo(appVersion.toString());
     }
+
+    @Test
+    void migrationFailure(@TempDir Path dataDir) throws IOException {
+        Path versionFilePath = dataDir.resolve("version");
+        Version dataDirVersion = new Version("2.1.0");
+        Files.writeString(versionFilePath, dataDirVersion.toString());
+
+        Version appVersion = ApplicationVersion.getVersion();
+        var migration = new Migration() {
+            @Override
+            public void run(Path dataDir) {
+                throw new MigrationFailedException("Migration failed.");
+            }
+
+            @Override
+            public Version getVersion() {
+                return new Version("2.1.1");
+            }
+        };
+
+        Migrator migrator = new Migrator(appVersion, dataDir, List.of(migration, migration));
+        migrator.migrate();
+
+        String readVersion = Files.readString(dataDir.resolve("version"));
+        assertThat(readVersion).isEqualTo(dataDirVersion.toString());
+    }
 }