Added chaining regression test (MongoEngine/mongoengine#135)

This commit is contained in:
Ross Lawley 2012-10-01 16:27:59 +00:00
parent 79098e997e
commit 12f3f8c694

View File

@ -230,6 +230,30 @@ class QuerySetTest(unittest.TestCase):
Blog.drop_collection() Blog.drop_collection()
def test_chaining(self):
class A(Document):
pass
class B(Document):
a = ReferenceField(A)
A.drop_collection()
B.drop_collection()
a1 = A().save()
a2 = A().save()
B(a=a1).save()
# Works
q1 = B.objects.filter(a__in=[a1, a2], a=a1)._query
# Doesn't work
q2 = B.objects.filter(a__in=[a1, a2])
q2 = q2.filter(a=a1)._query
self.assertEqual(q1, q2)
def test_update_write_options(self): def test_update_write_options(self):
"""Test that passing write_options works""" """Test that passing write_options works"""