Lines Matching refs:mock
5 from unittest.mock import AsyncMock, Mock, MagicMock, _magics
12 mock = Mock()
13 self.assertFalse(hasattr(mock, '__getitem__'))
15 mock.__getitem__ = Mock()
16 self.assertTrue(hasattr(mock, '__getitem__'))
18 del mock.__getitem__
19 self.assertFalse(hasattr(mock, '__getitem__'))
23 mock = MagicMock()
25 del mock.__getitem__
26 self.assertRaises(TypeError, lambda: mock['foo'])
28 mock = MagicMock()
30 mock['foo']
31 del mock.__getitem__
32 self.assertRaises(TypeError, lambda: mock['foo'])
36 mock = Mock()
40 mock.__getitem__ = f
41 self.assertIsNot(mock.__getitem__, f)
42 self.assertEqual(mock['foo'], (mock, 'fish'))
43 self.assertEqual(mock.__getitem__('foo'), (mock, 'fish'))
45 mock.__getitem__ = mock
46 self.assertIs(mock.__getitem__, mock)
59 mock = Mock()
60 self.assertEqual(repr(mock), "<Mock id='%s'>" % id(mock))
61 mock.__repr__ = lambda s: 'foo'
62 self.assertEqual(repr(mock), 'foo')
66 mock = Mock()
67 self.assertEqual(str(mock), object.__str__(mock))
68 mock.__str__ = lambda s: 'foo'
69 self.assertEqual(str(mock), 'foo')
73 mock = Mock()
75 self.assertRaises(TypeError, lambda: mock['foo'])
77 del mock['foo']
79 mock['foo'] = 3
91 mock.__setitem__ = setitem
92 mock.__getitem__ = getitem
93 mock.__delitem__ = delitem
95 self.assertRaises(KeyError, lambda: mock['foo'])
96 mock['foo'] = 'bar'
98 self.assertEqual(mock['foo'], 'bar')
99 del mock['foo']
104 original = mock = Mock()
105 mock.value = 0
107 self.assertRaises(TypeError, lambda: mock + 3)
110 mock.value += other
112 mock.__add__ = add
113 self.assertEqual(mock + 3, mock)
114 self.assertEqual(mock.value, 3)
116 del mock.__add__
117 def iadd(mock):
118 mock += 3
119 self.assertRaises(TypeError, iadd, mock)
120 mock.__iadd__ = add
121 mock += 6
122 self.assertEqual(mock, original)
123 self.assertEqual(mock.value, 9)
125 self.assertRaises(TypeError, lambda: 3 + mock)
126 mock.__radd__ = add
127 self.assertEqual(7 + mock, mock)
128 self.assertEqual(mock.value, 16)
131 original = mock = Mock()
132 mock.value = 32
133 self.assertRaises(TypeError, lambda: mock / 2)
136 mock.value /= other
138 mock.__truediv__ = truediv
139 self.assertEqual(mock / 2, mock)
140 self.assertEqual(mock.value, 16)
142 del mock.__truediv__
143 def itruediv(mock):
144 mock /= 4
145 self.assertRaises(TypeError, itruediv, mock)
146 mock.__itruediv__ = truediv
147 mock /= 8
148 self.assertEqual(mock, original)
149 self.assertEqual(mock.value, 2)
151 self.assertRaises(TypeError, lambda: 8 / mock)
152 mock.__rtruediv__ = truediv
153 self.assertEqual(0.5 / mock, mock)
154 self.assertEqual(mock.value, 4)
157 mock = Mock()
159 self.assertEqual(hash(mock), Mock.__hash__(mock))
163 mock.__hash__ = _hash
164 self.assertEqual(hash(mock), 3)
176 mock = Mock()
179 mock.__lt__ = mock.__gt__ = mock.__le__ = mock.__ge__ = comp
180 self. assertTrue(mock < 3)
181 self. assertTrue(mock > 3)
182 self. assertTrue(mock <= 3)
183 self. assertTrue(mock >= 3)
200 for mock in Mock(), MagicMock():
201 self.assertEqual(mock == mock, True)
202 self.assertIsInstance(mock == mock, bool)
203 self.assertEqual(mock != mock, False)
204 self.assertIsInstance(mock != mock, bool)
205 self.assertEqual(mock == object(), False)
206 self.assertEqual(mock != object(), True)
210 mock.__eq__ = eq
211 self.assertTrue(mock == 3)
212 self.assertFalse(mock == 4)
216 mock.__ne__ = ne
217 self.assertTrue(mock != 3)
218 self.assertFalse(mock != 4)
220 mock = MagicMock()
221 mock.__eq__.return_value = True
222 self.assertIsInstance(mock == 3, bool)
223 self.assertEqual(mock == 3, True)
225 mock.__ne__.return_value = False
226 self.assertIsInstance(mock != 3, bool)
227 self.assertEqual(mock != 3, False)
231 mock = Mock()
233 self.assertRaises(TypeError, len, mock)
234 self.assertRaises(TypeError, iter, mock)
235 self.assertRaises(TypeError, lambda: 'foo' in mock)
237 mock.__len__ = lambda s: 6
238 self.assertEqual(len(mock), 6)
240 mock.__contains__ = lambda s, o: o == 3
241 self.assertIn(3, mock)
242 self.assertNotIn(6, mock)
244 mock.__iter__ = lambda s: iter('foobarbaz')
245 self.assertEqual(list(mock), list('foobarbaz'))
249 mock = MagicMock()
251 mock.__iter__.return_value = iter([1, 2, 3])
252 self.assertEqual(list(mock), [1, 2, 3])
254 getattr(mock, '__bool__').return_value = False
255 self.assertFalse(hasattr(mock, '__nonzero__'))
256 self.assertFalse(bool(mock))
259 self.assertTrue(hasattr(mock, entry))
260 self.assertFalse(hasattr(mock, '__imaginary__'))
264 mock = MagicMock()
265 self.assertIsInstance(mock == object(), bool)
266 self.assertIsInstance(mock != object(), bool)
268 self.assertEqual(mock == object(), False)
269 self.assertEqual(mock != object(), True)
270 self.assertEqual(mock == mock, True)
271 self.assertEqual(mock != mock, False)
274 mock = AsyncMock()
275 self.assertEqual(int(mock), 1)
276 self.assertEqual(complex(mock), 1j)
277 self.assertEqual(float(mock), 1.0)
278 self.assertNotIn(object(), mock)
279 self.assertEqual(len(mock), 0)
280 self.assertEqual(list(mock), [])
281 self.assertEqual(hash(mock), object.__hash__(mock))
282 self.assertEqual(str(mock), object.__str__(mock))
283 self.assertTrue(bool(mock))
284 self.assertEqual(round(mock), mock.__round__())
285 self.assertEqual(math.trunc(mock), mock.__trunc__())
286 self.assertEqual(math.floor(mock), mock.__floor__())
287 self.assertEqual(math.ceil(mock), mock.__ceil__())
288 self.assertTrue(iscoroutinefunction(mock.__aexit__))
289 self.assertTrue(iscoroutinefunction(mock.__aenter__))
290 self.assertIsInstance(mock.__aenter__, AsyncMock)
291 self.assertIsInstance(mock.__aexit__, AsyncMock)
295 self.assertEqual(oct(mock), '0o1')
296 self.assertEqual(hex(mock), '0x1')
300 mock = MagicMock()
301 self.assertEqual(int(mock), 1)
302 self.assertEqual(complex(mock), 1j)
303 self.assertEqual(float(mock), 1.0)
304 self.assertNotIn(object(), mock)
305 self.assertEqual(len(mock), 0)
306 self.assertEqual(list(mock), [])
307 self.assertEqual(hash(mock), object.__hash__(mock))
308 self.assertEqual(str(mock), object.__str__(mock))
309 self.assertTrue(bool(mock))
310 self.assertEqual(round(mock), mock.__round__())
311 self.assertEqual(math.trunc(mock), mock.__trunc__())
312 self.assertEqual(math.floor(mock), mock.__floor__())
313 self.assertEqual(math.ceil(mock), mock.__ceil__())
314 self.assertTrue(iscoroutinefunction(mock.__aexit__))
315 self.assertTrue(iscoroutinefunction(mock.__aenter__))
316 self.assertIsInstance(mock.__aenter__, AsyncMock)
317 self.assertIsInstance(mock.__aexit__, AsyncMock)
321 self.assertEqual(oct(mock), '0o1')
322 self.assertEqual(hex(mock), '0x1')
327 mock = MagicMock()
328 expected_path = mock.__fspath__()
329 mock.reset_mock()
331 self.assertEqual(os.fspath(mock), expected_path)
332 mock.__fspath__.assert_called_once()
339 mock = Mock(spec=Iterable)
340 self.assertRaises(AttributeError, lambda: mock.__iter__)
342 mock.__iter__ = Mock(return_value=iter([]))
343 self.assertEqual(list(mock), [])
347 mock = Mock(spec=NonIterable)
348 self.assertRaises(AttributeError, lambda: mock.__iter__)
351 mock.__int__ = Mock(return_value=iter([]))
354 mock = MagicMock(spec=Iterable)
355 self.assertEqual(list(mock), [])
363 mock = Mock(spec_set=Iterable)
364 self.assertRaises(AttributeError, lambda: mock.__iter__)
366 mock.__iter__ = Mock(return_value=iter([]))
367 self.assertEqual(list(mock), [])
371 mock = Mock(spec_set=NonIterable)
372 self.assertRaises(AttributeError, lambda: mock.__iter__)
375 mock.__int__ = Mock(return_value=iter([]))
378 mock = MagicMock(spec_set=Iterable)
379 self.assertEqual(list(mock), [])
384 mock = MagicMock()
386 mock.__setattr__ = lambda self, name: None
394 mock = MagicMock()
395 attr = mock.foo
397 # the type of every mock (or magicmock) is a custom subclass
402 returned = mock()
407 mock = MagicMock()
408 self.assertIsInstance(mock.__getitem__, MagicMock)
410 mock[1][2].__getitem__.return_value = 3
411 self.assertEqual(mock[1][2][3], 3)
415 mock = MagicMock()
416 str(mock)
417 self.assertTrue(mock.__str__.called)
418 mock.reset_mock()
419 self.assertFalse(mock.__str__.called)
424 for mock in Mock(), MagicMock():
427 mock.__dir__ = _dir
428 self.assertEqual(dir(mock), ['foo'])