|
|
|
@ -184,3 +184,14 @@ class CrawlTaskCeleryTests(APITestCase): |
|
|
|
self.run.refresh_from_db() |
|
|
|
self.assertEqual(self.run.status, 'FAILED') |
|
|
|
self.assertEqual(self.run.error_log, "CrawlTask is inactive.") |
|
|
|
|
|
|
|
@patch('crawler.tasks.run_crawl_pipeline.delay') |
|
|
|
def test_check_and_schedule_crawl_tasks(self, mock_delay): |
|
|
|
from crawler.tasks import check_and_schedule_crawl_tasks |
|
|
|
CrawlRun.objects.all().delete() |
|
|
|
|
|
|
|
check_and_schedule_crawl_tasks() |
|
|
|
|
|
|
|
self.assertEqual(CrawlRun.objects.count(), 1) |
|
|
|
mock_delay.assert_called_once() |
|
|
|
|