Python Mock库

import unittest import pmock import testsupport class VerificationErrorTest(unittest.TestCase): def test_create_error(self): class Mocker: def __str__(self): return "matchers" error = pmock.VerificationError.create_error("problem", Mocker()) self.assertEqual(error.msg, "problem: matchers") class MatchErrorTest(unittest.TestCase): class MockInvocation: def __str__(self): return "call" class Mock: def __init__(self, invokables_str): self._str = invokables_str def invokables_str(self): return self._str def test_empty_invokables(self): error = pmock.MatchError.create_error("msg", self.MockInvocation(), self.Mock("")) self.assertEqual(error.msg, "msg\ninvoked call") def test_non_empty_invokables(self): error = pmock.MatchError.create_error("msg", self.MockInvocation(), self.Mock("invokables")) self.assertEqual(error.msg, "msg\ninvoked call\nin:\ninvokables") class ArgumentsMatcherTestMixin(object): def __init__(self, matcher_class): self.matcher_class = matcher_class def test_matches_arguments(self): arg1 = [] args_matcher = self.matcher_class((pmock.same(arg1),), {}) self.assert_( args_matcher.matches(pmock.Invocation("snake", (arg1,), {}))) def test_matches_keyword_arguments(self): arg2 = [] args_matcher = self.matcher_class((), {"food": pmock.same(arg2)}) invocation = pmock.Invocation("snake", (), {"food": arg2}) self.assert_(args_matcher.matches(invocation)) def test_matches_both_types_of_arguments(self): arg1 = [] arg2 = [] args_matcher = self.matcher_class((pmock.same(arg1),), {"food": pmock.same(arg2)}) invocation = pmock.Invocation("snake", (arg1,), {"food": arg2}) self.assert_(args_matcher.matches(invocation)) def test_insufficient_arguments(self): args_matcher = self.matcher_class((pmock.eq("slither"),), {}) self.assert_( not args_matcher.matches(pmock.Invocation("snake", (), {}))) def test_insufficient_keyword_arguments(self): args_matcher = self.matcher_class((), {"food": pmock.eq("goat")}) self.assert_( not args_matcher.matches(pmock.Invocation("snake", (), {}))) def test_unmatched_argument(self): class UnmatchedConstraint: def eval(self, invocation): return False args_matcher = self.matcher_class((UnmatchedConstraint(),), {}) invocation = pmock.Invocation("snake", ("slither",), {}) self.assert_(not args_matcher.matches(invocation)) def test_unmatched_keyword_argument(self): class UnmatchedConstraint: def eval(self, invocation): return False args_matcher = self.matcher_class((), {"food": UnmatchedConstraint()}) invocation = pmock.Invocation("snake", (), {"food": "goat"}) self.assert_(not args_matcher.matches(invocation)) class AllArgumentsMatcherTest(ArgumentsMatcherTestMixin, unittest.TestCase): def __init__(self, *args): ArgumentsMatcherTestMixin.__init__(self, pmock.AllArgumentsMatcher) unittest.TestCase.__init__(self, *args) def test_extra_arguments(self): args_matcher = pmock.AllArgumentsMatcher((pmock.eq("slither"),), {}) self.assert_(not args_matcher.matches( pmock.Invocation("snake", ("slither", "hiss"), {}))) def test_extra_keyword_arguments(self): args_matcher = pmock.AllArgumentsMatcher((), {"food": pmock.eq("goat")}) self.assert_(not args_matcher.matches( pmock.Invocation("snake", (), {"food": "goat", "colour": "red"}))) def test_no_arguments(self): args_matcher = pmock.AllArgumentsMatcher() self.assert_(args_matcher.matches(pmock.Invocation("snake", (), {}))) self.assert_( not args_matcher.matches(pmock.Invocation("snake", ("hiss",), {}))) self.assert_(not args_matcher.matches( pmock.Invocation("snake", (), {"food": "goat"}))) def test_str(self): matcher = pmock.AllArgumentsMatcher((pmock.eq("slither"),), {"food": pmock.eq("goat")}) self.assertEqual(str(matcher), "(pmock.eq('slither'), food=pmock.eq('goat'))") def test_empty_str(self): self.assertEqual(str(pmock.AllArgumentsMatcher()), "()") class LeastArgumentsMatcherTest(ArgumentsMatcherTestMixin, unittest.TestCase): def __init__(self, *args): ArgumentsMatcherTestMixin.__init__(self, pmock.LeastArgumentsMatcher) unittest.TestCase.__init__(self, *args) def test_extra_arguments(self): args_matcher = pmock.LeastArgumentsMatcher((pmock.eq("slither"),), {}) self.assert_(args_matcher.matches( pmock.Invocation("snake", ("slither", "hiss"), {}))) def test_extra_keyword_arguments(self): args_matcher = pmock.LeastArgumentsMatcher((), {"food": pmock.eq("goat")}) self.assert_(args_matcher.matches( pmock.Invocation("snake", (), {"food": "goat", "colour": "red"}))) def test_any_arguments(self): args_matcher = pmock.LeastArgumentsMatcher() self.assert_(args_matcher.matches(pmock.Invocation("snake", (), {}))) self.assert_( args_matcher.matches(pmock.Invocation("snake", ("hiss",), {}))) self.assert_(args_matcher.matches( pmock.Invocation("snake", ("constrict",), {"food": "goat"}))) def test_str(self): matcher = pmock.LeastArgumentsMatcher((pmock.eq("slither"),), {"food": pmock.eq("goat")}) self.assertEqual(str(matcher), "(pmock.eq('slither'), food=pmock.eq('goat'), ...)") def test_empty_str(self): self.assertEqual(str(pmock.LeastArgumentsMatcher()), "(...)") class InvocationMockerTest(unittest.TestCase): class MockMatcher: def __init__(self, matches): self._matches = matches def matches(self, invocation): self.matches_invocation = invocation return self._matches def invoked(self, invocation): self.invoked_invocation = invocation def test_matches(self): mocker = pmock.InvocationMocker(self.MockMatcher(True)) self.assert_(mocker.matches(pmock.Invocation("duck", (), {}))) def test_unmatched(self): mocker = pmock.InvocationMocker(self.MockMatcher(False)) self.assert_(not mocker.matches(pmock.Invocation("duck", (), {}))) def test_added_matcher_unmatched(self): mocker = pmock.InvocationMocker(self.MockMatcher(True)) mocker.add_matcher(self.MockMatcher(False)) self.assert_(not mocker.matches(pmock.Invocation("duck", (), {}))) def test_matches_passes_invocation_to_matcher(self): matcher = self.MockMatcher(True) mocker = pmock.InvocationMocker(matcher) invocation = pmock.Invocation("duck", (), {}) mocker.matches(invocation) self.assertEqual(matcher.matches_invocation, invocation) def test_no_stub_returns_none(self): mocker = pmock.InvocationMocker(self.MockMatcher(True)) self.assert_(mocker.invoke(pmock.Invocation("duck", (), {})) is None) def test_invoke_returns_stubs_value(self): class MockStub: def invoke(self, invocation): return 'value' mocker = pmock.InvocationMocker(self.MockMatcher(True)) mocker.set_stub(MockStub()) self.assert_(mocker.invoke(pmock.Invocation("duck", (), {}))