def test_user_log_command(bot_user):
log_command(bot_user, '/start', 'Hi')
log_command(bot_user, '/stop', 'Bye')
all_commands = LogCommandItem.objects(user=bot_user.id)
assert all_commands.count() == 2
assert all_commands.first()['message'] == 'Hi'
def test_user_log_command(bot_user):
log_command(bot_user, '/start', 'Hi')
log_command(bot_user, '/stop', 'Bye')
all_commands = LogCommandItem.filter(user=bot_user.chat_id)
assert all_commands.count() == 2
assert all_commands.first().message == 'Hi'
def _create_telegram_catalog_logs(logs_count=1,
command='/start',
message='message',
status='success'):
for _ in range(logs_count):
cmd = log_command(bot_user, command, message)
cmd.set_status(status)
def test_schedule_category_export_with_exception(mocked_send_message,
mocked_category_export,
bot_user):
mocked_category_export.side_effect = Exception(
'Spider wb has more than SCHEDULED_JOBS_THRESHOLD queued jobs')
log_item = log_command(bot_user, 'wb_catalog', 'la-la-la')
schedule_category_export('https://www.wildberries/category/url',
bot_user.chat_id, log_item.id)
mocked_category_export.assert_called()
assert 'мы сейчас не можем обработать ваш запрос' in mocked_send_message.call_args.kwargs[
'text']
def test_schedule_category_export_correct(
mocked_send_message, mocked_check_requests_count_recovered, bot_user,
set_scrapinghub_requests_mock):
set_scrapinghub_requests_mock(job_id='123/1/1234')
log_item = log_command(bot_user, 'wb_catalog', 'la-la-la')
schedule_category_export('https://www.wildberries/category/url',
bot_user.chat_id, log_item.id)
assert 'Мы обрабатываем ваш запрос' in mocked_send_message.call_args.kwargs[
'text']
mocked_check_requests_count_recovered.assert_called()