Rspec绕过内部函数并返回模拟数据

Question

在 Rails 中，我正在使用 Rspec 为控制器方法 search_backups 编写测试：

      def elastic_mongo_lookup(search_term) 
        devices_ids_from_elastic = ConfigTextSearch.search search_term
        puts devices_ids_from_elastic
        device_ids = devices_ids_from_elastic.map { |device| device._source.device_id }
        csv_string = CSV.generate do |csv|
          Device.where(:_id.in => device_ids).each do |device|
            csv << [device.logical_name, device.primary_ip]
          end
        end
        return csv_string
      end

  def search_backups
    authorize! :read, :custom_report
    csv_string = elastic_mongo_lookup params[:search_term]
    if csv_string.blank?
      flash[:notice] = "No results were found"
      redirect_to reports_path
    else 
      render text: "DeviceID, primary_ip\n" + csv_string
    end
  end#search_backups

---

  describe "try controller method" do
    let(:reports_controller) { ReportsController.new }
    before do
      allow(CSV).to receive(:generate).and_return("1234", "blah")
      allow(ConfigTextSearch).to receive(:search).and_return(['"hits": [ {"_source":{"device_id":"54afe167b3000006"}]'])
      allow(:devices_ids_from_elastic).to receive(:map).and_return('54afe167b3000006')
      stub_request(:get, "http://localhost:9200/mongo_index/config_files/_search?q=").
        with(:headers => {'Expect'=>'', 'User-Agent'=>'Faraday v0.9.1'}).
        to_return(:status => 200, :body => '', :headers => {})
    end

    it "allows people to search backups" do
      reports = double(ReportsController)
      post 'search_backups'
    end
  end

问题是ConfigTextSearch.search search_term 返回一个elasticsearch ORM 对象.. 这意味着我不能存根它，因为devices_ids_from_elastic.map 上的.map() 方法是唯一的，因为它嵌套了_source 方法。

我怎样才能完全绕过elastic_mongo_lookup，只返回一个模拟的csv_string 到search_backups？

Answer 1

在 RSpec 控制器测试中，controller 被定义为被测控制器。因此，您可以通过以下方式实现您的要求：

allow(controller).to receive(:elastic_mongo_lookup).and_return('whatever string you choose')