Moved test date check into a dedicated function

This commit is contained in:
Eliot Berriot 2019-07-29 10:44:01 +02:00
parent b3de851e6f
commit 32445c5113
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
8 changed files with 59 additions and 52 deletions

12
api/tests/utils.py Normal file
View file

@ -0,0 +1,12 @@
import datetime
from rest_framework import fields as rest_fields
def to_api_date(value):
if isinstance(value, datetime.datetime):
f = rest_fields.DateTimeField()
return f.to_representation(value)
if isinstance(value, datetime.date):
f = rest_fields.DateField()
return f.to_representation(value)
raise ValueError("Invalid value: {}".format(value))