1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
from .ut_utils import ForgeTestCase
from forge import UnexpectedSetattr
class WildcardTest(ForgeTestCase):
def tearDown(self):
self.forge.verify()
self.assertNoMoreCalls()
super(WildcardTest, self).tearDown()
def test__wildcard_functions(self):
wcf = self.forge.create_wildcard_function_stub()
self._do_all_sorts_of_calls(wcf)
self.forge.replay()
self._do_all_sorts_of_calls(wcf)
def _do_all_sorts_of_calls(self, func):
func(1, 2, 3)
func(1, 2)
func()
func(a=2)
func(1, 2, a=2, d=3)
def test__wildcard_record_replay(self):
wc = self.forge.create_wildcard_mock()
wc.f(1, 2, 3)
wc.g(1, 2, 3, d=4)
self.forge.replay()
wc.f(1, 2, 3)
wc.g(1, 2, 3, d=4)
def test__wildcard_function_names(self):
wc = self.forge.create_wildcard_function_stub('some_name')
self.assertIn('some_name', str(wc))
self.assertIn('some_name', repr(wc))
def test__wildcard_access_to_unrecorded_methods(self):
wc = self.forge.create_wildcard_mock()
self.forge.replay()
with self.assertRaises(AttributeError):
wc.f
def test__wildcard_access_to_unrecorded_methods_getattr_in_record(self):
wc = self.forge.create_wildcard_mock()
wc.f # no call!
self.forge.replay()
with self.assertRaises(AttributeError):
wc.f # todo: this shouldn't be ok... to be solved when setattr expectations are added
def test__setattr_forbidden(self):
wc = self.forge.create_wildcard_mock()
wc.a = 2
self.forge.replay()
self.assertEqual(wc.a, 2)
with self.assertRaises(UnexpectedSetattr):
wc.a = 2
with self.assertRaises(UnexpectedSetattr):
wc.a = 3
with self.assertRaises(UnexpectedSetattr):
wc.b = 3
def test__expect_setattr(self):
wc = self.forge.create_wildcard_mock()
wc.__forge__.expect_setattr("a", 2)
self.forge.replay()
wc.a = 2
self.assertEqual(wc.a, 2)
def test__repr(self):
name = 'some_name'
wc = self.forge.create_wildcard_mock(name)
self.assertIn(name, str(wc))
self.assertIn(name, repr(wc))
def test__method_repr(self):
wc = self.forge.create_wildcard_mock()
m = wc.method_with_specific_name
self.assertTrue("method_with_specific_name" in repr(m))
self.assertTrue("method_with_specific_name" in str(m))
def test__special_methods_ok(self):
wc = self.forge.create_wildcard_mock()
f = self.forge.create_wildcard_function_stub()
with wc:
f()
wc.__len__().and_return(666)
wc.__iter__().and_return(iter(range(10)))
self.forge.replay()
with wc:
f()
self.assertEqual(len(wc), 666)
self.assertEqual([x for x in wc], list(range(10)))
|
