Skip to content

Commit 71c3c63

Browse files
author
Erdenezul Batmunkh
committed
add test case for reverse_delete_rule with pull #1519
1 parent 2f1fe54 commit 71c3c63

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
@@ -1866,6 +1866,25 @@ class BlogPost(Document):
18661866
author.delete()
18671867
self.assertEqual(BlogPost.objects.count(), 0)
18681868

1869+
def test_reverse_delete_rule_pull(self):
1870+
"""Ensure that a referenced document is also deleted with
1871+
pull.
1872+
"""
1873+
class Record(Document):
1874+
name = StringField()
1875+
children = ListField(ReferenceField('self', reverse_delete_rule=PULL))
1876+
1877+
Record.drop_collection()
1878+
1879+
parent_record = Record(name='parent').save()
1880+
child_record = Record(name='child').save()
1881+
parent_record.children.append(child_record)
1882+
parent_record.save()
1883+
1884+
child_record.delete()
1885+
self.assertEqual(Record.objects(name='parent').get().children, [])
1886+
1887+
18691888
def test_reverse_delete_rule_with_custom_id_field(self):
18701889
"""Ensure that a referenced document with custom primary key
18711890
is also deleted upon deletion.

0 commit comments

Comments
 (0)