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

Merge pull request #692 from dod-ccpo/fix-seed-sample

Browse Source 
Create all users before creating portfolios
This commit is contained in:
leigh-mil 2019-03-08 10:16:25 -05:00 committed by GitHub
commit ae56b74e1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
script/seed_sample.py
View File

@ -53,41 +53,6 @@ PORTFOLIO_USERS = [
},
]
PORTFOLIO_INVITED_USERS = [
{
"first_name": "Frederick",
"last_name": "Fitzgerald",
"email": "frederick@mil.gov",
"portfolio_role": "developer",
"dod_id": "0000000004",
"status": InvitationStatus.REJECTED_WRONG_USER,
},
{
"first_name": "Gina",
"last_name": "Guzman",
"email": "gina@mil.gov",
"portfolio_role": "developer",
"dod_id": "0000000005",
"status": InvitationStatus.REJECTED_EXPIRED,
},
{
"first_name": "Hector",
"last_name": "Harper",
"email": "hector@mil.gov",
"portfolio_role": "developer",
"dod_id": "0000000006",
"status": InvitationStatus.REVOKED,
},
{
"first_name": "Isabella",
"last_name": "Ingram",
"email": "isabella@mil.gov",
"portfolio_role": "developer",
"dod_id": "0000000007",
"status": InvitationStatus.PENDING,
},
]
def get_users():
users = []
@ -102,7 +67,6 @@ def get_users():
def add_members_to_portfolio(portfolio):
get_users()
for portfolio_role in PORTFOLIO_USERS:
ws_role = Portfolios.create_member(portfolio.owner, portfolio, portfolio_role)
db.session.refresh(ws_role)
@ -111,18 +75,6 @@ def add_members_to_portfolio(portfolio):
db.session.commit()
def invite_members_to_portfolio(portfolio):
get_users()
for portfolio_role in PORTFOLIO_INVITED_USERS:
ws_role = Portfolios.create_member(portfolio.owner, portfolio, portfolio_role)
invitation = InvitationFactory.build(
portfolio_role=ws_role, status=portfolio_role["status"]
)
db.session.add(invitation)
db.session.commit()
def add_task_orders_to_portfolio(portfolio, to_length=90, clin_01=None, clin_03=None):
active_to_offset = random.randint(10, 31)
# exp TO ends same day as active TO starts
@ -200,6 +152,7 @@ def create_demo_portfolio(name, data):
def seed_db():
get_users()
amanda = Users.get_by_dod_id("2345678901")
application_info = [
{