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

formatting and add financial details to requests in seed_sample script

Browse Source
This commit is contained in:
dandds 2018-09-24 09:47:59 -04:00
parent 430efd211c
commit e668cb93d0
3 changed files with 33 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
atst/models/request.py
View File

@ -220,9 +220,10 @@ class Request(Base, mixins.TimestampsMixin):
@property @property
def contracting_officer_full_name(self): def contracting_officer_full_name(self):
if self.latest_revision.fname_co: if self.latest_revision.fname_co:
return "{} {}".format(self.latest_revision.fname_co, self.latest_revision.lname_co) return "{} {}".format(
self.latest_revision.fname_co, self.latest_revision.lname_co
)
@property @property
def contracting_officer_email(self): def contracting_officer_email(self):
return self.latest_revision.email_co return self.latest_revision.email_co

16
script/seed_sample.py
View File

@ -21,24 +21,25 @@ WORKSPACE_USERS = [
"last_name": "Knight", "last_name": "Knight",
"email": "knight@mil.gov", "email": "knight@mil.gov",
"workspace_role": "developer", "workspace_role": "developer",
"dod_id": "0000000001" "dod_id": "0000000001",
}, },
{ {
"first_name": "Mario", "first_name": "Mario",
"last_name": "Hudson", "last_name": "Hudson",
"email": "hudson@mil.gov", "email": "hudson@mil.gov",
"workspace_role": "ccpo", "workspace_role": "ccpo",
"dod_id": "0000000002" "dod_id": "0000000002",
}, },
{ {
"first_name": "Louise", "first_name": "Louise",
"last_name": "Greer", "last_name": "Greer",
"email": "greer@mil.gov", "email": "greer@mil.gov",
"workspace_role": "admin", "workspace_role": "admin",
"dod_id": "0000000003" "dod_id": "0000000003",
}, },
] ]
def seed_db(): def seed_db():
users = [] users = []
for dev_user in DEV_USERS.values(): for dev_user in DEV_USERS.values():
@ -65,8 +66,13 @@ def seed_db():
request = requests[0] request = requests[0]
request.task_order = TaskOrderFactory.build() request.task_order = TaskOrderFactory.build()
request = Requests.update(
request.id, {"financial_verification": RequestFactory.mock_financial_data()}
)
workspace = Workspaces.create(request, name="{}'s workspace".format(user.first_name)) workspace = Workspaces.create(
request, name="{}'s workspace".format(user.first_name)
)
for workspace_user in WORKSPACE_USERS: for workspace_user in WORKSPACE_USERS:
Workspaces.create_member(user, workspace, workspace_user) Workspaces.create_member(user, workspace, workspace_user)
@ -75,7 +81,7 @@ def seed_db():
workspace=workspace, workspace=workspace,
name="First Project", name="First Project",
description="This is our first project.", description="This is our first project.",
environment_names=["dev", "staging", "prod"] environment_names=["dev", "staging", "prod"],
) )

19
tests/factories.py
View File

@ -3,6 +3,7 @@ import string
import factory import factory
from uuid import uuid4 from uuid import uuid4
import datetime import datetime
from faker import Faker as _Faker
from atst.forms.data import SERVICE_BRANCHES from atst.forms.data import SERVICE_BRANCHES
from atst.models.request import Request from atst.models.request import Request
@ -159,6 +160,24 @@ class RequestFactory(Base):
) )
return request return request
@classmethod
def mock_financial_data(cls):
fake = _Faker()
return {
"pe_id": "0101110F",
"fname_co": fake.first_name(),
"lname_co": fake.last_name(),
"email_co": fake.email(),
"office_co": fake.phone_number(),
"fname_cor": fake.first_name(),
"lname_cor": fake.last_name(),
"email_cor": fake.email(),
"office_cor": fake.phone_number(),
"uii_ids": "123abc",
"treasury_code": "00123456",
"ba_code": "02A",
}
class PENumberFactory(Base): class PENumberFactory(Base):
class Meta: class Meta: