Юнит-тестирование на Python, вложенное "async with", как макетировать / патчить - PullRequest
Новая
       18

Юнит-тестирование на Python, вложенное "async with", как макетировать / патчить

0 голосов
/ 28 ноября 2018

Я немного боролся с модульным тестированием фрагмента асинхронного кода, который использует вложенную «асинхронность с».

Python версии 3.6.3 aiohttp версия 3.4.4

версия функции с голыми суставами, которую я хочу выполнить модульным тестом: 

async def main():

    url = 'http://www.google.com'
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as response:
            return await response.read()

А урезанный код модульного теста выглядит так: 

class AsyncMock(MagicMock):
    async def __call__(self, *args, **kwargs):
        return super(AsyncMock, self).__call__(*args, **kwargs)


class TestAsyncTest(unittest.TestCase):

    @patch('aiohttp.ClientSession', new_callable=AsyncMock)
    def test_async_test(self, mock_session):

        loop = asyncio.get_event_loop()
        result = loop.run_until_complete(main())
        print('result={}'.format(result))
        loop.close()

Вопрос :Как патчить вложенные вызовы, я хочу, чтобы функция «get» вызывала исключение.Я подозреваю, что это должно выглядеть примерно так: 

mock_session.__aenter__().get().__aenter__.side_effect = asyncio.TimeoutError()

Но это дает мне ошибку: 

E
======================================================================
ERROR: test_async_test (test_test_async_unittest.TestAsyncTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/lib/python3.6/unittest/mock.py", line 1179, in patched
    return func(*args, **keywargs)
  File "/ntfs/projects/gsmg/scratch/test_test_async_unittest.py", line 18, in test_async_test
    mock_session.__aenter__().get().__aenter__.side_effect = asyncio.TimeoutError()
  File "/usr/lib/python3.6/unittest/mock.py", line 584, in __getattr__
    raise AttributeError(name)
AttributeError: __aenter__

----------------------------------------------------------------------
Ran 1 test in 0.004s

FAILED (errors=1)

Тем не менее, если я запускаю его без этой строки (только код, размещенный выше)Я получаю эту ошибку: 

E
======================================================================
ERROR: test_async_test (test_test_async_unittest.TestAsyncTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/lib/python3.6/unittest/mock.py", line 1179, in patched
    return func(*args, **keywargs)
  File "/ntfs/projects/gsmg/scratch/test_test_async_unittest.py", line 19, in test_async_test
    result = loop.run_until_complete(main())
  File "/usr/lib/python3.6/asyncio/base_events.py", line 473, in run_until_complete
    return future.result()
  File "/ntfs/projects/gsmg/scratch/test_async_unittest.py", line 8, in main
    async with aiohttp.ClientSession() as session:
AttributeError: __aexit__

----------------------------------------------------------------------
Ran 1 test in 0.007s

FAILED (errors=1)
...