diff --git a/tests/test_fsm/storage/test_storages.py b/tests/test_fsm/storage/test_storages.py index d9874c68..1c1f87a2 100644 --- a/tests/test_fsm/storage/test_storages.py +++ b/tests/test_fsm/storage/test_storages.py @@ -39,6 +39,8 @@ class TestStorages: ): assert await storage.get_data(key=storage_key) == {} assert await storage.update_data(key=storage_key, data={"foo": "bar"}) == {"foo": "bar"} + assert await storage.update_data(key=storage_key, data={}) == {"foo": "bar"} + assert await storage.get_data(key=storage_key) == {"foo": "bar"} assert await storage.update_data(key=storage_key, data={"baz": "spam"}) == { "foo": "bar", "baz": "spam", @@ -47,3 +49,11 @@ class TestStorages: "foo": "bar", "baz": "spam", } + assert await storage.update_data(key=storage_key, data={"baz": "test"}) == { + "foo": "bar", + "baz": "test", + } + assert await storage.get_data(key=storage_key) == { + "foo": "bar", + "baz": "test", + }