13
13
from io import StringIO
14
14
15
15
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
+
16
24
def test_task (job = None ):
17
25
pass # pragma: no cover
18
26
@@ -189,7 +197,7 @@ def test_get_next_ready_job(self):
189
197
Job .objects .create (name = "testjob" , state = Job .STATES .READY )
190
198
Job .objects .create (name = "testjob" , state = Job .STATES .PROCESSING )
191
199
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 )
193
201
expected .save ()
194
202
195
203
self .assertEqual (Job .objects .get_ready_or_none ("default" ), expected )
@@ -231,7 +239,9 @@ def test_gets_jobs_in_priority_and_date_order(self):
231
239
232
240
def test_ignores_jobs_until_run_after_is_in_the_past (self ):
233
241
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
+ )
235
245
236
246
with freezegun .freeze_time (datetime (2021 , 11 , 4 , 7 )):
237
247
self .assertEqual (
@@ -256,7 +266,7 @@ def test_get_next_ready_job_created(self):
256
266
Job .objects .create (name = "testjob" , state = Job .STATES .NEW )
257
267
Job .objects .create (name = "testjob" , state = Job .STATES .PROCESSING )
258
268
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 )
260
270
expected .save ()
261
271
262
272
self .assertEqual (Job .objects .get_ready_or_none ("default" ), expected )
@@ -336,7 +346,7 @@ def test_failure_hook(self):
336
346
@override_settings (JOBS = {"testjob" : {"tasks" : ["a" ]}})
337
347
class DeleteOldJobsTestCase (TestCase ):
338
348
def test_delete_old_jobs (self ):
339
- two_days_ago = datetime . utcnow () - timedelta (days = 2 )
349
+ two_days_ago = timezone . now () - timedelta (days = 2 )
340
350
341
351
j1 = Job .objects .create (name = "testjob" , state = Job .STATES .COMPLETE )
342
352
j1 .created = two_days_ago
0 commit comments