More tests (Helper and contextinstance)

tests/test_utils/test_mixins.py

@@ -0,0 +1,51 @@
+import pytest
+from aiogram.utils.mixins import ContextInstanceMixin, DataMixin
+
+
+class DataObject(DataMixin):
+    pass
+
+
+class ContextObject(ContextInstanceMixin):
+    pass
+
+
+class TestDataMixin:
+    def test_store_value(self):
+        obj = DataObject()
+        obj["foo"] = 42
+
+        assert "foo" in obj
+        assert obj["foo"] == 42
+        assert len(obj.data) == 1
+
+    def test_remove_value(self):
+        obj = DataObject()
+        obj["foo"] = 42
+        del obj["foo"]
+
+        assert "key" not in obj
+        assert len(obj.data) == 0
+
+    def test_getter(self):
+        obj = DataObject()
+        obj["foo"] = 42
+
+        assert obj.get("foo") == 42
+        assert obj.get("bar") is None
+        assert obj.get("baz", "test") == "test"
+
+
+class TestContextInstanceMixin:
+    def test_empty(self):
+        obj = ContextObject()
+
+        assert obj.get_current(no_error=True) is None
+        with pytest.raises(LookupError):
+            assert obj.get_current(no_error=False)
+
+    def test_set_wrong_type(self):
+        obj = ContextObject()
+
+        with pytest.raises(TypeError, match=r"Value should be instance of 'ContextObject' not '.+'"):
+            obj.set_current(42)