tests for db_field

This commit is contained in:
Wilson Júnior 2011-12-13 11:56:35 -02:00
parent ca7b2371fb
commit 7c1afd0031

View File

@ -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()