如何測試服務對象

Question

我已經寫下了服務對象。

class LeadGeneration 
    def initialize contact_id 
    @lead = Lead.find_or_initialize_by(contact_id: contact_id) 
    end 

    def lead 
    @lead 
    end 

    def activate_lead 
    lead.active! 
    end 

    def stale_lead 
    lead.stale! 
    end 
end 

我對如何測試它有點困惑。我寫了下面的規格

require 'spec_helper' 

describe LeadGeneration do 
    let(:contact) { FactoryGirl.create(:contact) } 

it "#activate_lead" do 
    lg = LeadGeneration.new(contact.id) 
    expect(lg.lead).to receive(:active!) 
    lg.activate_lead 
    end 

    it "#stale_lead" do 
    lg = LeadGeneration.new(contact.id) 
    expect(lg.lead).to receive(:stale!) 
    lg.stale_lead 
    end 
end 

該規範工作正常，但我想這樣做而不暴露生成的鉛。我究竟如何去做這件事。我可以使用

expect(Lead.any_instance).to receive(:active!) 

但是從我的閱讀中，這是不好的做法。有什麼想法嗎？

Answer 1

因爲此刻你正在測試是鉛對象接收郵件，您可以存根#lead返回您設置的期望模仿對象：

it "#activate_lead" do 
    lead = double('lead') 
    lg = LeadGeneration.new(contact.id) 
    # make #lead return our test double whenever it's called 
    allow(lg).to receive(:lead) { lead } 
    # our test double should receive the message 
    expect(lead).to receive(:active!) 
    lg.activate_lead 
end 

這意味着你可以移動#lead方法轉換爲私有/受保護的空間，因爲您不必直接調用它。