Skip to content

Commit 67a1755

Browse files
IvanAnishchukjleclanche
authored andcommitted
Fix timezone warnings in models tests (#748)
Use timezone-aware datetimes.
1 parent 6bd1f8f commit 67a1755

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

tests/test_models.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
from datetime import datetime as dt
2-
31
import pytest
42
from django.contrib.auth import get_user_model
53
from django.core.exceptions import ImproperlyConfigured, ValidationError
@@ -305,23 +303,23 @@ def setUp(self):
305303
AccessToken.objects.create(
306304
id=1,
307305
token="555",
308-
expires=dt.now(),
306+
expires=timezone.now(),
309307
scope=2,
310308
application_id=3,
311309
user_id=1,
312-
created=dt.now(),
313-
updated=dt.now(),
310+
created=timezone.now(),
311+
updated=timezone.now(),
314312
source_refresh_token_id="0",
315313
)
316314
AccessToken.objects.create(
317315
id=2,
318316
token="666",
319-
expires=dt.now(),
317+
expires=timezone.now(),
320318
scope=2,
321319
application_id=3,
322320
user_id=1,
323-
created=dt.now(),
324-
updated=dt.now(),
321+
created=timezone.now(),
322+
updated=timezone.now(),
325323
source_refresh_token_id="1",
326324
)
327325

@@ -340,9 +338,9 @@ def test_clear_expired_tokens_with_tokens(self):
340338
self.client.login(username="test_user", password="123456")
341339
oauth2_settings.REFRESH_TOKEN_EXPIRE_SECONDS = 0
342340
ttokens = AccessToken.objects.count()
343-
expiredt = AccessToken.objects.filter(expires__lte=dt.now()).count()
341+
expiredt = AccessToken.objects.filter(expires__lte=timezone.now()).count()
344342
assert ttokens == 2
345343
assert expiredt == 2
346344
clear_expired()
347-
expiredt = AccessToken.objects.filter(expires__lte=dt.now()).count()
345+
expiredt = AccessToken.objects.filter(expires__lte=timezone.now()).count()
348346
assert expiredt == 0

0 commit comments

Comments
 (0)