20

最初の呼び出しでのみメソッドをスタブし、2 回目の呼び出しでは期待どおりに動作するようにするにはどうすればよいですか?

私は次の方法を持っています：

def method
  do_stuff
rescue => MyException
  sleep rand
  retry
end

do_stuff最初の呼び出しでraiseを実行したいのですMyExceptionが、2 回目の呼び出しでは正常に動作します。rescue無限ループにならずにブロックをテストするには、これを達成する必要があります。

これを達成する方法はありますか？

4

1 に答える 1

18

スタブが呼び出されたときに呼び出されるスタブにブロックを渡すことができます。その後、必要なことを行うだけでなく、そこでスタブ解除を実行できます。

class Foo
  def initialize
    @calls = 0
  end

  def be_persistent
    begin
      increment
    rescue
      retry
    end
  end

  def increment
    @calls += 1
  end
end

describe "Stub once" do
  let(:f) { Foo.new }
  before {
    f.stub(:increment) { f.unstub(:increment); raise "boom" }
  }

  it "should only stub once" do
    f.be_persistent.should == 1
  end
end

ここではうまく機能しているようです。

$ rspec stub.rb -f doc

Stub once
  should only stub once

Finished in 0.00058 seconds
1 example, 0 failures

または、呼び出し回数を追跡し、呼び出し回数に基づいてスタブの異なる結果を返すこともできます。

describe "Stub once" do
  let(:f) { Foo.new }

  it "should return different things when re-called" do
    call_count = 0
    f.should_receive(:increment).twice {
      if (call_count += 1) == 1
        raise "boom"
      else
        "success!"
      end
    }

    f.be_persistent.should == "success!"
  end
end
于 2013-08-21T23:12:17.600 に答える