Sustain unset @reflects_state ivar

grosser · Mar 11, 2015 · 2c02a10 · 2c02a10
1 parent 3cc9e21
commit 2c02a10
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/test_after_commit.rb b/lib/test_after_commit.rb
@@ -54,7 +54,12 @@ def test_commit_records
   ActiveRecord::Base.class_eval do
     alias_method :sync_with_transaction_state_with_state, :sync_with_transaction_state
     def sync_with_transaction_state
-      @reflects_state[0] = true
+      if @reflects_state
+        @reflects_state[0] = true
+      else
+        @reflects_state = [true]
+      end
+
       sync_with_transaction_state_with_state
     end
   end