From d37e89f128e437a9173848d85f94e1189b733c07 Mon Sep 17 00:00:00 2001 From: Andrey Tikhonov Date: Tue, 18 May 2021 14:59:00 +0300 Subject: [PATCH] check reset --- tests/contrib/fsm_storage/test_storage.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/contrib/fsm_storage/test_storage.py b/tests/contrib/fsm_storage/test_storage.py index 3935e422..4ae7b269 100644 --- a/tests/contrib/fsm_storage/test_storage.py +++ b/tests/contrib/fsm_storage/test_storage.py @@ -51,6 +51,12 @@ class TestStorage: @pytest.mark.asyncio async def test_reset(self, store): + await store.set_data(chat='1234', data={'foo': 'bar'}) + await store.reset_data(chat='1234') + assert await store.get_data(chat='1234') == {} + + @pytest.mark.asyncio + async def test_reset_empty(self, store): await store.reset_data(chat='1234') assert await store.get_data(chat='1234') == {}