1313from io import StringIO
1414
1515
16+ try :
17+ utc = timezone .utc
18+ except AttributeError :
19+ from datetime import timezone as datetime_timezone
20+
21+ utc = datetime_timezone .utc
22+
23+
1624def test_task (job = None ):
1725 pass # pragma: no cover
1826
@@ -189,7 +197,7 @@ def test_get_next_ready_job(self):
189197 Job .objects .create (name = "testjob" , state = Job .STATES .READY )
190198 Job .objects .create (name = "testjob" , state = Job .STATES .PROCESSING )
191199 expected = Job .objects .create (name = "testjob" , state = Job .STATES .READY )
192- expected .created = datetime .now () - timedelta (minutes = 1 )
200+ expected .created = timezone .now () - timedelta (minutes = 1 )
193201 expected .save ()
194202
195203 self .assertEqual (Job .objects .get_ready_or_none ("default" ), expected )
@@ -231,7 +239,9 @@ def test_gets_jobs_in_priority_and_date_order(self):
231239
232240 def test_ignores_jobs_until_run_after_is_in_the_past (self ):
233241 job_1 = Job .objects .create (name = "testjob" )
234- job_2 = Job .objects .create (name = "testjob" , run_after = datetime (2021 , 11 , 4 , 8 ))
242+ job_2 = Job .objects .create (
243+ name = "testjob" , run_after = datetime (2021 , 11 , 4 , 8 , tzinfo = utc )
244+ )
235245
236246 with freezegun .freeze_time (datetime (2021 , 11 , 4 , 7 )):
237247 self .assertEqual (
@@ -256,7 +266,7 @@ def test_get_next_ready_job_created(self):
256266 Job .objects .create (name = "testjob" , state = Job .STATES .NEW )
257267 Job .objects .create (name = "testjob" , state = Job .STATES .PROCESSING )
258268 expected = Job .objects .create (name = "testjob" , state = Job .STATES .NEW )
259- expected .created = datetime .now () - timedelta (minutes = 1 )
269+ expected .created = timezone .now () - timedelta (minutes = 1 )
260270 expected .save ()
261271
262272 self .assertEqual (Job .objects .get_ready_or_none ("default" ), expected )
@@ -336,7 +346,7 @@ def test_failure_hook(self):
336346@override_settings (JOBS = {"testjob" : {"tasks" : ["a" ]}})
337347class DeleteOldJobsTestCase (TestCase ):
338348 def test_delete_old_jobs (self ):
339- two_days_ago = datetime . utcnow () - timedelta (days = 2 )
349+ two_days_ago = timezone . now () - timedelta (days = 2 )
340350
341351 j1 = Job .objects .create (name = "testjob" , state = Job .STATES .COMPLETE )
342352 j1 .created = two_days_ago
0 commit comments