diff --git a/atst/models/request.py b/atst/models/request.py index ebba9cc3..9926aa12 100644 --- a/atst/models/request.py +++ b/atst/models/request.py @@ -220,9 +220,10 @@ class Request(Base, mixins.TimestampsMixin): @property def contracting_officer_full_name(self): 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 def contracting_officer_email(self): return self.latest_revision.email_co - diff --git a/script/seed_sample.py b/script/seed_sample.py index ee113afd..237b10a3 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -21,24 +21,25 @@ WORKSPACE_USERS = [ "last_name": "Knight", "email": "knight@mil.gov", "workspace_role": "developer", - "dod_id": "0000000001" + "dod_id": "0000000001", }, { "first_name": "Mario", "last_name": "Hudson", "email": "hudson@mil.gov", "workspace_role": "ccpo", - "dod_id": "0000000002" + "dod_id": "0000000002", }, { "first_name": "Louise", "last_name": "Greer", "email": "greer@mil.gov", "workspace_role": "admin", - "dod_id": "0000000003" + "dod_id": "0000000003", }, ] + def seed_db(): users = [] for dev_user in DEV_USERS.values(): @@ -65,8 +66,13 @@ def seed_db(): request = requests[0] 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: Workspaces.create_member(user, workspace, workspace_user) @@ -75,7 +81,7 @@ def seed_db(): workspace=workspace, name="First Project", description="This is our first project.", - environment_names=["dev", "staging", "prod"] + environment_names=["dev", "staging", "prod"], ) diff --git a/tests/factories.py b/tests/factories.py index 2f8677f0..ab8f17ca 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -3,6 +3,7 @@ import string import factory from uuid import uuid4 import datetime +from faker import Faker as _Faker from atst.forms.data import SERVICE_BRANCHES from atst.models.request import Request @@ -159,6 +160,24 @@ class RequestFactory(Base): ) 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 Meta: