From b4f34a25e061d3bdf6e1b47ef62b962833175872 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 4 Sep 2018 11:44:02 -0400 Subject: [PATCH] Add workspace members to seed script --- script/seed.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/script/seed.py b/script/seed.py index e69694a0..01050427 100644 --- a/script/seed.py +++ b/script/seed.py @@ -15,6 +15,29 @@ from atst.domain.exceptions import AlreadyExistsError from tests.factories import RequestFactory 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(): users = [] @@ -41,6 +64,9 @@ def seed_db(): requests.append(request) 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( workspace=workspace, name="First Project",