Fix tests that were broken by new constraints

This commit is contained in:
richard-dds
2018-09-21 13:54:46 -04:00
parent de338f675f
commit 519b5caa4c
7 changed files with 54 additions and 55 deletions

View File

@@ -1,9 +1,10 @@
from atst.domain.projects import Projects
from tests.factories import WorkspaceFactory
from atst.domain.workspaces import Workspaces
from tests.factories import RequestFactory
def test_create_project_with_multiple_environments():
workspace = WorkspaceFactory.create()
workspace = Workspaces.create(RequestFactory.create())
project = Projects.create(
workspace.owner, workspace, "My Test Project", "Test", ["dev", "prod"]
)

View File

@@ -8,13 +8,13 @@ DOD_ID = "my_dod_id"
def test_create_user():
user = Users.create("developer")
user = Users.create("developer", dod_id=DOD_ID)
assert user.atat_role.name == "developer"
def test_create_user_with_nonexistent_role():
with pytest.raises(NotFoundError):
Users.create("nonexistent")
Users.create("nonexistent", dod_id=DOD_ID)
def test_get_or_create_nonexistent_user():
@@ -29,13 +29,13 @@ def test_get_or_create_existing_user():
def test_get_user():
new_user = Users.create("developer")
new_user = Users.create("developer", dod_id=DOD_ID)
user = Users.get(new_user.id)
assert user.id == new_user.id
def test_get_nonexistent_user():
Users.create("developer")
Users.create("developer", dod_id=DOD_ID)
with pytest.raises(NotFoundError):
Users.get(uuid4())
@@ -47,19 +47,19 @@ def test_get_user_by_dod_id():
def test_update_user():
new_user = Users.create("developer")
new_user = Users.create("developer", dod_id=DOD_ID)
updated_user = Users.update(new_user.id, "ccpo")
assert updated_user.atat_role.name == "ccpo"
def test_update_nonexistent_user():
Users.create("developer")
Users.create("developer", dod_id=DOD_ID)
with pytest.raises(NotFoundError):
Users.update(uuid4(), "ccpo")
def test_update_existing_user_with_nonexistent_role():
new_user = Users.create("developer")
new_user = Users.create("developer", dod_id=DOD_ID)
with pytest.raises(NotFoundError):
Users.update(new_user.id, "nonexistent")

View File

@@ -1,23 +1,23 @@
from atst.domain.workspace_users import WorkspaceUsers
from atst.domain.users import Users
from tests.factories import WorkspaceFactory
from tests.factories import WorkspaceFactory, UserFactory
def test_can_create_new_workspace_user():
workspace = WorkspaceFactory.create()
new_user = Users.create("developer")
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 == "developer"
assert workspace_users[0].workspace_role.role.name == "owner"
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 = Users.create("developer")
new_user = UserFactory.create()
WorkspaceUsers.add_many(
workspace.id, [{"id": new_user.id, "workspace_role": "owner"}]
@@ -26,5 +26,5 @@ def test_can_update_existing_workspace_user():
workspace.id, [{"id": new_user.id, "workspace_role": "developer"}]
)
assert workspace_users[0].user.id == new_user.id
assert workspace_users[0].workspace_role.role.name == "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