Rehabilitate test coverage in ussd and cic-notify

2022-01-04 16:51:02 +00:00
parent fe3f2c2549
commit 837e8da650
63 changed files with 1890 additions and 498 deletions
--- a/apps/cic-ussd/cic_ussd/db/models/account.py
+++ b/apps/cic-ussd/cic_ussd/db/models/account.py
@@ -63,10 +63,7 @@ class Account(SessionBase):
    def remove_guardian(self, phone_number: str):
        set_guardians = self.guardians.split(',')
        set_guardians.remove(phone_number)
-        if len(set_guardians) > 1:
-            self.guardians = ','.join(set_guardians)
-        else:
-            self.guardians = set_guardians[0]
+        self.guardians = ','.join(set_guardians)

    def get_guardians(self) -> list:
        return self.guardians.split(',') if self.guardians else []