💩 First iteration

This commit is contained in:
Boger 2020-03-18 17:04:11 +03:00
parent 0bd7fc2c7e
commit fac69e52b7
25 changed files with 427 additions and 273 deletions

View file

@ -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()