Reset state for lock timeout retries

lib/strong_migrations/checker.rb

@@ -11,10 +11,15 @@ class << self
 
     def initialize(migration)
       @migration = migration
+      reset
+    end
+
+    def reset
       @new_tables = []
       @new_columns = []
       @timeouts_set = false
       @committed = false
+      @transaction_disabled = false
     end
 
     def self.safety_assured

lib/strong_migrations/migrator.rb

@@ -12,7 +12,7 @@ def ddl_transaction(migration, ...)
       # retry migration since the entire transaction needs to be rerun
       checker.retry_lock_timeouts(check_committed: true) do
         # failed transaction reverts timeout, so need to re-apply
-        checker.timeouts_set = false
+        checker.reset
 
         super(migration, ...)
       end

test/timeouts_test.rb

@@ -170,6 +170,22 @@ def test_lock_timeout_retries_add_index
     migrate AddIndexConcurrently, direction: :down
   end
 
+  def test_lock_timeout_retries_add_index_safe_by_default
+    skip unless postgresql?
+
+    StrongMigrations.safe_by_default = true
+
+    assert_raises(ActiveRecord::LockWaitTimeout) do
+      with_lock_timeout_retries do
+        migrate AddIndex
+      end
+    end
+
+    migrate AddIndex, direction: :down
+  ensure
+    StrongMigrations.safe_by_default = false
+  end
+
   def reset_timeouts
     StrongMigrations.lock_timeout = nil
     StrongMigrations.statement_timeout = nil