Override revert method to check for version safety

ankane · Oct 16, 2023 · ef01e2c · ef01e2c
1 parent 54a6107
commit ef01e2c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/strong_migrations/migration.rb b/lib/strong_migrations/migration.rb
@@ -20,6 +20,14 @@ def method_missing(method, *args)
     end
     ruby2_keywords(:method_missing) if respond_to?(:ruby2_keywords, true)
 
+    def revert(*migration_classes, &block)
+      if strong_migrations_checker.version_safe?
+        safety_assured { super(*migration_classes, &block) }
+      else
+        super(*migration_classes, &block)
+      end
+    end
+
     def safety_assured
       strong_migrations_checker.class.safety_assured do
         yield