From 02f61c323df31e6966e93ebd18c64ad9da7650ae Mon Sep 17 00:00:00 2001 From: emilecaron Date: Thu, 25 Jun 2015 18:26:52 +0000 Subject: [PATCH] update test --- tests/queryset/queryset.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py index 1e279744..050b2d03 100644 --- a/tests/queryset/queryset.py +++ b/tests/queryset/queryset.py @@ -1417,15 +1417,18 @@ class QuerySetTest(unittest.TestCase): """Ensure reference cascading doesn't loop if reference graph isn't a tree """ - class Category(Document): + class Dummy(Document): reference = ReferenceField('self', reverse_delete_rule=CASCADE) - base = Category().save() - other = Category(reference=base).save() + base = Dummy().save() + other = Dummy(reference=base).save() base.reference = other base.save() - self.assertEqual(2, base.delete()) + base.delete() + + self.assertRaises(DoesNotExist, base.reload) + self.assertRaises(DoesNotExist, other.reload) def test_reverse_delete_rule_cascade_self_referencing(self): """Ensure self-referencing CASCADE deletes do not result in infinite