fix tests

This commit is contained in:
Dmitry Afanasyev 2023-04-08 15:07:14 +03:00
parent ecbfbdc7d8
commit 585367e621

View File

@ -56,7 +56,7 @@ class TransportBot:
'т19', 'т19',
] ]
text = await TransportBot._get_buses_data(url=url, message=message, buses=buses) text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
return await TransportBot.bot.send_message( return await TransportBot.bot.send_message(
query.message.chat.id, text, reply_markup=TransportBot.get_keyboard() query.message.chat.id, text, reply_markup=TransportBot.get_keyboard()
@ -74,7 +74,7 @@ class TransportBot:
'т19', 'т19',
] ]
text = await TransportBot._get_buses_data(url=url, message=message, buses=buses) text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
return await TransportBot.bot.send_message( return await TransportBot.bot.send_message(
query.message.chat.id, text, reply_markup=TransportBot.get_keyboard() query.message.chat.id, text, reply_markup=TransportBot.get_keyboard()
@ -103,7 +103,7 @@ class TransportBot:
'т19', 'т19',
] ]
text = await TransportBot._get_buses_data(url=url, message=message, buses=buses) text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
kwargs = {'reply_markup': TransportBot.get_keyboard()} if show_keyboard else {} kwargs = {'reply_markup': TransportBot.get_keyboard()} if show_keyboard else {}
@ -120,7 +120,7 @@ class TransportBot:
) )
@staticmethod @staticmethod
async def _get_buses_data(url: str, message: str, buses: list[str]) -> str: async def get_buses_data(url: str, message: str, buses: list[str]) -> str:
driver = WebParser.get_driver() driver = WebParser.get_driver()
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
return await loop.run_in_executor( return await loop.run_in_executor(