NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-08 08:38:16 -06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Add test cases for BackgroundJob handling

Browse Source
This commit is contained in:
Alexander Haase 2024-07-25 14:50:56 +02:00
parent b17b2050df
commit 60e4e81020
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
netbox/utilities/tests/test_jobs.py
View File

@ -6,6 +6,7 @@ from django_rq import get_queue
from ..jobs import * from ..jobs import *
from core.models import Job from core.models import Job
from core.choices import JobStatusChoices
class TestBackgroundJob(BackgroundJob): class TestBackgroundJob(BackgroundJob):
@ -44,6 +45,24 @@ class BackgroundJobTest(BackgroundJobTestCase):
self.assertEqual(NamedBackgroundJob.name, 'TestName') self.assertEqual(NamedBackgroundJob.name, 'TestName')
def test_handle(self):
job = TestBackgroundJob.enqueue(immediate=True)
self.assertEqual(job.status, JobStatusChoices.STATUS_COMPLETED)
def test_handle_errored(self):
class ErroredBackgroundJob(TestBackgroundJob):
EXP = Exception('Test error')
@classmethod
def run(cls, *args, **kwargs):
raise cls.EXP
job = ErroredBackgroundJob.enqueue(immediate=True)
self.assertEqual(job.status, JobStatusChoices.STATUS_ERRORED)
self.assertEqual(job.error, repr(ErroredBackgroundJob.EXP))
class EnqueueTest(BackgroundJobTestCase): class EnqueueTest(BackgroundJobTestCase):
""" """