66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
from atst.domain.workspace_users import WorkspaceUsers
|
|
from atst.domain.users import Users
|
|
from atst.models.workspace_role import Status as WorkspaceRoleStatus
|
|
from atst.domain.roles import Roles
|
|
|
|
from tests.factories import (
|
|
WorkspaceFactory,
|
|
UserFactory,
|
|
InvitationFactory,
|
|
WorkspaceRoleFactory,
|
|
)
|
|
|
|
|
|
def test_can_create_new_workspace_user():
|
|
workspace = WorkspaceFactory.create()
|
|
new_user = UserFactory.create()
|
|
|
|
workspace_user_dicts = [{"id": new_user.id, "workspace_role": "owner"}]
|
|
workspace_users = WorkspaceUsers.add_many(workspace.id, workspace_user_dicts)
|
|
|
|
assert workspace_users[0].user.id == new_user.id
|
|
assert workspace_users[0].user.atat_role.name == new_user.atat_role.name
|
|
assert (
|
|
workspace_users[0].workspace_role.role.name
|
|
== new_user.workspace_roles[0].role.name
|
|
)
|
|
|
|
|
|
def test_can_update_existing_workspace_user():
|
|
workspace = WorkspaceFactory.create()
|
|
new_user = UserFactory.create()
|
|
|
|
WorkspaceUsers.add_many(
|
|
workspace.id, [{"id": new_user.id, "workspace_role": "owner"}]
|
|
)
|
|
workspace_users = WorkspaceUsers.add_many(
|
|
workspace.id, [{"id": new_user.id, "workspace_role": "developer"}]
|
|
)
|
|
|
|
assert workspace_users[0].user.atat_role.name == new_user.atat_role.name
|
|
assert (
|
|
workspace_users[0].workspace_role.role.name
|
|
== new_user.workspace_roles[0].role.name
|
|
)
|
|
|
|
|
|
def test_workspace_user_permissions():
|
|
workspace_one = WorkspaceFactory.create()
|
|
workspace_two = WorkspaceFactory.create()
|
|
new_user = UserFactory.create()
|
|
WorkspaceRoleFactory.create(
|
|
workspace=workspace_one,
|
|
user=new_user,
|
|
role=Roles.get("developer"),
|
|
status=WorkspaceRoleStatus.ACTIVE,
|
|
)
|
|
WorkspaceRoleFactory.create(
|
|
workspace=workspace_two,
|
|
user=new_user,
|
|
role=Roles.get("developer"),
|
|
status=WorkspaceRoleStatus.PENDING,
|
|
)
|
|
|
|
assert WorkspaceUsers.workspace_user_permissions(workspace_one, new_user)
|
|
assert not WorkspaceUsers.workspace_user_permissions(workspace_two, new_user)
|