From e41f01398c2ba6e18d31b94ce801d48ea4df04e4 Mon Sep 17 00:00:00 2001
From: Patrick Smith <patrick+dds@promptworks.com>
Date: Tue, 15 Jan 2019 11:16:40 -0500
Subject: [PATCH] Update seed sample to create multiple task orders for each
 portfolio

---
 script/seed_sample.py | 52 ++++++++++++++++++++++++++-----------------
 tests/factories.py    |  4 ++++
 2 files changed, 36 insertions(+), 20 deletions(-)

diff --git a/script/seed_sample.py b/script/seed_sample.py
index 40c82759..1395d7b0 100644
--- a/script/seed_sample.py
+++ b/script/seed_sample.py
@@ -14,7 +14,14 @@ from atst.domain.applications import Applications
 from atst.domain.portfolio_roles import PortfolioRoles
 from atst.models.invitation import Status as InvitationStatus
 from atst.domain.exceptions import AlreadyExistsError
-from tests.factories import RequestFactory, LegacyTaskOrderFactory, InvitationFactory
+from tests.factories import (
+    InvitationFactory,
+    RequestFactory,
+    TaskOrderFactory,
+    random_future_date,
+    random_past_date,
+    random_task_order_number,
+)
 from atst.routes.dev import _DEV_USERS as DEV_USERS
 
 PORTFOLIO_USERS = [
@@ -88,25 +95,6 @@ def seed_db():
         users.append(user)
 
     for user in users:
-        if Requests.get_many(creator=user):
-            continue
-
-        requests = []
-        for dollar_value in [1, 200, 3000, 40000, 500_000, 1_000_000]:
-            request = RequestFactory.build(creator=user)
-            request.latest_revision.dollar_value = dollar_value
-            db.session.add(request)
-            db.session.commit()
-
-            Requests.submit(request)
-            requests.append(request)
-
-        request = requests[0]
-        request.legacy_task_order = LegacyTaskOrderFactory.build()
-        request = Requests.update(
-            request.id, {"financial_verification": RequestFactory.mock_financial_data()}
-        )
-
         portfolio = Portfolios.create(
             user, name="{}'s portfolio".format(user.first_name)
         )
@@ -122,6 +110,30 @@ def seed_db():
             )
             db.session.add(invitation)
 
+        [expired_start, expired_end] = sorted(
+            [
+                random_past_date(year_max=2, year_min=1),
+                random_past_date(year_max=1, year_min=1),
+            ]
+        )
+        active_start = expired_end
+        active_end = random_future_date(year_min=1, year_max=1)
+
+        date_ranges = [(expired_start, expired_end), (active_start, active_end)]
+        for (start_date, end_date) in date_ranges:
+            task_order = TaskOrderFactory.build(
+                start_date=start_date,
+                end_date=end_date,
+                number=random_task_order_number(),
+                portfolio=portfolio,
+            )
+            db.session.add(task_order)
+
+        pending_task_order = TaskOrderFactory.build(
+            start_date=None, end_date=None, number=None, portfolio=portfolio
+        )
+        db.session.add(pending_task_order)
+
         db.session.commit()
 
         Applications.create(
diff --git a/tests/factories.py b/tests/factories.py
index 75ce839e..0f37d4e9 100644
--- a/tests/factories.py
+++ b/tests/factories.py
@@ -42,6 +42,10 @@ def random_phone_number():
     return "".join(random.choices(string.digits, k=10))
 
 
+def random_task_order_number():
+    return "-".join([str(random.randint(100, 999)) for _ in range(4)])
+
+
 def random_past_date(year_min=1, year_max=5):
     return _random_date(year_min, year_max, operator.sub)