diff --git a/encrypted_fields/tests.py b/encrypted_fields/tests.py index 446dd1a..5f5450d 100644 --- a/encrypted_fields/tests.py +++ b/encrypted_fields/tests.py @@ -1,9 +1,9 @@ import re -from datetime import datetime from django.db import models, connection from django.test import TestCase +from django.utils import timezone from .fields import ( EncryptedCharField, @@ -66,7 +66,7 @@ def test_text_field_encrypted(self): self.assertEqual(fresh_model.text, plaintext) def test_datetime_field_encrypted(self): - plaintext = datetime.now() + plaintext = timezone.now() model = TestModel() model.datetime = plaintext