[TIP] Return different values based on an argument passed to a method

[Marcin Zajączkowski]

Hi,

Is it possible to simple stub a method's return value when called with an argument (any argument or specific one)?

I tried with:
mock = Mock()
mock.foo(ANY).return_value(False)

but in a debugger I see that in my code mock.foo(<<some_object>>) returns a mock instead of False value. If not the returning value I could use assert_called_with with custom mather, but how to return different value based on an argument?

Regards
Marcin