Handle DBRefs correctly within Q objects. Closes #55

This commit is contained in:
flosch
2010-07-25 18:22:26 +02:00
parent 13316e5380
commit 327452622e
2 changed files with 25 additions and 0 deletions

View File

@@ -1304,6 +1304,20 @@ class QTest(unittest.TestCase):
query = ['(', {'age__gte': 18}, '&&', {'name': 'test'}, ')']
self.assertEqual((q1 & q2 & q3 & q4 & q5).query, query)
def test_q_with_dbref(self):
"""Ensure Q objects handle DBRefs correctly"""
class User(Document):
pass
class Post(Document):
created_user = ReferenceField(User)
user = User.objects.create()
Post.objects.create(created_user=user)
self.assertEqual(Post.objects.filter(created_user=user).count(), 1)
self.assertEqual(Post.objects.filter(Q(created_user=user)).count(), 1)
if __name__ == '__main__':
unittest.main()