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()