mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-12 10:11:52 +00:00
a7b92bb050
* PoC: Mount objects to the Bot instance, bind shortcuts to configured instance * Fixe docstring of the bind method * Pass Bot instance explicitly to the URLInputFile * Added tests * Added changelog * Refactor aiogram client and update tests Refactored base.py to improve code readability by separating response_type operation from model_validate(). Also, adjusted the parameters in URLInputFile() within test_input_file.py for better test coverage. Updated input_file.py to streamline read method and avoid unnecessary instantiation of Bot class. Lastly, adjusted typing in methods/base.py to enhance code clarity. * Update changelog
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from aiogram.client.context_controller import BotContextController
|
|
from tests.mocked_bot import MockedBot
|
|
|
|
|
|
class MyModel(BotContextController):
|
|
id: int
|
|
|
|
|
|
class TestBotContextController:
|
|
def test_via_model_validate(self, bot: MockedBot):
|
|
my_model = MyModel.model_validate({"id": 1}, context={"bot": bot})
|
|
assert my_model.id == 1
|
|
assert my_model._bot == bot
|
|
|
|
def test_via_model_validate_none(self):
|
|
my_model = MyModel.model_validate({"id": 1}, context={})
|
|
assert my_model.id == 1
|
|
assert my_model._bot is None
|
|
|
|
def test_as(self, bot: MockedBot):
|
|
my_model = MyModel(id=1).as_(bot)
|
|
assert my_model.id == 1
|
|
assert my_model._bot == bot
|
|
|
|
def test_as_none(self):
|
|
my_model = MyModel(id=1).as_(None)
|
|
assert my_model.id == 1
|
|
assert my_model._bot is None
|
|
|
|
def test_replacement(self, bot: MockedBot):
|
|
my_model = MyModel(id=1).as_(bot)
|
|
assert my_model.id == 1
|
|
assert my_model._bot == bot
|
|
my_model = my_model.as_(None)
|
|
assert my_model.id == 1
|
|
assert my_model._bot is None
|