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

Use WorkspaceRoleFactory in tests

Browse Source
This commit is contained in:
Montana 2018-11-16 15:34:42 -05:00
parent e73690986a
commit 3b2594283f
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/test_workspace_role.py
View File

@ -5,6 +5,7 @@ from atst.domain.workspaces import Workspaces
from atst.domain.projects import Projects from atst.domain.projects import Projects
from atst.domain.workspace_roles import WorkspaceRoles from atst.domain.workspace_roles import WorkspaceRoles
from atst.models.workspace_role import Status from atst.models.workspace_role import Status
from atst.models.role import Role
from atst.models.invitation import Status as InvitationStatus from atst.models.invitation import Status as InvitationStatus
from tests.factories import ( from tests.factories import (
RequestFactory, RequestFactory,
@ -36,8 +37,8 @@ def test_has_role_history(session):
user = UserFactory.create() user = UserFactory.create()
workspace = Workspaces.create(RequestFactory.create(creator=owner)) workspace = Workspaces.create(RequestFactory.create(creator=owner))
workspace_role = WorkspaceRoles.add(user, workspace.id, "developer") role = session.query(Role).filter(Role.name == 'developer').one()
# why not use workspace_role = WorkspaceRoleFactory.create()? workspace_role = WorkspaceRoleFactory.create(workspace=workspace, user=user, role=role)
WorkspaceRoles.update_role(workspace_role, "admin") WorkspaceRoles.update_role(workspace_role, "admin")
changed_events = ( changed_events = (
session.query(AuditEvent) session.query(AuditEvent)
@ -57,10 +58,7 @@ def test_has_status_history(session):
user = UserFactory.create() user = UserFactory.create()
workspace = Workspaces.create(RequestFactory.create(creator=owner)) workspace = Workspaces.create(RequestFactory.create(creator=owner))
workspace_role = WorkspaceRoles.add(user, workspace.id, "developer") workspace_role = WorkspaceRoleFactory.create(workspace=workspace, user=user)
import ipdb
ipdb.set_trace()
WorkspaceRoles.enable(workspace_role) WorkspaceRoles.enable(workspace_role)
changed_events = ( changed_events = (
session.query(AuditEvent) session.query(AuditEvent)