fix(test): flaky test_truncate_tables_with_query_cache

This fixes a bug where our `TestRetryHelper` logic combined
with `reset_fixtures` trying to reset a table without foreign
keys from another table.
It would first crash, removing the foreign key constraint (due
to how we handle `disable_referential_integrity`). And then pass,
since the foreign key constraint is gone. But we need that
constraint in later tests.

Author: BuonOmo

test/excludes/ActiveRecord/AdapterTestWithoutTransaction.rb

@@ -1 +1,18 @@
 exclude :test_reset_empty_table_with_custom_pk, "The test fails because serial primary keys in CockroachDB are created with unique_rowid() where PostgreSQL will create them with a sequence. See https://www.cockroachlabs.com/docs/v19.2/serial.html#modes-of-operation"
+
+require "support/copy_cat"
+
+# This fixes a bug where our `TestRetryHelper` logic combined
+# with `reset_fixtures` trying to reset a table without foreign
+# keys from another table.
+# It would first crash, removing the foreign key constraint (due
+# to how we handle `disable_referential_integrity`). And then pass,
+# since the foreign key constraint is gone. But we need that
+# constraint in later tests.
+CopyCat.copy_methods(self, self, :reset_fixtures) do
+  def on_block(node)
+    return unless node in [:block, [:send, [:lvar, :fixture_names], :each], *]
+
+    replace(node.loc.expression, "ActiveRecord::FixtureSet.create_fixtures(FIXTURES_ROOT, fixture_names)")
+  end
+end

test/excludes/PostgreSQLReferentialIntegrityTest.rb

@@ -1,5 +1,5 @@
 exclude :test_should_reraise_invalid_foreign_key_exception_and_show_warning,
-    "CockroachDB has a different limitation as there is no" \
+    "CockroachDB has a different limitation as there is no " \
     "'DISABLE TRIGGER' statement."
 
 break_tx = "CockroachDB will always alter transactions when " \

test/support/copy_cat.rb

@@ -25,6 +25,9 @@ def warn(message, category: nil, **kwargs)
   # Once debug is set, you can check the closest node you want to edit
   # and then create a method `on_<node_type>` to handle it.
   def copy_methods(new_klass, old_klass, *methods, debug: false, &block)
+    if debug and not block_given?
+      puts "You need to provide a block to debug."
+    end
     methods.each do |met|
       file, _ = old_klass.instance_method(met).source_location
       ast = find_method(Prism::Translation::Parser.parse_file(file), met)