pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Create test case for failing saves (wrong delta) with dbref=False

Browse Source
This commit is contained in:
Paul Uithol 2013-07-25 14:30:20 +02:00
parent f22737d6a4
commit dae9e662a5
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/document/delta.py
View File

@ -313,17 +313,17 @@ class DeltaTest(unittest.TestCase):
self.circular_reference_deltas_2(DynamicDocument, Document) self.circular_reference_deltas_2(DynamicDocument, Document)
self.circular_reference_deltas_2(DynamicDocument, DynamicDocument) self.circular_reference_deltas_2(DynamicDocument, DynamicDocument)
def circular_reference_deltas_2(self, DocClass1, DocClass2): def circular_reference_deltas_2(self, DocClass1, DocClass2, dbref=True):
class Person(DocClass1): class Person(DocClass1):
name = StringField() name = StringField()
owns = ListField(ReferenceField('Organization')) owns = ListField(ReferenceField('Organization', dbref=dbref))
employer = ReferenceField('Organization') employer = ReferenceField('Organization', dbref=dbref)
class Organization(DocClass2): class Organization(DocClass2):
name = StringField() name = StringField()
owner = ReferenceField('Person') owner = ReferenceField('Person', dbref=dbref)
employees = ListField(ReferenceField('Person')) employees = ListField(ReferenceField('Person', dbref=dbref))
Person.drop_collection() Person.drop_collection()
Organization.drop_collection() Organization.drop_collection()
@ -355,6 +355,8 @@ class DeltaTest(unittest.TestCase):
self.assertEqual(o.owner, p) self.assertEqual(o.owner, p)
self.assertEqual(e.employer, o) self.assertEqual(e.employer, o)
return person, organization, employee
def test_delta_db_field(self): def test_delta_db_field(self):
self.delta_db_field(Document) self.delta_db_field(Document)
self.delta_db_field(DynamicDocument) self.delta_db_field(DynamicDocument)
@ -686,6 +688,19 @@ class DeltaTest(unittest.TestCase):
self.assertEqual(doc._get_changed_fields(), ['list_field']) self.assertEqual(doc._get_changed_fields(), ['list_field'])
self.assertEqual(doc._delta(), ({}, {'list_field': 1})) self.assertEqual(doc._delta(), ({}, {'list_field': 1}))
def test_delta_with_dbref_true(self):
person, organization, employee = self.circular_reference_deltas_2(Document, Document, True)
employee.name = 'test'
changed = organization._get_changed_fields()
delta = organization._delta()
organization.save()
def test_delta_with_dbref_false(self):
person, organization, employee = self.circular_reference_deltas_2(Document, Document, False)
employee.name = 'test'
changed = organization._get_changed_fields()
delta = organization._delta()
organization.save()
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()