From 329c7f4c9746e26e30ce4c38f434da29b5c93994 Mon Sep 17 00:00:00 2001
From: Rishat Fayzullin <nilluziaf@gmail.com>
Date: Wed, 6 Mar 2024 16:53:56 +0300
Subject: [PATCH] Added few additional checks in storages test

---
 tests/test_fsm/storage/test_storages.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

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",
+        }