python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix redis store close check

Browse source
This commit is contained in:
Andrey Tikhonov 2021-05-18 15:01:55 +03:00
parent d37e89f128
commit 97e33f809a
1 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/contrib/fsm_storage/test_storage.py
View file

@ -61,12 +61,19 @@ class TestStorage:
assert await store.get_data(chat='1234') == {}
@pytest.mark.parametrize(
"store", [
pytest.lazy_fixture('redis_store'),
pytest.lazy_fixture('redis_store2'),
]
)
class TestRedisStorage2:
@pytest.mark.asyncio
async def test_close_and_open_connection(self, redis_store):
await redis_store.set_data(chat='1234', data={'foo': 'bar'})
assert await redis_store.get_data(chat='1234') == {'foo': 'bar'}
pool_id = id(redis_store._redis)
await redis_store.close()
assert await redis_store.get_data(chat='1234') == {'foo': 'bar'} # new pool was opened at this point
assert id(redis_store._redis) != pool_id
async def test_close_and_open_connection(self, store):
await store.set_data(chat='1234', data={'foo': 'bar'})
assert await store.get_data(chat='1234') == {'foo': 'bar'}
pool_id = id(store._redis)
await store.close()
assert await store.get_data(chat='1234') == {
'foo': 'bar'} # new pool was opened at this point
assert id(store._redis) != pool_id