Replace setup with WorkspaceFactory

This commit is contained in:
richard-dds 2018-10-08 10:44:38 -04:00
parent 40557a5494
commit 7df8d3bfa4

View File

@ -1,5 +1,5 @@
from atst.domain.projects import Projects from atst.domain.projects import Projects
from tests.factories import RequestFactory, UserFactory from tests.factories import RequestFactory, UserFactory, WorkspaceFactory
from atst.domain.workspaces import Workspaces from atst.domain.workspaces import Workspaces
@ -18,12 +18,17 @@ def test_create_project_with_multiple_environments():
def test_workspace_owner_can_view_environments(): def test_workspace_owner_can_view_environments():
owner = UserFactory.create() owner = UserFactory.create()
request = RequestFactory.create(creator=owner) workspace = WorkspaceFactory.create(
workspace = Workspaces.create(request) owner=owner,
_project = Projects.create( projects=[
owner, workspace, "My Test Project", "Test", ["dev", "prod"] {
"environments": [
{"name": "dev"},
{"name": "prod"}
]
}
]
) )
project = Projects.get(owner, workspace, workspace.projects[0].id)
project = Projects.get(owner, workspace, _project.id)
assert len(project.environments) == 2 assert len(project.environments) == 2