use real atat and workspace roles in tests

2019-01-10 11:23:11 -05:00
parent 66ada94dcc
commit e2e4b5ae70
4 changed files with 16 additions and 18 deletions
--- a/tests/domain/test_workspace_roles.py
+++ b/tests/domain/test_workspace_roles.py
@@ -55,5 +55,11 @@ def test_workspace_role_permissions():
        status=WorkspaceRoleStatus.PENDING,
    )

-    assert WorkspaceRoles.workspace_role_permissions(workspace_one, new_user)
-    assert not WorkspaceRoles.workspace_role_permissions(workspace_two, new_user)
+    default_perms = set(new_user.atat_role.permissions)
+    assert len(
+        WorkspaceRoles.workspace_role_permissions(workspace_one, new_user)
+    ) > len(default_perms)
+    assert (
+        WorkspaceRoles.workspace_role_permissions(workspace_two, new_user)
+        == default_perms
+    )