mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
73 lines
1.6 KiB
Python
73 lines
1.6 KiB
Python
import pytest
|
|
|
|
from aiogram.utils.mixins import ContextInstanceMixin, DataMixin, KeepRefsMixin
|
|
|
|
|
|
class ContextObject(ContextInstanceMixin["ContextObject"]):
|
|
pass
|
|
|
|
|
|
class DataObject(DataMixin):
|
|
pass
|
|
|
|
|
|
class RefsObject(KeepRefsMixin):
|
|
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)
|
|
|
|
|
|
class TestKeepRefsMixin:
|
|
def test_refs_are_saved(self):
|
|
obj = RefsObject()
|
|
|
|
assert obj in RefsObject.get_instances()
|
|
|
|
def test_refs_are_deleted(self):
|
|
obj = RefsObject()
|
|
size_with_obj = len(RefsObject.get_instances())
|
|
del obj
|
|
size_without_obj = len(RefsObject.get_instances())
|
|
|
|
assert size_with_obj - 1 == size_without_obj
|