From 97475ee64f5480cfbb0dc612672c3b6227bf8c7d Mon Sep 17 00:00:00 2001
From: richard-dds <richard-ctr@friends.dds.mil>
Date: Thu, 1 Nov 2018 10:12:58 -0400
Subject: [PATCH] WIP tests

---
 atst/models/workspace_role.py       |  1 +
 tests/models/test_workspace_user.py | 12 ++++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/atst/models/workspace_role.py b/atst/models/workspace_role.py
index d7ae975d..30b75837 100644
--- a/atst/models/workspace_role.py
+++ b/atst/models/workspace_role.py
@@ -38,6 +38,7 @@ class WorkspaceRole(Base, mixins.TimestampsMixin, mixins.AuditableMixin):
 
     @property
     def display_status(self):
+        import ipdb; ipdb.set_trace()
         if self.status == Status.ACTIVE:
             return "Active"
         else:
diff --git a/tests/models/test_workspace_user.py b/tests/models/test_workspace_user.py
index da1f08d0..b6256ed5 100644
--- a/tests/models/test_workspace_user.py
+++ b/tests/models/test_workspace_user.py
@@ -1,8 +1,9 @@
 from atst.domain.environments import Environments
 from atst.domain.workspaces import Workspaces
 from atst.domain.projects import Projects
-from atst.models.workspace_user import WorkspaceUser
-from tests.factories import RequestFactory, UserFactory
+from atst.domain.workspace_users import WorkspaceUsers
+from atst.models.invitation import Status
+from tests.factories import RequestFactory, UserFactory, InvitationFactory, WorkspaceRoleFactory
 
 
 def test_has_no_environment_roles():
@@ -54,3 +55,10 @@ def test_role_displayname():
     workspace_user = Workspaces.create_member(owner, workspace, developer_data)
 
     assert workspace_user.role_displayname == "Developer"
+
+
+def test_status_when_member_has_pending_invitation():
+    workspace_role = WorkspaceRoleFactory.create(
+        invitations=[InvitationFactory.create(status=Status.ACCEPTED)]
+    )
+    assert workspace_role.display_status == "Accepted"