1 files changed, 18 insertions, 17 deletions
diff --git a/spec/rubocop/cop/migration/with_lock_retries_with_change_spec.rb b/spec/rubocop/cop/migration/with_lock_retries_with_change_spec.rb
index f0be14c8ee9..f2e84a8697c 100644
--- a/spec/rubocop/cop/migration/with_lock_retries_with_change_spec.rb
+++ b/spec/rubocop/cop/migration/with_lock_retries_with_change_spec.rb
@@ -1,40 +1,41 @@
 # frozen_string_literal: true
 
 require 'fast_spec_helper'
-require 'rubocop'
 require_relative '../../../../rubocop/cop/migration/with_lock_retries_with_change'
 
 RSpec.describe RuboCop::Cop::Migration::WithLockRetriesWithChange 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` is used inside a `change` method' do
-      inspect_source('def change; with_lock_retries {}; 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` cannot be used within `change` [...]
+          with_lock_retries {}
+        end
+      RUBY
     end
 
     it 'registers no offense when `with_lock_retries` is used inside an `up` method' do
-      inspect_source('def up; with_lock_retries {}; end')
-
-      expect(cop.offenses.size).to eq(0)
+      expect_no_offenses(<<~RUBY)
+        def up
+          with_lock_retries {}
+        end
+      RUBY
     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 {}; end')
-
-      expect(cop.offenses.size).to eq(0)
+      expect_no_offenses(<<~RUBY)
+        def change
+          with_lock_retries {}
+        end
+      RUBY
     end
   end
 end