aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Akarys42 <[email protected]>2019-10-25 19:16:21 +0200
committerGravatar Akarys42 <[email protected]>2019-10-25 19:16:21 +0200
commitf42c936bddc43b30a8245306c86b859185b5e913 (patch)
tree629f748280b3d3d92cb2e8a738cde7bcca964ebc
parentWrite tests for the future date validator (diff)
Change OffensiveMessage.delete_date from datetime.date to datetime.datetime
-rw-r--r--pydis_site/apps/api/models/bot/offensive_message.py4
-rw-r--r--pydis_site/apps/api/tests/test_validators.py6
2 files changed, 5 insertions, 5 deletions
diff --git a/pydis_site/apps/api/models/bot/offensive_message.py b/pydis_site/apps/api/models/bot/offensive_message.py
index 523fd482..cee5b32c 100644
--- a/pydis_site/apps/api/models/bot/offensive_message.py
+++ b/pydis_site/apps/api/models/bot/offensive_message.py
@@ -9,7 +9,7 @@ from pydis_site.apps.api.models.utils import ModelReprMixin
def future_date_validator(date: datetime.date) -> None:
"""Raise ValidationError if the date isn't a future date."""
- if date < datetime.date.today():
+ if date < datetime.datetime.utcnow():
raise ValidationError("Date must be a future date")
@@ -38,7 +38,7 @@ class OffensiveMessage(ModelReprMixin, models.Model):
),
)
)
- delete_date = models.DateField(
+ delete_date = models.DateTimeField(
help_text="The date on which the message will be auto-deleted.",
validators=(future_date_validator,)
)
diff --git a/pydis_site/apps/api/tests/test_validators.py b/pydis_site/apps/api/tests/test_validators.py
index 6e86759c..9d993839 100644
--- a/pydis_site/apps/api/tests/test_validators.py
+++ b/pydis_site/apps/api/tests/test_validators.py
@@ -1,4 +1,4 @@
-from datetime import date
+from datetime import datetime
from django.core.exceptions import ValidationError
from django.test import TestCase
@@ -252,8 +252,8 @@ class TagEmbedValidatorTests(TestCase):
class OffensiveMessageValidatorsTests(TestCase):
def test_accepts_future_date(self):
- future_date_validator(date(3000, 1, 1))
+ future_date_validator(datetime(3000, 1, 1))
def test_rejects_non_future_date(self):
with self.assertRaises(ValidationError):
- future_date_validator(date(1000, 1, 1))
+ future_date_validator(datetime(1000, 1, 1))