pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add test on datetime field - parse datetime as str

Browse Source
This commit is contained in:
Bastien Gérard 2019-03-02 23:17:24 +01:00
parent 5bf1dd55b1
commit 7ed5829b2c
1 changed files with 44 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
tests/fields/test_datetime_field.py
View File

@ -1,5 +1,5 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import datetime import datetime as dt
import six import six
try: try:
@ -41,13 +41,13 @@ class TestDateTimeField(MongoDBTestCase):
a document. a document.
""" """
class Person(Document): class Person(Document):
created = DateTimeField(default=datetime.datetime.utcnow) created = DateTimeField(default=dt.datetime.utcnow)
utcnow = datetime.datetime.utcnow() utcnow = dt.datetime.utcnow()
person = Person() person = Person()
person.validate() person.validate()
person_created_t0 = person.created person_created_t0 = person.created
self.assertLess(person.created - utcnow, datetime.timedelta(seconds=1)) self.assertLess(person.created - utcnow, dt.timedelta(seconds=1))
self.assertEqual(person_created_t0, person.created) # make sure it does not change self.assertEqual(person_created_t0, person.created) # make sure it does not change
self.assertEqual(person._data['created'], person.created) self.assertEqual(person._data['created'], person.created)
@ -65,15 +65,15 @@ class TestDateTimeField(MongoDBTestCase):
# Test can save dates # Test can save dates
log = LogEntry() log = LogEntry()
log.date = datetime.date.today() log.date = dt.date.today()
log.save() log.save()
log.reload() log.reload()
self.assertEqual(log.date.date(), datetime.date.today()) self.assertEqual(log.date.date(), dt.date.today())
# Post UTC - microseconds are rounded (down) nearest millisecond and # Post UTC - microseconds are rounded (down) nearest millisecond and
# dropped # dropped
d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 999) d1 = dt.datetime(1970, 1, 1, 0, 0, 1, 999)
d2 = datetime.datetime(1970, 1, 1, 0, 0, 1) d2 = dt.datetime(1970, 1, 1, 0, 0, 1)
log = LogEntry() log = LogEntry()
log.date = d1 log.date = d1
log.save() log.save()
@ -82,8 +82,8 @@ class TestDateTimeField(MongoDBTestCase):
self.assertEqual(log.date, d2) self.assertEqual(log.date, d2)
# Post UTC - microseconds are rounded (down) nearest millisecond # Post UTC - microseconds are rounded (down) nearest millisecond
d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 9999) d1 = dt.datetime(1970, 1, 1, 0, 0, 1, 9999)
d2 = datetime.datetime(1970, 1, 1, 0, 0, 1, 9000) d2 = dt.datetime(1970, 1, 1, 0, 0, 1, 9000)
log.date = d1 log.date = d1
log.save() log.save()
log.reload() log.reload()
@ -93,8 +93,8 @@ class TestDateTimeField(MongoDBTestCase):
if not six.PY3: if not six.PY3:
# Pre UTC dates microseconds below 1000 are dropped # Pre UTC dates microseconds below 1000 are dropped
# This does not seem to be true in PY3 # This does not seem to be true in PY3
d1 = datetime.datetime(1969, 12, 31, 23, 59, 59, 999) d1 = dt.datetime(1969, 12, 31, 23, 59, 59, 999)
d2 = datetime.datetime(1969, 12, 31, 23, 59, 59) d2 = dt.datetime(1969, 12, 31, 23, 59, 59)
log.date = d1 log.date = d1
log.save() log.save()
log.reload() log.reload()
@ -108,7 +108,7 @@ class TestDateTimeField(MongoDBTestCase):
LogEntry.drop_collection() LogEntry.drop_collection()
d1 = datetime.datetime(1970, 1, 1, 0, 0, 1) d1 = dt.datetime(1970, 1, 1, 0, 0, 1)
log = LogEntry() log = LogEntry()
log.date = d1 log.date = d1
log.validate() log.validate()
@ -124,7 +124,7 @@ class TestDateTimeField(MongoDBTestCase):
# create additional 19 log entries for a total of 20 # create additional 19 log entries for a total of 20
for i in range(1971, 1990): for i in range(1971, 1990):
d = datetime.datetime(i, 1, 1, 0, 0, 1) d = dt.datetime(i, 1, 1, 0, 0, 1)
LogEntry(date=d).save() LogEntry(date=d).save()
self.assertEqual(LogEntry.objects.count(), 20) self.assertEqual(LogEntry.objects.count(), 20)
@ -143,15 +143,15 @@ class TestDateTimeField(MongoDBTestCase):
i += 1 i += 1
# Test searching # Test searching
logs = LogEntry.objects.filter(date__gte=datetime.datetime(1980, 1, 1)) logs = LogEntry.objects.filter(date__gte=dt.datetime(1980, 1, 1))
self.assertEqual(logs.count(), 10) self.assertEqual(logs.count(), 10)
logs = LogEntry.objects.filter(date__lte=datetime.datetime(1980, 1, 1)) logs = LogEntry.objects.filter(date__lte=dt.datetime(1980, 1, 1))
self.assertEqual(logs.count(), 10) self.assertEqual(logs.count(), 10)
logs = LogEntry.objects.filter( logs = LogEntry.objects.filter(
date__lte=datetime.datetime(1980, 1, 1), date__lte=dt.datetime(1980, 1, 1),
date__gte=datetime.datetime(1975, 1, 1), date__gte=dt.datetime(1975, 1, 1),
) )
self.assertEqual(logs.count(), 5) self.assertEqual(logs.count(), 5)
@ -163,20 +163,20 @@ class TestDateTimeField(MongoDBTestCase):
time = DateTimeField() time = DateTimeField()
log = LogEntry() log = LogEntry()
log.time = datetime.datetime.now() log.time = dt.datetime.now()
log.validate() log.validate()
log.time = datetime.date.today() log.time = dt.date.today()
log.validate() log.validate()
log.time = datetime.datetime.now().isoformat(' ') log.time = dt.datetime.now().isoformat(' ')
log.validate() log.validate()
log.time = '2019-05-16 21:42:57.897847' log.time = '2019-05-16 21:42:57.897847'
log.validate() log.validate()
if dateutil: if dateutil:
log.time = datetime.datetime.now().isoformat('T') log.time = dt.datetime.now().isoformat('T')
log.validate() log.validate()
log.time = -1 log.time = -1
@ -190,6 +190,26 @@ class TestDateTimeField(MongoDBTestCase):
log.time = '2019-05-16 21:42:57.123.456' log.time = '2019-05-16 21:42:57.123.456'
self.assertRaises(ValidationError, log.validate) self.assertRaises(ValidationError, log.validate)
def test_parse_valid_datetime_str(self):
class DTDoc(Document):
date = DateTimeField()
# make sure that passing a parsable datetime works
dtd = DTDoc()
now = dt.datetime.utcnow()
dtd.date = str(now)
self.assertIsInstance(dtd.date, six.string_types)
dtd.save()
dtd.reload()
self.assertIsInstance(dtd.date, dt.datetime)
self.assertNotEqual(dtd.date, now) # microseconds differ as its not stored in mongo
self.assertEqual(
dtd.date.strftime('%Y-%m-%d %H:%M:%S'),
now.strftime('%Y-%m-%d %H:%M:%S')
)
class TestDateTimeTzAware(MongoDBTestCase): class TestDateTimeTzAware(MongoDBTestCase):
def test_datetime_tz_aware_mark_as_changed(self): def test_datetime_tz_aware_mark_as_changed(self):
@ -205,8 +225,8 @@ class TestDateTimeTzAware(MongoDBTestCase):
LogEntry.drop_collection() LogEntry.drop_collection()
LogEntry(time=datetime.datetime(2013, 1, 1, 0, 0, 0)).save() LogEntry(time=dt.datetime(2013, 1, 1, 0, 0, 0)).save()
log = LogEntry.objects.first() log = LogEntry.objects.first()
log.time = datetime.datetime(2013, 1, 1, 0, 0, 0) log.time = dt.datetime(2013, 1, 1, 0, 0, 0)
self.assertEqual(['time'], log._changed_fields) self.assertEqual(['time'], log._changed_fields)