From a0e3f382cd4169cfd50243bf3694b4f29d87883f Mon Sep 17 00:00:00 2001 From: Ross Lawley Date: Fri, 24 Aug 2012 09:20:14 +0100 Subject: [PATCH] Added testcase ref MongoEngine/mongoengine#92 --- tests/test_fields.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/test_fields.py b/tests/test_fields.py index e7936ca6..ce80bf9c 100644 --- a/tests/test_fields.py +++ b/tests/test_fields.py @@ -2066,6 +2066,27 @@ class FieldTest(unittest.TestCase): c = self.db['mongoengine.counters'].find_one({'_id': 'person.id'}) self.assertEqual(c['next'], 10) + def test_sequence_field_sequence_name(self): + class Person(Document): + id = SequenceField(primary_key=True, sequence_name='jelly') + name = StringField() + + self.db['mongoengine.counters'].drop() + Person.drop_collection() + + for x in xrange(10): + p = Person(name="Person %s" % x) + p.save() + + c = self.db['mongoengine.counters'].find_one({'_id': 'jelly.id'}) + self.assertEqual(c['next'], 10) + + ids = [i.id for i in Person.objects] + self.assertEqual(ids, range(1, 11)) + + c = self.db['mongoengine.counters'].find_one({'_id': 'jelly.id'}) + self.assertEqual(c['next'], 10) + def test_multiple_sequence_fields(self): class Person(Document): id = SequenceField(primary_key=True)