From c814fae395f3ca836e6119dc083cddbf322a55fc Mon Sep 17 00:00:00 2001
From: hinashi <k.hinagawa@gmail.com>
Date: Fri, 16 Sep 2022 11:53:36 +0900
Subject: [PATCH] Changed to separate may_remove_referral method

---
 entry/models.py           | 15 +++++++------
 entry/tests/test_model.py | 45 +++++++++++++++++++++++++++++++++------
 2 files changed, 47 insertions(+), 13 deletions(-)

diff --git a/entry/models.py b/entry/models.py
index 20f7fc5a4..155715116 100644
--- a/entry/models.py
+++ b/entry/models.py
@@ -1159,21 +1159,18 @@ def add_to_attrv(self, user, referral=None, value="", boolean=False):
             if updated_data and self.is_updated(updated_data):
                 self.add_value(user, updated_data, boolean=attrv.boolean)
 
-    # NOTE: Type-Write
-    def delete(self):
-        super(Attribute, self).delete()
-
+    def may_remove_referral(self):
         def _may_remove_referral(referral):
             if not referral:
                 # the case this refers no entry, do nothing
                 return
 
-            entry = Entry.objects.filter(id=referral.id, is_active=True).first()
+            entry: Entry = Entry.objects.filter(id=referral.id, is_active=True).first()
             if not entry:
                 # the case referred entry is already deleted, do nothing
                 return
 
-            if entry.get_referred_objects().count() > 0:
+            if entry.get_referred_objects().exclude(id=self.parent_entry.id).count() > 0:
                 # the case other entries also refer target referral, do nothing
                 return
 
@@ -1188,6 +1185,12 @@ def _may_remove_referral(referral):
             else:
                 _may_remove_referral(attrv.referral)
 
+    # NOTE: Type-Write
+    def delete(self):
+        super(Attribute, self).delete()
+
+        self.may_remove_referral()
+
     # implementation for Attribute
     def check_duplication_entry_at_restoring(self, entry_chain):
         def _check(referral):
diff --git a/entry/tests/test_model.py b/entry/tests/test_model.py
index c5a9f140d..71d115224 100644
--- a/entry/tests/test_model.py
+++ b/entry/tests/test_model.py
@@ -20,16 +20,16 @@ class ModelTest(AironeTestCase):
     def setUp(self):
         super(ModelTest, self).setUp()
 
-        self._user = User(username="test")
+        self._user: User = User(username="test")
         self._user.save()
 
-        self._entity = Entity(name="entity", created_user=self._user)
+        self._entity: Entity = Entity(name="entity", created_user=self._user)
         self._entity.save()
 
-        self._entry = Entry(name="entry", created_user=self._user, schema=self._entity)
+        self._entry: Entry = Entry(name="entry", created_user=self._user, schema=self._entity)
         self._entry.save()
 
-        self._attr = self.make_attr("attr")
+        self._attr: Attribute = self.make_attr("attr")
         self._attr.save()
 
         # clear all cache before start
@@ -38,7 +38,7 @@ def setUp(self):
         self._org_auto_complement_user = settings.AIRONE["AUTO_COMPLEMENT_USER"]
 
         # make auto complement user
-        self._complement_user = User(
+        self._complement_user: User = User(
             username=self._org_auto_complement_user,
             email="hoge@example.com",
             is_superuser=True,
@@ -93,7 +93,7 @@ def create_entity_with_all_type_attributes(self, user, ref_entity=None):
         This is a test helper method to add attributes of all attribute-types
         to specified entity.
         """
-        entity = Entity.objects.create(name="entity", created_user=user)
+        entity = Entity.objects.create(name="all_attr_entity", created_user=user)
         attr_info = {
             "str": AttrTypeValue["string"],
             "text": AttrTypeValue["text"],
@@ -856,6 +856,37 @@ def test_delete_entry_in_chain(self):
         self.assertFalse(ref_entries[1].is_active)
         self.assertTrue(ref_entries[2].is_active)
 
+    def test_may_remove_referral(self):
+        entity: Entity = self.create_entity_with_all_type_attributes(self._user, self._entity)
+        entry: Entry = Entry.objects.create(name="e1", schema=entity, created_user=self._user)
+        entry.complement_attrs(self._user)
+
+        attr_info = [
+            {"name": "obj", "val": self._entry, "del": ""},
+            {
+                "name": "name",
+                "val": {"name": "new_value", "id": self._entry},
+                "del": {"name": "", "id": ""},
+            },
+            {"name": "arr_obj", "val": [self._entry], "del": []},
+            {"name": "arr_name", "val": [{"name": "new_value", "id": self._entry}], "del": []},
+        ]
+        for info in attr_info:
+            entity_attr: EntityAttr = entity.attrs.get(name=info["name"])
+            entity_attr.is_delete_in_chain = True
+            entity_attr.save()
+
+            attr: Attribute = entry.attrs.get(schema=entity_attr)
+            attr.add_value(self._user, info["val"])
+            attr.may_remove_referral()
+
+            self._entry.refresh_from_db()
+            self.assertFalse(self._entry.is_active)
+
+            # restore to pre-test state
+            attr.add_value(self._user, info["del"])
+            self._entry.restore()
+
     def test_order_of_array_named_ref_entries(self):
         ref_entity = Entity.objects.create(name="referred_entity", created_user=self._user)
         ref_entry = Entry.objects.create(
@@ -4258,7 +4289,7 @@ def test_get_es_document_without_attribute_value(self):
         self.assertEqual(
             result["ret_values"][0],
             {
-                "entity": {"id": entity.id, "name": "entity"},
+                "entity": {"id": entity.id, "name": "all_attr_entity"},
                 "entry": {"id": entry.id, "name": "entry"},
                 "is_readble": True,
                 "attrs": {