73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
import datetime
|
|
import json
|
|
|
|
from django.core import exceptions, serializers
|
|
from django.db import models
|
|
from django.test import TestCase
|
|
|
|
from .models import DurationModel, NullDurationModel
|
|
|
|
|
|
class TestSaveLoad(TestCase):
|
|
|
|
def test_simple_roundtrip(self):
|
|
duration = datetime.timedelta(days=123, seconds=123, microseconds=123)
|
|
DurationModel.objects.create(field=duration)
|
|
loaded = DurationModel.objects.get()
|
|
self.assertEqual(loaded.field, duration)
|
|
|
|
def test_create_empty(self):
|
|
NullDurationModel.objects.create()
|
|
loaded = NullDurationModel.objects.get()
|
|
self.assertEqual(loaded.field, None)
|
|
|
|
|
|
class TestQuerying(TestCase):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.objs = [
|
|
DurationModel.objects.create(field=datetime.timedelta(days=1)),
|
|
DurationModel.objects.create(field=datetime.timedelta(seconds=1)),
|
|
DurationModel.objects.create(field=datetime.timedelta(seconds=-1)),
|
|
]
|
|
|
|
def test_exact(self):
|
|
self.assertSequenceEqual(
|
|
DurationModel.objects.filter(field=datetime.timedelta(days=1)),
|
|
[self.objs[0]]
|
|
)
|
|
|
|
def test_gt(self):
|
|
self.assertSequenceEqual(
|
|
DurationModel.objects.filter(field__gt=datetime.timedelta(days=0)),
|
|
[self.objs[0], self.objs[1]]
|
|
)
|
|
|
|
|
|
class TestSerialization(TestCase):
|
|
test_data = '[{"fields": {"field": "1 01:00:00"}, "model": "model_fields.durationmodel", "pk": null}]'
|
|
|
|
def test_dumping(self):
|
|
instance = DurationModel(field=datetime.timedelta(days=1, hours=1))
|
|
data = serializers.serialize('json', [instance])
|
|
self.assertEqual(json.loads(data), json.loads(self.test_data))
|
|
|
|
def test_loading(self):
|
|
instance = list(serializers.deserialize('json', self.test_data))[0].object
|
|
self.assertEqual(instance.field, datetime.timedelta(days=1, hours=1))
|
|
|
|
|
|
class TestValidation(TestCase):
|
|
|
|
def test_invalid_string(self):
|
|
field = models.DurationField()
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
field.clean('not a datetime', None)
|
|
self.assertEqual(cm.exception.code, 'invalid')
|
|
self.assertEqual(
|
|
cm.exception.message % cm.exception.params,
|
|
"'not a datetime' value has an invalid format. "
|
|
"It must be in [DD] [HH:[MM:]]ss[.uuuuuu] format."
|
|
)
|