spec/rubocop/cop/rspec/yield_spec.rb in rubocop-rspec-1.32.0 vs spec/rubocop/cop/rspec/yield_spec.rb in rubocop-rspec-1.33.0

@@ -1,47 +1,69 @@
+# frozen_string_literal: true
+
 RSpec.describe RuboCop::Cop::RSpec::Yield do
   subject(:cop) { described_class.new }
 
   it 'flags `block.call`' do
     expect_offense(<<-RUBY)
       allow(foo).to receive(:bar) { |&block| block.call }
                                   ^^^^^^^^^^^^^^^^^^^^^^^ Use `.and_yield`.
     RUBY
+
+    expect_correction(<<-RUBY)
+      allow(foo).to receive(:bar).and_yield
+    RUBY
   end
 
   it 'flags multiple `block.call`' do
     expect_offense(<<-RUBY)
       allow(foo).to receive(:bar) do |&block|
                                   ^^^^^^^^^^^ Use `.and_yield`.
         block.call
         block.call
       end
     RUBY
+
+    expect_correction(<<-RUBY)
+      allow(foo).to receive(:bar).and_yield.and_yield
+    RUBY
   end
 
   it 'flags `block.call` with arguments' do
     expect_offense(<<-RUBY)
       allow(foo).to receive(:bar) { |&block| block.call(1, 2) }
                                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Use `.and_yield`.
     RUBY
+
+    expect_correction(<<-RUBY)
+      allow(foo).to receive(:bar).and_yield(1, 2)
+    RUBY
   end
 
   it 'flags multiple `block.call` with arguments' do
     expect_offense(<<-RUBY)
       allow(foo).to receive(:bar) do |&block|
                                   ^^^^^^^^^^^ Use `.and_yield`.
          block.call(1)
          block.call(2)
        end
     RUBY
+
+    expect_correction(<<-RUBY)
+      allow(foo).to receive(:bar).and_yield(1).and_yield(2)
+    RUBY
   end
 
   it 'flags `block.call` when `receive` is chained' do
     expect_offense(<<-RUBY)
       allow(foo).to receive(:bar).with(anything) { |&block| block.call }
                                                  ^^^^^^^^^^^^^^^^^^^^^^^ Use `.and_yield`.
     RUBY
+
+    expect_correction(<<-RUBY)
+      allow(foo).to receive(:bar).with(anything).and_yield
+    RUBY
   end
 
   it 'ignores `receive` with no block arguments' do
     expect_no_offenses(<<-RUBY)
       allow(foo).to receive(:bar) { |block| block.call }
@@ -54,27 +76,6 @@
         result = block.call
         transform(result)
       end
     RUBY
   end
-
-  include_examples 'autocorrect',
-                   'allow(foo).to receive(:bar) { |&block| block.call }',
-                   'allow(foo).to receive(:bar).and_yield'
-
-  include_examples 'autocorrect',
-                   'allow(foo).to receive(:bar) { |&block| block.call(1, 2) }',
-                   'allow(foo).to receive(:bar).and_yield(1, 2)'
-
-  bad_code = <<-RUBY
-    allow(foo).to receive(:bar) do |&block|
-      block.call(1)
-      block.call(2)
-    end
-  RUBY
-
-  good_code = <<-RUBY
-    allow(foo).to receive(:bar).and_yield(1).and_yield(2)
-  RUBY
-
-  include_examples 'autocorrect', bad_code, good_code
 end