Ejemplo n.º 1
0
 async def test_call(self, *args):
     with asynctest.mock.patch('aiohttp.request', create=True) as aiohttp_req_mock:
         mock_req = CoroutineMock()
         mock_req.__aexit__ = CoroutineMock()
         mock_req.__aenter__ = CoroutineMock()
         aiohttp_req_mock.return_value = mock_req
         expected_method = 'some-value'
         expected_url = 'some-value'
         expected_params = {}
         expected_data = {}
         expected_json = {}
         expected_cookies = {}
         expected_headers = {}
         expected_auth = {}
         await Api.call(method=expected_method, url=expected_url, params=expected_params, data=expected_data, json=expected_json, cookies=expected_cookies, headers=expected_headers, auth=expected_auth)
         aiohttp_req_mock.assert_called_with(
             method=expected_method,
             url=expected_url,
             params=expected_params,
             data=expected_data,
             json=expected_json,
             cookies=expected_cookies,
             headers=expected_headers,
             auth=expected_auth)
Ejemplo n.º 2
0
 def make_mock(exc):
     cm = CoroutineMock()
     cm.__aenter__ = CoroutineMock(side_effect=exc)
     cm.__aexit__ = CoroutineMock()
     return cm