Lines Matching refs:mock
13 from unittest.mock import (ANY, call, AsyncMock, patch, MagicMock, Mock,
172 mock = AsyncMock()
173 self.assertTrue(iscoroutinefunction(mock))
177 mock = AsyncMock(foo)
178 self.assertTrue(iscoroutinefunction(mock))
179 self.assertTrue(inspect.iscoroutinefunction(mock))
182 mock = AsyncMock()
183 m = mock()
186 self.assertIn('assert_awaited', dir(mock))
190 mock = AsyncMock(foo)
191 self.assertTrue(iscoroutinefunction(mock))
192 self.assertTrue(inspect.iscoroutinefunction(mock))
199 mock = AsyncMock(fut)
200 self.assertIsInstance(mock, asyncio.Future)
257 self.assertIsInstance(mock_method.mock, AsyncMock)
263 # Verify the default values during mock setup
290 mock = mock_type(AsyncClass)
291 self.assertIsInstance(mock.async_method, AsyncMock)
292 self.assertIsInstance(mock.normal_method, MagicMock)
299 mock = Mock(AsyncClass)
300 self.assertIsInstance(mock.async_method, AsyncMock)
301 self.assertIsInstance(mock.normal_method, Mock)
304 mock = Mock(AsyncClass)
305 seal(mock)
307 mock.normal_method
309 mock.async_method
340 mock = AsyncMock(spec=normal_func)
341 self.assertIsInstance(mock, AsyncMock)
342 m = mock()
347 mock = AsyncMock(normal_func)
348 self.assertIsInstance(mock, AsyncMock)
349 m = mock()
400 mock = AsyncMock(spec_set=AsyncClass.async_method)
401 self.assertTrue(iscoroutinefunction(mock))
402 self.assertIsInstance(mock, AsyncMock)
405 mock = MagicMock(spec_set=AsyncClass)
406 self.assertTrue(iscoroutinefunction(mock.async_method))
407 self.assertFalse(iscoroutinefunction(mock.normal_method))
408 self.assertIsInstance(mock.async_method, AsyncMock)
409 self.assertIsInstance(mock.normal_method, MagicMock)
410 self.assertIsInstance(mock, MagicMock)
424 mock = AsyncMock(addition, return_value=10)
425 output = await mock(5)
431 mock = AsyncMock(addition, side_effect=Exception('err'))
433 await mock(5)
438 mock = AsyncMock(side_effect=addition)
439 result = await mock(5)
445 mock = AsyncMock(side_effect=addition)
446 result = await mock(5)
451 mock = AsyncMock(side_effect=vals)
453 self.assertEqual(await mock(), item)
456 await mock()
463 mock = AsyncMock(side_effect=vals)
464 self.assertEqual(await mock(), 1)
467 await mock()
471 mock = AsyncMock(return_value=value)
472 result = await mock()
478 mock = AsyncMock(return_value=fut)
479 result = await mock()
486 mock = AsyncMock(side_effect=[fut])
487 result = await mock()
491 await mock()
495 mock = AsyncMock(side_effect=effect)
497 result = await mock()
509 mock = AsyncMock(wraps=inner)
510 result = await mock()
512 mock.assert_awaited()
524 mock = AsyncMock(wraps=inner)
525 result = await mock()
527 mock.assert_awaited()
635 # We mock __aenter__ so it does not return self
714 # Check that the mock and the real thing bahave the same
764 self.mock = AsyncMock()
767 await self.mock(*args, **kwargs)
773 mock = AsyncMock(AsyncClass)
775 mock.async_method()
776 self.assertTrue(iscoroutinefunction(mock.async_method))
777 mock.async_method.assert_called()
778 mock.async_method.assert_called_once()
779 mock.async_method.assert_called_once_with()
781 mock.assert_awaited()
783 mock.async_method.assert_awaited()
786 mock = AsyncMock(AsyncClass)
787 mock_coroutine = mock.async_method()
788 mock.async_method.assert_called()
789 mock.async_method.assert_called_once()
790 mock.async_method.assert_called_once_with()
792 mock.async_method.assert_awaited()
796 mock.async_method.assert_called_once()
797 mock.async_method.assert_awaited()
798 mock.async_method.assert_awaited_once()
799 mock.async_method.assert_awaited_once_with()
803 self.mock.assert_awaited()
806 self.mock.assert_called()
809 self.mock.assert_called_once()
810 self.mock.assert_awaited_once()
813 mock = AsyncMock(AsyncClass)
814 coroutine = mock.async_method()
818 mock.async_method()
820 mock.async_method.assert_awaited()
821 mock.async_method.assert_called()
823 mock.async_method.assert_awaited()
824 mock.async_method.assert_awaited_once()
827 mock = AsyncMock(AsyncClass)
828 coroutine = mock.async_method()
829 mock.async_method.assert_called_once()
834 mock.async_method.assert_awaited()
838 self.mock.assert_awaited()
840 self.mock.assert_called()
843 run(self._await_coroutine(self.mock))
846 self.mock.assert_awaited()
848 self.mock.assert_called()
853 self.mock('foo')
854 self.mock.assert_has_calls(kalls)
856 self.mock.assert_has_awaits(kalls)
860 self.mock()
862 self.assertEqual(self.mock.mock_calls, kalls_empty)
865 self.mock('foo')
867 self.mock('baz')
869 self.assertEqual(self.mock.mock_calls, mock_kalls)
888 self.mock.something(3, fish=None)
890 self.mock.something_else.something(6, cake=sentinel.Cake)
892 self.assertEqual(self.mock.method_calls, [
897 self.assertEqual(self.mock.something_else.method_calls,
902 def assert_attrs(mock):
905 attr = getattr(mock, name)
910 assert_attrs(self.mock)
912 self.mock()
914 self.mock(1, 2)
916 self.mock(a=3)
918 self.mock.reset_mock()
919 assert_attrs(self.mock)
932 self.mock.assert_awaited()
935 self.mock.assert_awaited()
939 self.mock.assert_awaited_once()
942 self.mock.assert_awaited_once()
946 self.mock.assert_awaited_once()
951 self.mock.assert_awaited_with('foo')
956 self.mock.assert_awaited_with('foo')
959 self.mock.assert_awaited_with('foo')
963 self.mock.assert_awaited_with('foo')
967 self.mock.assert_awaited_once_with('foo')
970 self.mock.assert_awaited_once_with('foo')
974 self.mock.assert_awaited_once_with('foo')
978 self.mock.assert_any_await('foo')
982 self.mock.assert_any_await('foo')
985 self.mock.assert_any_await('foo')
988 self.mock.assert_any_await('foo')
994 self.mock.assert_has_awaits(calls)
999 self.mock.assert_has_awaits(calls)
1003 self.mock.assert_has_awaits(calls)
1006 self.mock.assert_has_awaits(calls)
1009 self.mock.assert_has_awaits(calls)
1017 self.mock.assert_has_awaits([call(ANY, 1)])
1018 self.mock.assert_awaited_with(ANY, 1)
1019 self.mock.assert_any_await(ANY, 1)
1038 self.mock.assert_has_awaits(calls, any_order=True)
1042 self.mock.assert_has_awaits(calls, any_order=True)
1046 self.mock.assert_has_awaits(calls, any_order=True)
1049 self.mock.assert_has_awaits(calls, any_order=True)
1052 self.mock.assert_has_awaits(calls, any_order=True)
1055 self.mock.assert_not_awaited()
1059 self.mock.assert_not_awaited()
1064 self.mock = AsyncMock(spec=f)
1073 self.mock.assert_has_awaits([call()])
1085 self.mock.assert_has_awaits([call(), call(1, 2)])