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

Add workspace members to seed script

Browse Source
This commit is contained in:
richard-dds
2018-09-04 11:44:02 -04:00
parent a0b0b54146
commit b4f34a25e0
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
script/seed.py
View File

@@ -15,6 +15,29 @@ from atst.domain.exceptions import AlreadyExistsError
from tests.factories import RequestFactory from tests.factories import RequestFactory
from atst.routes.dev import _DEV_USERS as DEV_USERS from atst.routes.dev import _DEV_USERS as DEV_USERS
WORKSPACE_USERS = [
{
"first_name": "Danny",
"last_name": "Knight",
"email": "knight@mil.gov",
"workspace_role": "developer",
"dod_id": "0000000001"
},
{
"first_name": "Mario",
"last_name": "Hudson",
"email": "hudson@mil.gov",
"workspace_role": "ccpo",
"dod_id": "0000000002"
},
{
"first_name": "Louise",
"last_name": "Greer",
"email": "greer@mil.gov",
"workspace_role": "admin",
"dod_id": "0000000003"
},
]
def seed_db(): def seed_db():
users = [] users = []
@@ -41,6 +64,9 @@ def seed_db():
requests.append(request) requests.append(request)
workspace = Workspaces.create(requests[0], name="{}'s workspace".format(user.first_name)) workspace = Workspaces.create(requests[0], name="{}'s workspace".format(user.first_name))
for workspace_user in WORKSPACE_USERS:
Workspaces.create_member(user, workspace, workspace_user)
Projects.create( Projects.create(
workspace=workspace, workspace=workspace,
name="First Project", name="First Project",