1 files changed, 14 insertions, 16 deletions
diff --git a/spec/rubocop/cop/avoid_becomes_spec.rb b/spec/rubocop/cop/avoid_becomes_spec.rb
index 07cf374faf5..401c694f373 100644
--- a/spec/rubocop/cop/avoid_becomes_spec.rb
+++ b/spec/rubocop/cop/avoid_becomes_spec.rb
@@ -2,33 +2,31 @@
 
 require 'fast_spec_helper'
 require 'rubocop'
-require 'rubocop/rspec/support'
 require_relative '../../../rubocop/cop/avoid_becomes'
 
 RSpec.describe RuboCop::Cop::AvoidBecomes do
-  include CopHelper
-
   subject(:cop) { described_class.new }
 
   it 'flags the use of becomes with a constant parameter' do
-    inspect_source('foo.becomes(Project)')
-
-    expect(cop.offenses.size).to eq(1)
+    expect_offense(<<~CODE)
+      foo.becomes(Project)
+      ^^^^^^^^^^^^^^^^^^^^ Avoid the use of becomes(SomeConstant), [...]
+    CODE
   end
 
   it 'flags the use of becomes with a namespaced constant parameter' do
-    inspect_source('foo.becomes(Namespace::Group)')
-
-    expect(cop.offenses.size).to eq(1)
+    expect_offense(<<~CODE)
+      foo.becomes(Namespace::Group)
+      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Avoid the use of becomes(SomeConstant), [...]
+    CODE
   end
 
   it 'flags the use of becomes with a dynamic parameter' do
-    inspect_source(<<~RUBY)
-    model = Namespace
-    project = Project.first
-    project.becomes(model)
-    RUBY
-
-    expect(cop.offenses.size).to eq(1)
+    expect_offense(<<~CODE)
+      model = Namespace
+      project = Project.first
+      project.becomes(model)
+      ^^^^^^^^^^^^^^^^^^^^^^ Avoid the use of becomes(SomeConstant), [...]
+    CODE
   end
 end