def setUp(self):
self.reserves = []
self.strings = RuReserve
self.chat = Chat()
self.chat.id = 101
self.user = User()
self.user.first_name = "Firstname"
self.user.id = 111
dp = Dispatcher()
self.data_adapter = MemoryDataAdapter()
self.state_manager = StateManager(self.data_adapter)
self.processor = ReserveProcessor(dp, self.state_manager, self.strings)
message = Message()
message.chat = self.chat
message.from_user = self.user
message.message_id = 121
message.text = "Some text"
message.answer = self.answer_mock
message.delete = self.delete_mock
message.edit_text = self.edit_text_mock
self.test_message = message
callback = CallbackQuery()
callback.bot = Bot()
callback.answer = self.callback_answer_mock
callback.message = message
callback.from_user = User()
callback.from_user.id = 101
self.test_callback_query = callback
async def answer_mock(self, text, parse_mode=None, reply_markup=None):
self.message = Message()
self.message.text = text
self.message.reply_markup = reply_markup
answer = Message()
answer.message_id = 1001
answer.from_user = self.user
answer.chat = self.chat
return answer
async def test_message_default(self):
"""Default message state"""
test_message = Message()
test_message.chat = self.chat
test_message.from_user = self.user
test_message.message_id = 123
await self.message_default(test_message)
passed, alert = self.assert_params(self.result_text, "Default message")
assert passed, alert
async def test_callback_query_reserve_book(self):
"""Reserve state main message"""
test_message = Message()
test_message.chat = self.chat
test_message.from_user = self.user
test_message.message_id = 122
test_callback_query = CallbackQuery()
test_callback_query.message = test_message
await self.callback_query_reserve_book(test_callback_query)
passed, alert = self.assert_params(self.result_text,
"Callback: reserve book")
assert passed, alert
async def test_message_reserve_main(self):
"""Reserve state main message"""
test_message = Message()
test_message.chat = self.chat
test_message.from_user = self.user
test_message.message_id = 121
test_message.text = "Message-121"
await self.message_reserve_main(test_message)
passed, alert = self.assert_params(self.result_text,
"Message-121: reserve main")
assert passed, alert
