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

Refactor test_update_environment_roles

Browse Source
This commit is contained in:
richard-dds 2018-10-07 11:52:28 -04:00
parent e383ee533a
commit 9061c3515c
1 changed files with 24 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
tests/domain/test_environments.py
View File

@ -1,51 +1,42 @@
import pytest
from uuid import uuid4
from atst.domain.environments import Environments from atst.domain.environments import Environments
from atst.domain.environment_roles import EnvironmentRoles from atst.domain.environment_roles import EnvironmentRoles
from atst.domain.projects import Projects
from atst.domain.roles import Roles
from atst.domain.workspaces import Workspaces
from atst.domain.workspace_users import WorkspaceUsers
from atst.domain.exceptions import NotFoundError
from atst.models.environment_role import EnvironmentRole
from tests.factories import ( from tests.factories import UserFactory, SuperWorkspaceFactory
RequestFactory,
UserFactory,
WorkspaceFactory,
EnvironmentFactory,
ProjectFactory,
SuperWorkspaceFactory,
)
def test_update_environment_roles(): def test_update_environment_roles():
owner = UserFactory.create() owner = UserFactory.create()
developer_data = { developer = UserFactory.from_atat_role("developer")
"dod_id": "1234567890",
"first_name": "Test",
"last_name": "User",
"email": "test.user@mail.com",
"workspace_role": "developer",
}
workspace = Workspaces.create(RequestFactory.create(creator=owner)) workspace = SuperWorkspaceFactory.create(
workspace_user = Workspaces.create_member(owner, workspace, developer_data) owner=owner,
project = Projects.create( members=[{"user": developer, "role_name": "developer"}],
owner, workspace, "my test project", "It's mine.", ["dev", "staging", "prod"] projects=[
{
"name": "project1",
"environments": [
{
"name": "project1 dev",
"members": [{"user": developer, "role_name": "devlops"}],
},
{
"name": "project1 staging",
"members": [{"user": developer, "role_name": "developer"}]
},
{"name": "project1 prod"},
],
}
],
) )
dev_env = project.environments[0] dev_env = workspace.projects[0].environments[0]
staging_env = project.environments[1] staging_env = workspace.projects[0].environments[1]
Environments.add_member(dev_env, workspace_user.user, "devops")
Environments.add_member(staging_env, workspace_user.user, "developer")
new_ids_and_roles = [ new_ids_and_roles = [
{"id": dev_env.id, "role": "billing_admin"}, {"id": dev_env.id, "role": "billing_admin"},
{"id": staging_env.id, "role": "developer"}, {"id": staging_env.id, "role": "developer"},
] ]
workspace_user = workspace.members[0]
Environments.update_environment_role(owner, new_ids_and_roles, workspace_user) Environments.update_environment_role(owner, new_ids_and_roles, workspace_user)
new_dev_env_role = EnvironmentRoles.get(workspace_user.user.id, dev_env.id) new_dev_env_role = EnvironmentRoles.get(workspace_user.user.id, dev_env.id)
staging_env_role = EnvironmentRoles.get(workspace_user.user.id, staging_env.id) staging_env_role = EnvironmentRoles.get(workspace_user.user.id, staging_env.id)
@ -57,7 +48,6 @@ def test_update_environment_roles():
def test_get_scoped_environments(db): def test_get_scoped_environments(db):
developer = UserFactory.create() developer = UserFactory.create()
workspace = SuperWorkspaceFactory.create( workspace = SuperWorkspaceFactory.create(
name="hey",
members=[{"user": developer, "role_name": "developer"}], members=[{"user": developer, "role_name": "developer"}],
projects=[ projects=[
{ {