1 files changed, 36 insertions, 0 deletions
diff --git a/spec/rubocop/cop/database/disable_referential_integrity_spec.rb b/spec/rubocop/cop/database/disable_referential_integrity_spec.rb
new file mode 100644
index 00000000000..9ac67363cb6
--- /dev/null
+++ b/spec/rubocop/cop/database/disable_referential_integrity_spec.rb
@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'fast_spec_helper'
+require_relative '../../../../rubocop/cop/database/disable_referential_integrity'
+
+RSpec.describe RuboCop::Cop::Database::DisableReferentialIntegrity do
+  subject(:cop) { described_class.new }
+
+  it 'does not flag the use of disable_referential_integrity with a send receiver' do
+    expect_offense(<<~SOURCE)
+      foo.disable_referential_integrity
+      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Do not use `disable_referential_integrity`, [...]
+    SOURCE
+  end
+
+  it 'flags the use of disable_referential_integrity with a full definition' do
+    expect_offense(<<~SOURCE)
+      ActiveRecord::Base.connection.disable_referential_integrity
+      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Do not use `disable_referential_integrity`, [...]
+    SOURCE
+  end
+
+  it 'flags the use of disable_referential_integrity with a nil receiver' do
+    expect_offense(<<~SOURCE)
+      class Foo ; disable_referential_integrity ; end
+                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Do not use `disable_referential_integrity`, [...]
+    SOURCE
+  end
+
+  it 'flags the use of disable_referential_integrity when passing a block' do
+    expect_offense(<<~SOURCE)
+      class Foo ; disable_referential_integrity { :foo } ; end
+                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Do not use `disable_referential_integrity`, [...]
+    SOURCE
+  end
+end