mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
af4ceef153
This commit fixes the style issues introduced indd50a9baccording to the output from black, autopep8 and isort. Details:4e472085-2e87-4f61-b7a1-ed208506962f/
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import pytest
|
|
|
|
from aiogram.contrib.fsm_storage.redis import RedisStorage2
|
|
|
|
|
|
@pytest.fixture()
|
|
async def store(redis_options):
|
|
s = RedisStorage2(**redis_options)
|
|
try:
|
|
yield s
|
|
finally:
|
|
conn = await s.redis()
|
|
await conn.flushdb()
|
|
await s.close()
|
|
await s.wait_closed()
|
|
|
|
|
|
@pytest.mark.redis
|
|
class TestRedisStorage2:
|
|
@pytest.mark.asyncio
|
|
async def test_set_get(self, store):
|
|
if await store.get_data(chat="1234") != {}:
|
|
raise AssertionError
|
|
await store.set_data(chat="1234", data={"foo": "bar"})
|
|
if await store.get_data(chat="1234") != {"foo": "bar"}:
|
|
raise AssertionError
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_close_and_open_connection(self, store):
|
|
await store.set_data(chat="1234", data={"foo": "bar"})
|
|
if await store.get_data(chat="1234") != {"foo": "bar"}:
|
|
raise AssertionError
|
|
pool_id = id(store._redis)
|
|
await store.close()
|
|
if await store.get_data(chat="1234") != {"foo": "bar"}:
|
|
raise AssertionError
|
|
if id(store._redis) == pool_id:
|
|
raise AssertionError
|