From a0bccc64a95d690acd3b3fd61f334fbe271a9ca2 Mon Sep 17 00:00:00 2001
From: graham-dds <graham.beckley-ctr@friends.dds.mil>
Date: Mon, 28 Oct 2019 13:14:39 -0400
Subject: [PATCH] Disable PortfolioRole when revoking an invite

---
 atst/domain/invitations.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/atst/domain/invitations.py b/atst/domain/invitations.py
index 663d2201..4b89917a 100644
--- a/atst/domain/invitations.py
+++ b/atst/domain/invitations.py
@@ -138,6 +138,12 @@ class PortfolioInvitations(BaseInvitations):
     model = PortfolioInvitation
     role_domain_class = PortfolioRoles
 
+    @classmethod
+    def revoke(cls, token):
+        invite = super().revoke(token)
+        PortfolioRoles.disable(invite.role)
+        return invite
+
 
 class ApplicationInvitations(BaseInvitations):
     model = ApplicationInvitation