django-model-utils/tests/test_models/test_timestamped_model.py
2017-02-15 23:00:10 +00:00

25 lines
650 B
Python

from __future__ import unicode_literals
from datetime import datetime
from freezegun import freeze_time
from django.test import TestCase
from tests.models import TimeStamp
class TimeStampedModelTests(TestCase):
def test_created(self):
with freeze_time(datetime(2016, 1, 1)):
t1 = TimeStamp.objects.create()
self.assertEqual(t1.created, datetime(2016, 1, 1))
def test_modified(self):
with freeze_time(datetime(2016, 1, 1)):
t1 = TimeStamp.objects.create()
with freeze_time(datetime(2016, 1, 2)):
t1.save()
self.assertEqual(t1.modified, datetime(2016, 1, 2))