jyapayne/mongoengine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added full_result kwarg to update (#380)

Browse source
This commit is contained in:
Ross Lawley 2013-06-21 10:19:40 +00:00
commit f1a1aa54d8
4 changed files with 37 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/queryset/queryset.py
View file

@ -536,6 +536,23 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(club.members['John']['gender'], "F")
self.assertEqual(club.members['John']['age'], 14)
def test_update_results(self):
self.Person.drop_collection()
result = self.Person(name="Bob", age=25).update(upsert=True, full_result=True)
self.assertIsInstance(result, dict)
self.assertTrue("upserted" in result)
self.assertFalse(result["updatedExisting"])
bob = self.Person.objects.first()
result = bob.update(set__age=30, full_result=True)
self.assertIsInstance(result, dict)
self.assertTrue(result["updatedExisting"])
self.Person(name="Bob", age=20).save()
result = self.Person.objects(name="Bob").update(set__name="bobby", multi=True)
self.assertEqual(result, 2)
def test_upsert(self):
self.Person.drop_collection()