def test_end_of_calls(self): call_stack = CallStack(self.fake) c = Call(self.fake) c.return_val = 1 call_stack.add_call(c) eq_(call_stack(), 1) call_stack()
def test_get_call_object(self): call_stack = CallStack(self.fake) c = Call(self.fake) call_stack.add_call(c) assert call_stack.get_call_object() is c d = Call(self.fake) call_stack.add_call(d) assert call_stack.get_call_object() is d
def test_reset(self): call_stack = CallStack(self.fake) c = Call(self.fake) c.return_val = 1 call_stack.add_call(c) c = Call(self.fake) c.return_val = 2 call_stack.add_call(c) eq_(call_stack(), 1) eq_(call_stack(), 2) call_stack.reset() eq_(call_stack(), 1) eq_(call_stack(), 2)
def test_with_initial_calls(self): c = Call(self.fake) c.return_val = 1 call_stack = CallStack(self.fake, initial_calls=[c]) eq_(call_stack(), 1)
def test_no_calls(self): call_stack = CallStack(self.fake) call_stack()