Added testcase for ListFields with just pks (#361)

This commit is contained in:
Ross Lawley 2013-06-06 17:19:19 +00:00
parent 06f5dc6ad7
commit 9f3394dc6d

View File

@ -9,6 +9,7 @@ import unittest
import uuid import uuid
from datetime import datetime from datetime import datetime
from bson import DBRef
from tests.fixtures import PickleEmbedded, PickleTest, PickleSignalsTest from tests.fixtures import PickleEmbedded, PickleTest, PickleSignalsTest
from mongoengine import * from mongoengine import *
@ -1107,11 +1108,16 @@ class InstanceTest(unittest.TestCase):
with query_counter() as q: with query_counter() as q:
self.assertEqual(q, 0) self.assertEqual(q, 0)
sub = UserSubscription(user=u1.pk, feed=f1.pk) sub = UserSubscription(user=u1.pk, feed=f1.pk)
sub.validate() self.assertEqual(q, 0)
self.assertEqual(q, 0) # Check no change
sub.save() sub.save()
self.assertEqual(q, 1) self.assertEqual(q, 1)
# Saving with just the refs on a ListField
with query_counter() as q:
self.assertEqual(q, 0)
User(name="Bob", orgs=[o1.pk, o2.pk]).save()
self.assertEqual(q, 1)
# Saving new objects # Saving new objects
with query_counter() as q: with query_counter() as q:
self.assertEqual(q, 0) self.assertEqual(q, 0)