WIP: start sketching out flway migrate on apply

manheim · Apr 12, 2021 · a9891c6 · a9891c6
1 parent 43f3365
commit a9891c6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/FlywayMigrationPlugin.groovy b/src/FlywayMigrationPlugin.groovy
@@ -8,6 +8,7 @@ class FlywayMigrationPlugin implements TerraformEnvironmentStagePlugin, Resettab
 
     public void apply(TerraformEnvironmentStage stage) {
         stage.decorate(TerraformEnvironmentStage.PLAN, flywayInfoClosure())
+        stage.decorate(TerraformEnvironmentStage.APPLY, flywayMigrateClosure())
     }
 
     public Closure flywayInfoClosure() {
@@ -22,6 +23,14 @@ class FlywayMigrationPlugin implements TerraformEnvironmentStagePlugin, Resettab
         }
     }
 
+    public Closure flywayMigrateClosure() {
+        return { innerClosure ->
+            innerClosure()
+
+            sh "echo run flyway migrate"
+        }
+    }
+
     public Collection buildEnvironmentVariableList(env) {
         def list = []
         if (passwordVariable) {

diff --git a/test/FlywayMigrationPluginTest.groovy b/test/FlywayMigrationPluginTest.groovy
@@ -39,6 +39,18 @@ class FlywayMigrationPluginTest {
 
             verify(stage).decorate(TerraformEnvironmentStage.PLAN, infoClosure)
         }
+
+        @Test
+        void addsFlywayMigrateClosureOnApply() {
+            def migrateClosure = { -> }
+            def plugin = spy(new FlywayMigrationPlugin())
+            doReturn(migrateClosure).when(plugin).flywayMigrateClosure()
+            def stage = mock(TerraformEnvironmentStage.class)
+
+            plugin.apply(stage)
+
+            verify(stage).decorate(TerraformEnvironmentStage.APPLY, migrateClosure)
+        }
     }
 
     @Nested