def test_fails_second_time(self):
self.raw_api.do_it.side_effect = [
InvalidAuthToken('message', 'code'),
InvalidAuthToken('message', 'code'),
]
with self.assertRaises(InvalidAuthToken):
self.session.do_it()
def __call__(self, *args, **kwargs):
if self.__called:
return self.__original_function(*args, **kwargs)
self.__called = True
raise InvalidAuthToken('message', 401)
def test_works_second_time(self):
self.raw_api.do_it.side_effect = [
InvalidAuthToken('message', 'code'),
'ok',
]
self.assertEqual('ok', self.session.do_it())