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

Add asserts to test_delta_with_dbref_*, instead of relying on exceptions

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

18
tests/document/delta.py
View File

@ -691,15 +691,25 @@ class DeltaTest(unittest.TestCase):
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()
self.assertEqual(organization._get_changed_fields(), ['employees.0.name'])
updates, removals = organization._delta()
self.assertEqual({}, removals)
self.assertIn('employees.0', updates)
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()
self.assertEqual(organization._get_changed_fields(), ['employees.0.name'])
updates, removals = organization._delta()
self.assertEqual({}, removals)
self.assertIn('employees.0', updates)
organization.save()
if __name__ == '__main__':