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

add task order data to seed workspace

Browse Source
This commit is contained in:
dandds 2018-09-20 11:42:03 -04:00
parent 9f7284070e
commit 2a67fb6197
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
script/seed.py
View File

@ -12,7 +12,7 @@ from atst.domain.requests import Requests
from atst.domain.workspaces import Workspaces from atst.domain.workspaces import Workspaces
from atst.domain.projects import Projects from atst.domain.projects import Projects
from atst.domain.exceptions import AlreadyExistsError 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 from atst.routes.dev import _DEV_USERS as DEV_USERS
WORKSPACE_USERS = [ WORKSPACE_USERS = [
@ -63,7 +63,10 @@ def seed_db():
Requests.submit(request) Requests.submit(request)
requests.append(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: for workspace_user in WORKSPACE_USERS:
Workspaces.create_member(user, workspace, workspace_user) Workspaces.create_member(user, workspace, workspace_user)

2
tests/factories.py
View File

@ -170,7 +170,7 @@ class TaskOrderFactory(Base):
source = Source.MANUAL source = Source.MANUAL
funding_type = FundingType.PROC funding_type = FundingType.PROC
funding_type_other = None funding_type_other = None
number = "toABC123" number = factory.Faker("md5")
expiration_date = factory.LazyFunction( expiration_date = factory.LazyFunction(
lambda: datetime.date( lambda: datetime.date(
datetime.date.today().year + random.randrange(1, 15), 1, 1 datetime.date.today().year + random.randrange(1, 15), 1, 1