jyapayne/mongoengine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added get_next_value to SequenceField

Browse source
This commit is contained in:
Wilson Júnior 2013-05-16 12:50:34 -03:00
commit 731d8fc6be
2 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/fields/fields.py
View file

@ -1917,6 +1917,23 @@ class FieldTest(unittest.TestCase):
c = self.db['mongoengine.counters'].find_one({'_id': 'person.id'})
self.assertEqual(c['next'], 1000)
def test_sequence_field_get_next_value(self):
class Person(Document):
id = SequenceField(primary_key=True)
name = StringField()
self.db['mongoengine.counters'].drop()
Person.drop_collection()
for x in xrange(10):
Person(name="Person %s" % x).save()
self.assertEqual(Person.id.get_next_value(), 10)
self.db['mongoengine.counters'].drop()
self.assertEqual(Person.id.get_next_value(), 1)
def test_sequence_field_sequence_name(self):
class Person(Document):
id = SequenceField(primary_key=True, sequence_name='jelly')