test - tdd ruby on rails

Método de código auxiliar solo en la primera llamada con Rspec

Question

¿Cómo puedo aplastar un método solo en la primera llamada y en la segunda debe comportarse como se espera?

Tengo el siguiente método:

def method
do_stuff
rescue => MyException
sleep rand
retry
end

Quiero la primera llamada de do_stuff para generar MyException , pero en la segunda llamada, se comporta normalmente. Necesito lograr esto para probar mi bloque de rescue sin obtener un bucle infinito.

¿Hay una manera de lograr esto?

Answer 1

Puede pasar un bloque a un código auxiliar que se invocará cuando se llame al código auxiliar. A continuación, puede realizar la desincrustación allí, además de hacer lo que necesite.

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

Parece que funciona muy bien aquí.

$ rspec stub.rb -f doc
Stub once
should only stub once
Finished in 0.00058 seconds
1 example, 0 failures

Alternativamente, puede simplemente rastrear la cantidad de llamadas y devolver diferentes resultados para el talón basado en el recuento de llamadas:

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