add WorkspaceUser tests from authz

This commit is contained in:
dandds
2018-07-30 14:19:38 -04:00
parent 202f88bae5
commit e270350925
2 changed files with 55 additions and 6 deletions

View File

@@ -0,0 +1,43 @@
import pytest
from uuid import uuid4
from atst.domain.workspace_users import WorkspaceUsers
from atst.domain.users import Users
@pytest.fixture()
def users_repo(db):
return Users(db)
@pytest.fixture()
def workspace_users_repo(db):
return WorkspaceUsers(db)
def test_can_create_new_workspace_user(users_repo, workspace_users_repo):
workspace_id = uuid4()
user = users_repo.create(uuid4(), "developer")
workspace_user_dicts = [
{"id": user.id, "workspace_role": "owner"}
]
workspace_users = workspace_users_repo.add_many(workspace_id, workspace_user_dicts)
assert workspace_users[0].user.id == user.id
assert workspace_users[0].user.atat_role.name == "developer"
assert workspace_users[0].workspace_role.role.name == "owner"
def test_can_update_existing_workspace_user(users_repo, workspace_users_repo):
workspace_id = uuid4()
user = users_repo.create(uuid4(), "developer")
workspace_users_repo.add_many(workspace_id, [
{"id": user.id, "workspace_role": "owner"}
])
workspace_users = workspace_users_repo.add_many(workspace_id, [
{"id": user.id, "workspace_role": "developer"}
])
assert workspace_users[0].user.id == user.id
assert workspace_users[0].workspace_role.role.name == "developer"