Fix bug in application member deletion

2019-06-10 16:43:47 -04:00
parent 0bd9d4bbb6
commit 30ef2a67ef
4 changed files with 33 additions and 12 deletions
--- a/atst/domain/applications.py
+++ b/atst/domain/applications.py
@@ -128,15 +128,11 @@ class Applications(BaseDomainClass):
        return invitation

    @classmethod
-    def remove_member(cls, application, user_id):
-        application_role = ApplicationRoles.get(
-            user_id=user_id, application_id=application.id
-        )
-
+    def remove_member(cls, application_role):
        application_role.status = ApplicationRoleStatus.DISABLED
        application_role.deleted = True

-        for env in application.environments:
+        for env in application_role.application.environments:
            EnvironmentRoles.delete(
                application_role_id=application_role.id, environment_id=env.id
            )