diff --git a/backend/crawler/tests.py b/backend/crawler/tests.py index 4697ed2..da9b7a4 100644 --- a/backend/crawler/tests.py +++ b/backend/crawler/tests.py @@ -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() +