def test_resets_args_called_on_stubs(self):
stub = Stub('stub')
stub.set_default_response('response')
self.test_environment.add_stub(self.obj, 'method', stub)
stub(123)
self.test_environment.reset()
self.assertEqual([], stub.was_called_with)
def test_resets_mock_expectations(self):
stub = Stub('stub')
stub.set_default_response('response')
expectation = ShouldReceiveExpectation(stub, AnyArgs)
self.test_environment.add_mock_expectation(expectation)
self.test_environment.reset()
self.test_environment.verify_expectations()
def test_sets_up_and_resets_stubs(self):
stub = Stub('stub')
stub.set_default_response('response')
original_methods = [self.obj.method1, self.obj.method2]
self.test_environment.add_stub(self.obj, 'method1', stub)
self.test_environment.add_stub(self.obj, 'method2', stub)
self.assertEqual('response', self.obj.method1('any', 'args'))
self.assertEqual('response', self.obj.method2('any', 'args'))
self.test_environment.reset()
self.assertEqual(original_methods, [self.obj.method1, self.obj.method2])
def test_verifies_a_method_was_called_with_the_right_arguments(self):
expectation = ShouldReceiveExpectation(self.stub, Args.make(1))
self.stub(1)
expectation.verify()
new_stub = Stub('new_stub')
new_stub.set_default_response('response')
expectation = ShouldReceiveExpectation(new_stub, Args.make(1))
new_stub('wrong', 'args')
try:
expectation.verify()
except AssertionError, e:
self.assertEqual("Expected new_stub(1) to be called but it wasn't.",
str(e))
def test_verifies_mock_expectations(self):
stub = Stub('stub')
stub.set_default_response('response')
passing_expectation = ShouldReceiveExpectation(stub, AnyArgs)
failing_expectation = ShouldReceiveExpectation(stub,
Args.make('some args'))
self.test_environment.add_mock_expectation(passing_expectation)
self.test_environment.add_mock_expectation(failing_expectation)
stub('random args')
try:
self.test_environment.verify_expectations()
except AssertionError, e:
self.assertEqual("Expected stub('some args') to be called but it "
"wasn't.", str(e))
def create_stub():
stub = Stub('stub')
stub.set_default_response('response')
return stub