Test Case for MongoEngine/mongoengine#83

test.py

import unittest

from mongoengine import *
from mongoengine.tests import query_counter


class Test(unittest.TestCase):

    def setUp(self):
        conn = connect(db='mongoenginetest')

    def test_cascade_dereferencing_complexfields(self):

        class Foo(Document):
            name = StringField()
            bars = ListField(ReferenceField('Bar'))

        class Bar(Document):
            name = StringField()
            quux = IntField()

        Foo.drop_collection()
        Bar.drop_collection()

        my_bars = [Bar(name=(u'bar no. %d' % i), quux=i) for i in range(10)]
        for bar in my_bars:
            bar.save()

        foo = Foo(name=u'test foo', bars=my_bars)
        foo.save()

        with query_counter() as q:
            self.assertEqual(q, 0)

            foo = Foo.objects.get(name=u'test foo')
            self.assertEqual(q, 1)

            foo.name = u'another name'
            self.assertEqual(q, 1)

            foo.save(validate=False, cascade=False)
            self.assertEqual(q, 2)

if __name__ == '__main__':
    unittest.main()