Convert Users, Roles, WorkspaceUsers, TaskOrders to use classmethods

This commit is contained in:
richard-dds
2018-08-03 13:12:54 -04:00
committed by dandds
parent 7b5d76e260
commit b72a16569f
8 changed files with 86 additions and 107 deletions

View File

@@ -5,37 +5,27 @@ from atst.domain.workspace_users import WorkspaceUsers
from atst.domain.users import Users
@pytest.fixture()
def users_repo(session):
return Users(session)
@pytest.fixture()
def workspace_users_repo(session):
return WorkspaceUsers(session)
def test_can_create_new_workspace_user(users_repo, workspace_users_repo):
def test_can_create_new_workspace_user():
workspace_id = uuid4()
user = users_repo.create(uuid4(), "developer")
user = Users.create(uuid4(), "developer")
workspace_user_dicts = [{"id": user.id, "workspace_role": "owner"}]
workspace_users = workspace_users_repo.add_many(workspace_id, workspace_user_dicts)
workspace_users = WorkspaceUsers.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):
def test_can_update_existing_workspace_user():
workspace_id = uuid4()
user = users_repo.create(uuid4(), "developer")
user = Users.create(uuid4(), "developer")
workspace_users_repo.add_many(
WorkspaceUsers.add_many(
workspace_id, [{"id": user.id, "workspace_role": "owner"}]
)
workspace_users = workspace_users_repo.add_many(
workspace_users = WorkspaceUsers.add_many(
workspace_id, [{"id": user.id, "workspace_role": "developer"}]
)