Merge sequence field changes

This commit is contained in:
Ross Lawley 2012-12-19 11:39:19 +00:00
parent 3425264077
commit 3e8f02c64b
2 changed files with 1 additions and 3 deletions

View File

@ -1368,8 +1368,7 @@ class SequenceField(BaseField):
"""
Generate and Increment the counter
"""
sequence_name = (self.sequence_name or
self.owner_document._get_collection_name())
sequence_name = self.get_sequence_name()
sequence_id = "%s.%s" % (sequence_name, self.name)
collection = get_db(alias=self.db_alias)[self.collection_name]
counter = collection.find_and_modify(query={"_id": sequence_id},

View File

@ -2241,7 +2241,6 @@ class FieldTest(unittest.TestCase):
Post(title="MongoEngine",
comments=[Comment(content="NoSQL Rocks"),
Comment(content="MongoEngine Rocks")]).save()
import ipdb; ipdb.set_trace();
c = self.db['mongoengine.counters'].find_one({'_id': 'comment.id'})
self.assertEqual(c['next'], 2)
post = Post.objects.first()