Metoda stubsowania tylko dla określonych argumentów

Question

Czy istnieje sposób na metodę stub dla tylko określonych argumentów. Coś takiego

boss.stub(:fire!).with(employee1).and_return(true) 

Jeśli jakikolwiek inny pracownik przechodzi do boss.fire! metodzie Wezmę boss received unexpected message błąd, ale to, co naprawdę chciałbym tylko, aby zastąpić metodę dla konkretnego argumentu, i zostawić to dla wszystkich innych .

Jakieś pomysły, jak to zrobić?

Answer 1

Możesz dodać domyślny niedopałek dla metody fire! które nazywamy oryginalna realizacja:

boss.stub(:fire!).and_call_original 
boss.stub(:fire!).with(employee1).and_return(true) 

rspec 3 Składnia (@ pk-nb)

allow(boss).to receive(:fire!).and_call_original 
allow(boss).to receive(:fire!).with(employee1).and_return(true) 

Answer 2

Można spróbować napisać własną metodę stubbing, z kodem jak ten

fire_method = boss.method(:fire!) 
boss.stub!(:fire!) do |employee| 
    if employee == employee1 
    true 
    else 
    fire_method.call(*args) 
    end 
end