Replace AsyncMock with simple mock class in FastAPI wiring tests for better compatibility

This commit is contained in:
Roman Mogylatov 2020-12-05 21:58:42 -05:00
parent 0b7cf3254e
commit c1e53e5edd

View File

@ -80,10 +80,11 @@ class WiringFastAPITest(AsyncTestCase):
super().tearDown() super().tearDown()
def test_depends_marker_injection(self): def test_depends_marker_injection(self):
service_mock = mock.AsyncMock(spec=web.Service) class ServiceMock:
service_mock.process.return_value = 'Foo' async def process(self):
return 'Foo'
with web.container.service.override(service_mock): with web.container.service.override(ServiceMock()):
response = self._run(self.client.get('/')) response = self._run(self.client.get('/'))
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)