Browse Source

test(crawler): add unit tests for periodic task scheduling

master
PouyaKhajavi 11 hours ago
parent
commit
bef4b50d6b
  1. 11
      backend/crawler/tests.py

11
backend/crawler/tests.py

@ -184,3 +184,14 @@ class CrawlTaskCeleryTests(APITestCase):
self.run.refresh_from_db() self.run.refresh_from_db()
self.assertEqual(self.run.status, 'FAILED') self.assertEqual(self.run.status, 'FAILED')
self.assertEqual(self.run.error_log, "CrawlTask is inactive.") 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()
Loading…
Cancel
Save