class RuboCop::Cop::RSpec::ReturnFromStub::BlockBodyCorrector

def call(corrector)
  # Heredoc autocorrection is not yet implemented.
  return if heredoc?
  corrector.replace(
    block,
    "#{block.send_node.source}.and_return(#{body.source})"
  )
end

def heredoc?
  body.loc.is_a?(Parser::Source::Map::Heredoc)
end

def initialize(block)
  @block = block
  @node = block.parent
  @body = block.body || NULL_BLOCK_BODY
end