1 files changed, 32 insertions, 36 deletions
diff --git a/spec/rubocop/cop/migration/with_lock_retries_disallowed_method_spec.rb b/spec/rubocop/cop/migration/with_lock_retries_disallowed_method_spec.rb
index 814d87ea24b..298ca273256 100644
--- a/spec/rubocop/cop/migration/with_lock_retries_disallowed_method_spec.rb
+++ b/spec/rubocop/cop/migration/with_lock_retries_disallowed_method_spec.rb
@@ -1,64 +1,58 @@
 # frozen_string_literal: true
 
 require 'fast_spec_helper'
-require 'rubocop'
 require_relative '../../../../rubocop/cop/migration/with_lock_retries_disallowed_method'
 
 RSpec.describe RuboCop::Cop::Migration::WithLockRetriesDisallowedMethod do
-  include CopHelper
-
   subject(:cop) { described_class.new }
 
-  context 'in migration' do
+  context 'when in migration' do
     before do
       allow(cop).to receive(:in_migration?).and_return(true)
     end
 
     it 'registers an offense when `with_lock_retries` block has disallowed method' do
-      inspect_source('def change; with_lock_retries { disallowed_method }; end')
-
-      aggregate_failures do
-        expect(cop.offenses.size).to eq(1)
-        expect(cop.offenses.map(&:line)).to eq([1])
-      end
+      expect_offense(<<~RUBY)
+        def change
+          with_lock_retries { disallowed_method }
+                              ^^^^^^^^^^^^^^^^^ The method is not allowed [...]
+        end
+      RUBY
     end
 
     it 'registers an offense when `with_lock_retries` block has disallowed methods' do
-      source = <<~HEREDOC
-      def change
-        with_lock_retries do
-          disallowed_method
+      expect_offense(<<~RUBY)
+        def change
+          with_lock_retries do
+            disallowed_method
+            ^^^^^^^^^^^^^^^^^ The method is not allowed [...]
 
-          create_table do |t|
-            t.text :text
-          end
+            create_table do |t|
+              t.text :text
+            end
 
-          other_disallowed_method
+            other_disallowed_method
+            ^^^^^^^^^^^^^^^^^^^^^^^ The method is not allowed [...]
 
-          add_column :users, :name
+            add_column :users, :name
+          end
         end
-      end
-      HEREDOC
-
-      inspect_source(source)
-
-      aggregate_failures do
-        expect(cop.offenses.size).to eq(2)
-        expect(cop.offenses.map(&:line)).to eq([3, 9])
-      end
+      RUBY
     end
 
     it 'registers no offense when `with_lock_retries` has only allowed method' do
-      inspect_source('def up; with_lock_retries { add_foreign_key :foo, :bar }; end')
-
-      expect(cop.offenses.size).to eq(0)
+      expect_no_offenses(<<~RUBY)
+        def up
+          with_lock_retries { add_foreign_key :foo, :bar }
+        end
+      RUBY
     end
 
     describe 'for `add_foreign_key`' do
       it 'registers an offense when more than two FKs are added' do
         message = described_class::MSG_ONLY_ONE_FK_ALLOWED
 
-        expect_offense <<~RUBY
+        expect_offense(<<~RUBY)
           with_lock_retries do
             add_foreign_key :imports, :projects, column: :project_id, on_delete: :cascade
             ^^^^^^^^^^^^^^^ #{message}
@@ -71,11 +65,13 @@ RSpec.describe RuboCop::Cop::Migration::WithLockRetriesDisallowedMethod do
     end
   end
 
-  context 'outside of migration' do
+  context 'when outside of migration' do
     it 'registers no offense' do
-      inspect_source('def change; with_lock_retries { disallowed_method }; end')
-
-      expect(cop.offenses.size).to eq(0)
+      expect_no_offenses(<<~RUBY)
+        def change
+          with_lock_retries { disallowed_method }
+        end
+      RUBY
     end
   end
 end