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=
'Произошла ошибка при запросе каталога, попробуйте запросить его позже'
)