From 3fc3d37919d90427ee7f1df21fa7fdace52d1e44 Mon Sep 17 00:00:00 2001 From: Alexander Haase Date: Thu, 25 Jul 2024 18:01:02 +0200 Subject: [PATCH] Test enqueue in combination with enqueue_once --- netbox/utilities/tests/test_jobs.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/netbox/utilities/tests/test_jobs.py b/netbox/utilities/tests/test_jobs.py index 462f53814..b4ae7374a 100644 --- a/netbox/utilities/tests/test_jobs.py +++ b/netbox/utilities/tests/test_jobs.py @@ -111,6 +111,23 @@ class EnqueueTest(BackgroundJobTestCase): self.assertRaises(Job.DoesNotExist, job1.refresh_from_db) self.assertEqual(TestBackgroundJob.get_jobs(instance).count(), 1) + def test_enqueue_once_with_enqueue(self): + instance = Job() + job1 = TestBackgroundJob.enqueue_once(instance, schedule_at=self.get_schedule_at(2)) + job2 = TestBackgroundJob.enqueue(instance, schedule_at=self.get_schedule_at()) + + self.assertNotEqual(job1, job2) + self.assertEqual(TestBackgroundJob.get_jobs(instance).count(), 2) + + def test_enqueue_once_after_enqueue(self): + instance = Job() + job1 = TestBackgroundJob.enqueue(instance, schedule_at=self.get_schedule_at()) + job2 = TestBackgroundJob.enqueue_once(instance, schedule_at=self.get_schedule_at(2)) + + self.assertNotEqual(job1, job2) + self.assertRaises(Job.DoesNotExist, job1.refresh_from_db) + self.assertEqual(TestBackgroundJob.get_jobs(instance).count(), 1) + def test_enqueue_system(self): job = TestBackgroundJob.enqueue_once(schedule_at=self.get_schedule_at())