Improve coverage in fields test
This commit is contained in:
		| @@ -208,10 +208,7 @@ class TestCachedReferenceField(MongoDBTestCase): | |||||||
|                 ('pj', "PJ") |                 ('pj', "PJ") | ||||||
|             ) |             ) | ||||||
|             name = StringField() |             name = StringField() | ||||||
|             tp = StringField( |             tp = StringField(choices=TYPES) | ||||||
|                 choices=TYPES |  | ||||||
|             ) |  | ||||||
|  |  | ||||||
|             father = CachedReferenceField('self', fields=('tp',)) |             father = CachedReferenceField('self', fields=('tp',)) | ||||||
|  |  | ||||||
|         Person.drop_collection() |         Person.drop_collection() | ||||||
| @@ -222,6 +219,9 @@ class TestCachedReferenceField(MongoDBTestCase): | |||||||
|         a2 = Person(name='Wilson Junior', tp='pf', father=a1) |         a2 = Person(name='Wilson Junior', tp='pf', father=a1) | ||||||
|         a2.save() |         a2.save() | ||||||
|  |  | ||||||
|  |         a2 = Person.objects.with_id(a2.id) | ||||||
|  |         self.assertEqual(a2.father.tp, a1.tp) | ||||||
|  |  | ||||||
|         self.assertEqual(dict(a2.to_mongo()), { |         self.assertEqual(dict(a2.to_mongo()), { | ||||||
|             "_id": a2.pk, |             "_id": a2.pk, | ||||||
|             "name": u"Wilson Junior", |             "name": u"Wilson Junior", | ||||||
| @@ -374,6 +374,9 @@ class TestCachedReferenceField(MongoDBTestCase): | |||||||
|         self.assertEqual(o.to_mongo()['animal']['tag'], 'heavy') |         self.assertEqual(o.to_mongo()['animal']['tag'], 'heavy') | ||||||
|         self.assertEqual(o.to_mongo()['animal']['owner']['t'], 'u') |         self.assertEqual(o.to_mongo()['animal']['owner']['t'], 'u') | ||||||
|  |  | ||||||
|  |         # Check to_mongo with fields | ||||||
|  |         self.assertNotIn('animal', o.to_mongo(fields=['person'])) | ||||||
|  |  | ||||||
|         # counts |         # counts | ||||||
|         Ocorrence(person="teste 2").save() |         Ocorrence(person="teste 2").save() | ||||||
|         Ocorrence(person="teste 3").save() |         Ocorrence(person="teste 3").save() | ||||||
|   | |||||||
| @@ -172,6 +172,9 @@ class TestDateTimeField(MongoDBTestCase): | |||||||
|         log.time = datetime.datetime.now().isoformat(' ') |         log.time = datetime.datetime.now().isoformat(' ') | ||||||
|         log.validate() |         log.validate() | ||||||
|  |  | ||||||
|  |         log.time = '2019-05-16 21:42:57.897847' | ||||||
|  |         log.validate() | ||||||
|  |  | ||||||
|         if dateutil: |         if dateutil: | ||||||
|             log.time = datetime.datetime.now().isoformat('T') |             log.time = datetime.datetime.now().isoformat('T') | ||||||
|             log.validate() |             log.validate() | ||||||
| @@ -180,6 +183,12 @@ class TestDateTimeField(MongoDBTestCase): | |||||||
|         self.assertRaises(ValidationError, log.validate) |         self.assertRaises(ValidationError, log.validate) | ||||||
|         log.time = 'ABC' |         log.time = 'ABC' | ||||||
|         self.assertRaises(ValidationError, log.validate) |         self.assertRaises(ValidationError, log.validate) | ||||||
|  |         log.time = '2019-05-16 21:GARBAGE:12' | ||||||
|  |         self.assertRaises(ValidationError, log.validate) | ||||||
|  |         log.time = '2019-05-16 21:42:57.GARBAGE' | ||||||
|  |         self.assertRaises(ValidationError, log.validate) | ||||||
|  |         log.time = '2019-05-16 21:42:57.123.456' | ||||||
|  |         self.assertRaises(ValidationError, log.validate) | ||||||
|  |  | ||||||
|  |  | ||||||
| class TestDateTimeTzAware(MongoDBTestCase): | class TestDateTimeTzAware(MongoDBTestCase): | ||||||
|   | |||||||
| @@ -508,6 +508,23 @@ class TestGenericLazyReferenceField(MongoDBTestCase): | |||||||
|         p = Ocurrence.objects.get() |         p = Ocurrence.objects.get() | ||||||
|         self.assertIs(p.animal, None) |         self.assertIs(p.animal, None) | ||||||
|  |  | ||||||
|  |     def test_generic_lazy_reference_accepts_string_instead_of_class(self): | ||||||
|  |         class Animal(Document): | ||||||
|  |             name = StringField() | ||||||
|  |             tag = StringField() | ||||||
|  |  | ||||||
|  |         class Ocurrence(Document): | ||||||
|  |             person = StringField() | ||||||
|  |             animal = GenericLazyReferenceField('Animal') | ||||||
|  |  | ||||||
|  |         Animal.drop_collection() | ||||||
|  |         Ocurrence.drop_collection() | ||||||
|  |  | ||||||
|  |         animal = Animal().save() | ||||||
|  |         Ocurrence(animal=animal).save() | ||||||
|  |         p = Ocurrence.objects.get() | ||||||
|  |         self.assertEqual(p.animal, animal) | ||||||
|  |  | ||||||
|     def test_generic_lazy_reference_embedded(self): |     def test_generic_lazy_reference_embedded(self): | ||||||
|         class Animal(Document): |         class Animal(Document): | ||||||
|             name = StringField() |             name = StringField() | ||||||
|   | |||||||
| @@ -39,9 +39,9 @@ class TestLongField(MongoDBTestCase): | |||||||
|  |  | ||||||
|         doc.value = -1 |         doc.value = -1 | ||||||
|         self.assertRaises(ValidationError, doc.validate) |         self.assertRaises(ValidationError, doc.validate) | ||||||
|         doc.age = 120 |         doc.value = 120 | ||||||
|         self.assertRaises(ValidationError, doc.validate) |         self.assertRaises(ValidationError, doc.validate) | ||||||
|         doc.age = 'ten' |         doc.value = 'ten' | ||||||
|         self.assertRaises(ValidationError, doc.validate) |         self.assertRaises(ValidationError, doc.validate) | ||||||
|  |  | ||||||
|     def test_long_ne_operator(self): |     def test_long_ne_operator(self): | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user