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