pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Workspaces.revoke_access

Browse Source
This commit is contained in:
richard-dds 2018-11-19 10:58:29 -05:00
parent 439f13237b
commit 3026771a38
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
atst/domain/workspaces/workspaces.py
View File

@ -138,3 +138,12 @@ class Workspaces(object):
workspace.name = new_data["name"]
WorkspacesQuery.add_and_commit(workspace)
@classmethod
def revoke_access(cls, user, workspace, target_workspace_role):
# TODO: What permission to here? Do we need a new one?
# Authorization.check_workspace_permission(
# user, workspace, Permissions.REQUEST_NEW_CSP_ROLE, "revoke workspace access"
# )
target_workspace_role.status = WorkspaceRoleStatus.DISABLED
return WorkspacesQuery.add_and_commit(target_workspace_role)

9
tests/domain/test_workspaces.py
View File

@ -11,8 +11,8 @@ from atst.models.workspace_role import Status as WorkspaceRoleStatus
from tests.factories import (
RequestFactory,
UserFactory,
InvitationFactory,
WorkspaceRoleFactory,
WorkspaceFactory,
)
@ -302,3 +302,10 @@ def test_can_create_workspaces_with_matching_names():
workspace_name = "Great Workspace"
Workspaces.create(RequestFactory.create(), name=workspace_name)
Workspaces.create(RequestFactory.create(), name=workspace_name)
def test_can_remove_workspace_access():
workspace = WorkspaceFactory.create()
workspace_role = WorkspaceRoleFactory.create(workspace=workspace)
Workspaces.revoke_access(workspace.owner, workspace, workspace_role)
assert Workspaces.for_user(workspace_role.user) == []