fixes for get_next_value

This commit is contained in:
Wilson Júnior 2013-05-16 13:05:07 -03:00
parent 0b1e11ba1f
commit f7e22d2b8b
2 changed files with 2 additions and 2 deletions

View File

@ -1449,7 +1449,7 @@ class SequenceField(BaseField):
data = collection.find_one({"_id": sequence_id})
if data:
return data['next']
return data['next']+1
return 1

View File

@ -1910,7 +1910,7 @@ class FieldTest(unittest.TestCase):
for x in xrange(10):
Person(name="Person %s" % x).save()
self.assertEqual(Person.id.get_next_value(), 10)
self.assertEqual(Person.id.get_next_value(), 11)
self.db['mongoengine.counters'].drop()
self.assertEqual(Person.id.get_next_value(), 1)