add new worker test

This commit is contained in:
Nick Sweeting 2024-12-12 21:41:34 -08:00
parent 5c06b8ff00
commit c11a1b54f1
No known key found for this signature in database
2 changed files with 26 additions and 22 deletions

View file

@ -1,3 +1,25 @@
from django.test import TestCase
# Create your tests here.
class CrawlActorTest(TestCase):
def test_crawl_creation(self):
seed = Seed.objects.create(uri='https://example.com')
Event.dispatch('CRAWL_CREATE', {'seed_id': seed.id})
crawl_actor = CrawlActor()
output_events = list(crawl_actor.process_next_event())
assert len(output_events) == 1
assert output_events[0].get('name', 'unset') == 'FS_WRITE'
assert output_events[0].get('path') == '/tmp/test_crawl/index.json'
output_events = list(crawl_actor.process_next_event())
assert len(output_events) == 1
assert output_events[0].get('name', 'unset') == 'CRAWL_CREATED'
assert Crawl.objects.filter(seed_id=seed.id).exists(), 'Crawl was not created'