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

Add some invited users to the seed script

Browse Source
This commit is contained in:
richard-dds 2018-11-05 15:41:39 -05:00
parent c3e1a7eb27
commit b1468f1376
1 changed files with 44 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
script/seed_sample.py
View File

@ -12,8 +12,9 @@ from atst.domain.requests import Requests
from atst.domain.workspaces import Workspaces 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.invitation import Status as InvitationStatus
from atst.domain.exceptions import AlreadyExistsError from atst.domain.exceptions import AlreadyExistsError
from tests.factories import RequestFactory, TaskOrderFactory from tests.factories import RequestFactory, TaskOrderFactory, InvitationFactory
from atst.routes.dev import _DEV_USERS as DEV_USERS from atst.routes.dev import _DEV_USERS as DEV_USERS
WORKSPACE_USERS = [ WORKSPACE_USERS = [
@ -40,6 +41,41 @@ WORKSPACE_USERS = [
}, },
] ]
WORKSPACE_INVITED_USERS = [
{
"first_name": "Frederick",
"last_name": "Fitzgerald",
"email": "frederick@mil.gov",
"workspace_role": "developer",
"dod_id": "0000000004",
"status": InvitationStatus.REJECTED_WRONG_USER
},
{
"first_name": "Gina",
"last_name": "Guzman",
"email": "gina@mil.gov",
"workspace_role": "developer",
"dod_id": "0000000005",
"status": InvitationStatus.REJECTED_EXPIRED
},
{
"first_name": "Hector",
"last_name": "Harper",
"email": "hector@mil.gov",
"workspace_role": "developer",
"dod_id": "0000000006",
"status": InvitationStatus.REVOKED
},
{
"first_name": "Isabella",
"last_name": "Ingram",
"email": "isabella@mil.gov",
"workspace_role": "developer",
"dod_id": "0000000007",
"status": InvitationStatus.PENDING
},
]
def seed_db(): def seed_db():
users = [] users = []
@ -78,6 +114,13 @@ def seed_db():
ws_role = Workspaces.create_member(user, workspace, workspace_role) ws_role = Workspaces.create_member(user, workspace, workspace_role)
WorkspaceRoles.enable(ws_role) WorkspaceRoles.enable(ws_role)
for workspace_role in WORKSPACE_INVITED_USERS:
ws_role = Workspaces.create_member(user, workspace, workspace_role)
invitation = InvitationFactory.build(workspace_role=ws_role, status=workspace_role["status"])
db.session.add(invitation)
db.session.commit()
Projects.create( Projects.create(
user, user,
workspace=workspace, workspace=workspace,