Ruby Rspec3期望一次接收內部方法調用

Question

我想嘗試測試對象是否在特定情況下使用優化。所以，我有這種代碼

class Bar 
    attr_reader :n 
    def initialize(n) 
    @n=n 
    end 

    def a 
    if @n <= 3 
     b1 
    else 
     b2 
    end 
    end 

    def b1 
    @n+=1 
    end 

    def b2 
    @n+=1 ##super fast addition 
    end 
end 

我嘗試寫這樣的一個rspec的

bar=Bar.new(5) 
allow(bar).to receive(:b2).and_call_original 

bar.a 
expect(bar).to receive(:b2).once 
expect(bar.n).to eq 6 

但它不工作...你kown這可能嗎？如果是的話，怎麼樣？

Answer 1

您應該將receive期望之前的方法調用。您也可以將allow和expect合併爲一行：

bar = Bar.new(5) 
expect(bar).to receive(:b2).once.and_call_original 
bar.a 
expect(bar.n).to eq 6