Skip to content

Commit da7d646

Browse files
author
erdenezul
authored
Merge pull request #1571 from erdenezul/reverse_delete_rule_with_pull
add test case for reverse_delete_rule with pull #1519
2 parents d19c6a1 + 71c3c63 commit da7d646

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

tests/document/instance.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1884,6 +1884,25 @@ class BlogPost(Document):
18841884
author.delete()
18851885
self.assertEqual(BlogPost.objects.count(), 0)
18861886

1887+
def test_reverse_delete_rule_pull(self):
1888+
"""Ensure that a referenced document is also deleted with
1889+
pull.
1890+
"""
1891+
class Record(Document):
1892+
name = StringField()
1893+
children = ListField(ReferenceField('self', reverse_delete_rule=PULL))
1894+
1895+
Record.drop_collection()
1896+
1897+
parent_record = Record(name='parent').save()
1898+
child_record = Record(name='child').save()
1899+
parent_record.children.append(child_record)
1900+
parent_record.save()
1901+
1902+
child_record.delete()
1903+
self.assertEqual(Record.objects(name='parent').get().children, [])
1904+
1905+
18871906
def test_reverse_delete_rule_with_custom_id_field(self):
18881907
"""Ensure that a referenced document with custom primary key
18891908
is also deleted upon deletion.

0 commit comments

Comments
 (0)