mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-06 07:50:32 +00:00
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
import pytest
|
|
|
|
from aiogram.utils.mixins import ContextInstanceMixin, DataMixin
|
|
|
|
|
|
class ContextObject(ContextInstanceMixin["ContextObject"]):
|
|
pass
|
|
|
|
|
|
class DataObject(DataMixin):
|
|
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)
|