Test to illustrate failure in changed attribute tracking
This commit is contained in:
parent
c0a5b16a7f
commit
5be5685a09
@ -735,5 +735,47 @@ class DeltaTest(unittest.TestCase):
|
|||||||
mydoc._clear_changed_fields()
|
mydoc._clear_changed_fields()
|
||||||
self.assertEqual([], mydoc._get_changed_fields())
|
self.assertEqual([], mydoc._get_changed_fields())
|
||||||
|
|
||||||
|
def test_referenced_object_changed_attributes(self):
|
||||||
|
"""Ensures that when you save a new reference to a field, the referenced object isn't altered"""
|
||||||
|
|
||||||
|
class Organization(Document):
|
||||||
|
name = StringField()
|
||||||
|
|
||||||
|
class User(Document):
|
||||||
|
name = StringField()
|
||||||
|
org = ReferenceField('Organization', required=True)
|
||||||
|
|
||||||
|
Organization.drop_collection()
|
||||||
|
User.drop_collection()
|
||||||
|
|
||||||
|
org1 = Organization(name='Org 1')
|
||||||
|
org1.save()
|
||||||
|
|
||||||
|
org2 = Organization(name='Org 2')
|
||||||
|
org2.save()
|
||||||
|
|
||||||
|
user = User(name='Fred', org=org1)
|
||||||
|
user.save()
|
||||||
|
|
||||||
|
org1.reload()
|
||||||
|
org2.reload()
|
||||||
|
user.reload()
|
||||||
|
self.assertEqual(org1.name, 'Org 1')
|
||||||
|
self.assertEqual(org2.name, 'Org 2')
|
||||||
|
self.assertEqual(user.name, 'Fred')
|
||||||
|
|
||||||
|
user.name = 'Harold'
|
||||||
|
user.org = org2
|
||||||
|
|
||||||
|
org2.name = 'New Org 2'
|
||||||
|
self.assertEqual(org2.name, 'New Org 2')
|
||||||
|
|
||||||
|
user.save()
|
||||||
|
org2.save()
|
||||||
|
|
||||||
|
self.assertEqual(org2.name, 'New Org 2')
|
||||||
|
org2.reload()
|
||||||
|
self.assertEqual(org2.name, 'New Org 2')
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user