RSpec/ReturnFromStub Optimize #on_block

Author: andrykonchin

lib/rubocop/cop/rspec/return_from_stub.rb

@@ -40,6 +40,7 @@ class ReturnFromStub < Cop
         MSG_BLOCK = 'Use block for static values.'
 
         def_node_search :contains_stub?, '(send nil? :receive (...))'
+        def_node_matcher :stub_with_block?, '(block #contains_stub? ...)'
         def_node_search :and_return_value, <<-PATTERN
           $(send _ :and_return $(...))
         PATTERN
@@ -53,7 +54,7 @@ def on_send(node)
 
         def on_block(node)
           return unless style == :and_return
-          return unless contains_stub?(node)
+          return unless stub_with_block?(node)
 
           check_block_body(node)
         end