Jak wyśmiać instrukcje "asynchroniczne z"?

Question

Próbuję napisać testy dla metody, która używa instrukcji "asynchronicznych z" (w tym przypadku puli połączeń aioredis), chcę kpić z połączenia z redis, ale mam problem z ustaleniem, w jaki sposób.

Oto co mam do tej pory:

from asyncio import Future 
from unittest.mock import MagicMock 

import pytest 

# The thing i'm trying to test 
async def set_value(redis, value): 
    # Do things 
    async with redis.get() as conn: 
     await conn.set("key", value) 

#My Mock classes 
class MockRedis(): 
    def get(self): 
     return MockAsyncPool() 


class MockAsyncPool(MagicMock): 
    async def __aenter__(self): 
     conn = MagicMock() 
     f = Future() 
     f.set_result(True) 
     conn.set = MagicMock(return_value=f) 
     return conn 

    def __aexit__(self, exc_type, exc_val, exc_tb): 
     pass 


# The actual test 
@pytest.mark.asyncio 
async def test_get_token(): 
    redis = MockRedis() 

    token = await set_value(redis, 'something') 
    assert token is not None 

uruchomić go z:

py.test path/to/file.py 

a ja otrzymuję ten błąd:

> await conn.set("key", value)

E TypeError: object NoneType can't be used in 'await' expression

Answer 1

__aexit__ musi być również asyncronous (needs to return an awaitable):

async def __aexit__(self, exc_type, exc_val, exc_tb): 
    pass 

Bez asynchroniczny niej wraca None zamiast współprogram więc podnosi błąd, co do bardzo mylących błędzie Stworzyłem this issue podkreślić, że musi się komunikat o błędzie, który zostanie ustalony.