Added test for check_down with safety_assured

ankane · Nov 9, 2024 · 3e7129e · 3e7129e
1 parent f802207
commit 3e7129e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/test/check_down_test.rb b/test/check_down_test.rb
@@ -35,6 +35,14 @@ def test_change_safe
     end
   end
 
+  def test_safety_assured
+    migrate CheckDownSafetyAssured
+    with_check_down do
+      assert_unsafe CheckDownSafetyAssured, direction: :down
+    end
+    assert_safe CheckDownSafetyAssured, direction: :down
+  end
+
   def test_add_column
     migrate AddColumnDefault
     with_check_down do

diff --git a/test/migrations/check_down.rb b/test/migrations/check_down.rb
@@ -37,3 +37,11 @@ def change
     remove_index :users, column: :name, algorithm: :concurrently
   end
 end
+
+class CheckDownSafetyAssured < TestMigration
+  def change
+    safety_assured do
+      add_column :users, :age, :integer
+    end
+  end
+end