Fixed as_pymongo to return the id (#386)

2013-07-10 13:38:53 +00:00 · 2013-07-10 13:38:53 +00:00 · f34e8a0ff6
commit f34e8a0ff6
parent 4209d61b13
3 changed files with 6 additions and 1 deletions
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@ -3227,6 +3227,9 @@ class QuerySetTest(unittest.TestCase):
        User(name="Bob Dole", age=89, price=Decimal('1.11')).save()
        User(name="Barack Obama", age=51, price=Decimal('2.22')).save()

+        results = User.objects.only('id', 'name').as_pymongo()
+        self.assertEqual(results[0].keys(), ['_id', 'name'])
+
        users = User.objects.only('name', 'price').as_pymongo()
        results = list(users)
        self.assertTrue(isinstance(results[0], dict))