def test_schedule_category_export_correct(mocked_send_message, mocked_category_export): mocked_category_export.return_value = 'https://dummy.url/' schedule_category_export('https://www.wildberries/category/url', '1423') mocked_category_export.assert_called() mocked_send_message.assert_called_with( chat_id='1423', text= 'Я поставил каталог в очередь на исследование. Скоро пришлю результаты.' )
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()
def test_schedule_category_export_with_exception(mocked_send_message, mocked_category_export): mocked_category_export.side_effect = Exception( 'Spider wb has more than 1 queued jobs') schedule_category_export('https://www.wildberries/category/url', '1423') mocked_category_export.assert_called() mocked_send_message.assert_called_with( chat_id='1423', text= 'Произошла ошибка при запросе каталога, попробуйте запросить его позже' )