Add/remove foreign keys when truncating during tests

Author: aidanharan

test/cases/coerced_tests.rb

@@ -48,6 +48,26 @@ def test_value_limit_violations_are_translated_to_specific_exception_coerced
         assert_not_nil error.cause
       end
     end
+
+    # SQL Server does not allow truncation of tables that are referenced by foreign key
+    # constraints. So manually remove/add foreign keys in test.
+    coerce_tests! test_truncate_tables
+    def test_truncate_tables_coerced
+      @connection.remove_foreign_key :authors, :author_addresses
+
+      assert_operator Post.count, :>, 0
+      assert_operator Author.count, :>, 0
+      assert_operator AuthorAddress.count, :>, 0
+
+      @connection.truncate_tables("author_addresses", "authors", "posts")
+
+      assert_equal 0, Post.count
+      assert_equal 0, Author.count
+      assert_equal 0, AuthorAddress.count
+    ensure
+      @connection.add_foreign_key :authors, :author_addresses
+      reset_fixtures("posts", "authors", "author_addresses")
+    end
   end
 end
 
@@ -1181,3 +1201,41 @@ class EagerLoadingTooManyIdsTest < ActiveRecord::TestCase
   # Temporarily coerce this test due to https://github.com/rails/rails/issues/34945
   coerce_tests! :test_eager_loading_too_may_ids
 end
+
+
+module ActiveRecord
+  class DatabaseTasksTruncateAllTest < ActiveRecord::TestCase
+
+    # SQL Server does not allow truncation of tables that are referenced by foreign key
+    # constraints. So manually remove/add foreign keys in test.
+    coerce_tests! test_truncate_tables
+    def test_truncate_tables_coerced
+      connection = ActiveRecord::Base.connection
+      connection.remove_foreign_key :authors, :author_addresses
+
+      assert_operator SchemaMigration.count, :>, 0
+      assert_operator InternalMetadata.count, :>, 0
+      assert_operator Author.count, :>, 0
+      assert_operator AuthorAddress.count, :>, 0
+
+      old_configurations = ActiveRecord::Base.configurations
+      configurations = { development: ActiveRecord::Base.configurations["arunit"] }
+      ActiveRecord::Base.configurations = configurations
+
+      ActiveRecord::Tasks::DatabaseTasks.stub(:root, nil) do
+        ActiveRecord::Tasks::DatabaseTasks.truncate_all(
+          ActiveSupport::StringInquirer.new("development")
+        )
+      end
+
+      assert_operator SchemaMigration.count, :>, 0
+      assert_operator InternalMetadata.count, :>, 0
+      assert_equal 0, Author.count
+      assert_equal 0, AuthorAddress.count
+    ensure
+      connection.add_foreign_key :authors, :author_addresses
+      ActiveRecord::Base.configurations = old_configurations
+    end
+
+  end
+end