formatting and add financial details to requests in seed_sample script

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
-

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"],
         )
 
 

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: