From 7df8d3bfa44ddb5c9f7cc963f1be55b5014117e8 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Mon, 8 Oct 2018 10:44:38 -0400 Subject: [PATCH] Replace setup with WorkspaceFactory --- tests/domain/test_projects.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/tests/domain/test_projects.py b/tests/domain/test_projects.py index e473f4df..480e588c 100644 --- a/tests/domain/test_projects.py +++ b/tests/domain/test_projects.py @@ -1,5 +1,5 @@ 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 @@ -18,12 +18,17 @@ def test_create_project_with_multiple_environments(): def test_workspace_owner_can_view_environments(): owner = UserFactory.create() - request = RequestFactory.create(creator=owner) - workspace = Workspaces.create(request) - _project = Projects.create( - owner, workspace, "My Test Project", "Test", ["dev", "prod"] + workspace = WorkspaceFactory.create( + owner=owner, + projects=[ + { + "environments": [ + {"name": "dev"}, + {"name": "prod"} + ] + } + ] ) - - project = Projects.get(owner, workspace, _project.id) + project = Projects.get(owner, workspace, workspace.projects[0].id) assert len(project.environments) == 2