def test_my_method_shouldReturnMultipleValues_whenMyMethodReturnsSomeValue(self, mock_my_method):
list_of_return_values= [True,False,False]
def side_effect():
return list_of_return_values.pop()
mock_my_method.side_effect = side_effect
some_other_class = SomeOtherClassThatUsesMyClass()
self.assertFalse(some_other_class.method_under_test())
self.assertFalse(some_other_class.method_under_test())
self.assertTrue(some_other_class.method_under_test())
def test_my_method_shouldReturnTrue_whenSomeOtherClassMethodIsCalledAndAReturnValueIsSet(self, mock_my_class):
mc = mock_my_class.return_value
mc.my_method.return_value = True
some_other_class = SomeOtherClassThatUsesMyClass()
result = some_other_class.method_under_test()
self.assertTrue(result)
def test_my_method_shouldCallMyClassMethodMyMethod_whenSomeOtherClassMethodIsCalled(self, mock_my_class):
some_other_class = SomeOtherClassThatUsesMyClass()
some_other_class.method_under_test()
self.assertTrue(mock_my_class.called)
def test_my_method_shouldReturnTrue_whenMyMethodReturnsSomeValue(self, mock_my_method):
mock_my_method.return_value=True
some_other_class = SomeOtherClassThatUsesMyClass()
result = some_other_class.method_under_test()
self.assertTrue(result)
