💩 First iteration

tests/test_api/test_client/test_session/test_base_session.py

@@ -6,7 +6,6 @@ import pytest
 from aiogram.api.client.session.base import BaseSession, T
 from aiogram.api.client.telegram import PRODUCTION, TelegramAPIServer
 from aiogram.api.methods import GetMe, Response, TelegramMethod
-from aiogram.utils.mixins import DataMixin
 
 try:
     from asynctest import CoroutineMock, patch
@@ -31,7 +30,7 @@ class CustomSession(BaseSession):
         yield b"\f" * 10
 
 
-class TestBaseSession(DataMixin):
+class TestBaseSession():
     def test_init_api(self):
         session = CustomSession()
         assert session.api == PRODUCTION

tests/test_utils/test_mixins.py

@@ -1,42 +1,11 @@
 import pytest
 
-from aiogram.utils.mixins import ContextInstanceMixin, DataMixin
+from aiogram.utils.mixins import ContextInstanceMixin
 
-
-class DataObject(DataMixin):
+class ContextObject(ContextInstanceMixin['ContextObject']):
     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()