From 2a67fb619794bbc17bdbacfa360a2a7435c3ff8d Mon Sep 17 00:00:00 2001
From: dandds <dan-ctr@friends.dds.mil>
Date: Thu, 20 Sep 2018 11:42:03 -0400
Subject: [PATCH] add task order data to seed workspace

---
 script/seed.py     | 7 +++++--
 tests/factories.py | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/script/seed.py b/script/seed.py
index cebe5528..ee113afd 100644
--- a/script/seed.py
+++ b/script/seed.py
@@ -12,7 +12,7 @@ from atst.domain.requests import Requests
 from atst.domain.workspaces import Workspaces
 from atst.domain.projects import Projects
 from atst.domain.exceptions import AlreadyExistsError
-from tests.factories import RequestFactory
+from tests.factories import RequestFactory, TaskOrderFactory
 from atst.routes.dev import _DEV_USERS as DEV_USERS
 
 WORKSPACE_USERS = [
@@ -63,7 +63,10 @@ def seed_db():
             Requests.submit(request)
             requests.append(request)
 
-        workspace = Workspaces.create(requests[0], name="{}'s workspace".format(user.first_name))
+        request = requests[0]
+        request.task_order = TaskOrderFactory.build()
+
+        workspace = Workspaces.create(request, name="{}'s workspace".format(user.first_name))
         for workspace_user in WORKSPACE_USERS:
             Workspaces.create_member(user, workspace, workspace_user)
 
diff --git a/tests/factories.py b/tests/factories.py
index dd342b09..b462701c 100644
--- a/tests/factories.py
+++ b/tests/factories.py
@@ -170,7 +170,7 @@ class TaskOrderFactory(Base):
     source = Source.MANUAL
     funding_type = FundingType.PROC
     funding_type_other = None
-    number = "toABC123"
+    number = factory.Faker("md5")
     expiration_date = factory.LazyFunction(
         lambda: datetime.date(
             datetime.date.today().year + random.randrange(1, 15), 1, 1