如何测试类是否在RSpec中收到消息?

时间:2016-03-08 16:26:05

标签: ruby-on-rails ruby rspec rspec-rails rspec-mocks

我有一个服务对象:

class SecurityQueryService
  class Contract::NotFound < StandardError; end

  attr_reader :ticker_name, :contract_type, :contract

  def initialize(user, params)
    @user          = user
    @ticker_name   = params[:ticker_name].upcase
    @contract_type = params[:type]
  end

  def call
    find_contract
  end

  private

  def find_contract
    @contract ||= contract_klass.find_by(ticker: ticker_name)
    fail(
      Contract::NotFound,
      "Cannot find contract with ticker_name #{ticker_name}"
    ) if @contract.nil?
  end

  def contract_klass
    return EquityContract if contract_type.nil?
    "#{contract_type.strip.capitalize}Contract".constantize
  end

end

我有以下相关规范:

require 'spec_helper'

describe SecurityQueryService do
  let(:user) { create(:customer) }
  let!(:equity_contract) { create(:equity_contract) }

  describe "#call" do
    describe "#find_contract" do
      it "returns contract based on contract type." do
        service = SecurityQueryService.new(user, {
          type: 'equity',
          ticker_name: equity_contract.ticker
        })
        service.call
        expect(EquityContract).to receive(:find_by)
      end    
    end
  end

end

每当我致电EquityContract时,我都希望find_by收到#call条消息。

在我运行规范的那一刻,我得到了:

 Failure/Error: expect(EquityContract).to receive(:find_by)
   (EquityContract(id: integer, ticker: string, name: string, country: string, currency: string, instrument_type: string, bloomberg_ticker: string, benchmark_index: string, skip_valuation: boolean, isin: string, currency_factor: decimal, daily_update: boolean, id_bb_unique: string, image: string, domain: string) (class)).find_by(*(any args))
       expected: 1 time with any arguments
       received: 0 times with any arguments

如何测试此行为?

1 个答案:

答案 0 :(得分:3)

您需要在调用方法之前设置模拟。 RSpec提供了两种方法:

  1. service.call之前移动期望:

    describe "#find_contract" do
      it "returns contract based on contract type." do
        service = SecurityQueryService.new(user,
          type: 'equity',
          ticker_name: equity_contract.ticker
        )
        expect(EquityContract).to receive(:find_by)
        service.call
      end    
    end
    
  2. 通过allow要调用的方法将方法设置为间谍,然后期望在事后调用它:

    describe "#find_contract" do
      it "returns contract based on contract type." do
        service = SecurityQueryService.new(user,
          type: 'equity',
          ticker_name: equity_contract.ticker
        )
        allow(EquityContract).to receive(:find_by)
        service.call
        expect(EquityContract).to have_received(:find_by)
      end    
    end
    
  3. 正如您所看到的,第一种方法需要最少的输入,但需要提前进行笨拙的思考。第二种方法更冗长,但更符合逻辑,因为它在方法调用之后提出了期望。