From 7c1afd00313c5e0eaad64079efa8be127a0a4f1b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wilson=20J=C3=BAnior?= Date: Tue, 13 Dec 2011 11:56:35 -0200 Subject: [PATCH] tests for db_field --- tests/queryset.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/queryset.py b/tests/queryset.py index 02e931e4..4b619560 100644 --- a/tests/queryset.py +++ b/tests/queryset.py @@ -3027,5 +3027,24 @@ class QueryFieldListTest(unittest.TestCase): plist = list(Person.objects.select('name', 'state')) self.assertEqual(plist, [[u'Wilson JR', s1]]) + + def test_select_db_field(self): + class TestDoc(Document): + x = IntField(db_field="y") + y = BooleanField(db_field="x") + + TestDoc.drop_collection() + + TestDoc(x=10, y=True).save() + TestDoc(x=20, y=False).save() + TestDoc(x=30, y=True).save() + + plist = list(TestDoc.objects.select('x', 'y')) + + self.assertEqual(len(plist), 3) + self.assertEqual(plist[0], [10, True]) + self.assertEqual(plist[1], [20, False]) + self.assertEqual(plist[2], [30, True]) + if __name__ == '__main__': unittest.main()