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

Fix seed script

Browse Source
This commit is contained in:
richard-dds 2019-08-13 10:16:25 -04:00
parent bc0fd4900a
commit 034645d013
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
script/seed_sample.py
View File

@ -5,6 +5,7 @@ from datetime import timedelta, date, timedelta
import random import random
from faker import Faker from faker import Faker
from werkzeug.datastructures import FileStorage from werkzeug.datastructures import FileStorage
from uuid import uuid4
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.append(parent_dir) sys.path.append(parent_dir)
@ -30,9 +31,9 @@ from atst.routes.dev import _DEV_USERS as DEV_USERS
from tests.factories import ( from tests.factories import (
random_service_branch, random_service_branch,
random_task_order_number,
TaskOrderFactory, TaskOrderFactory,
CLINFactory, CLINFactory,
AttachmentFactory
) )
fake = Faker() fake = Faker()
@ -167,8 +168,7 @@ def add_task_orders_to_portfolio(portfolio):
yesterday = today - timedelta(days=1) yesterday = today - timedelta(days=1)
five_days = timedelta(days=5) five_days = timedelta(days=5)
with open("tests/fixtures/sample.pdf", "rb") as fp: pdf = {"filename": "sample_task_order.pdf", "object_name": str(uuid4())}
pdf = FileStorage(fp, content_type="application/pdf")
draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None) draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None)
unsigned_to = TaskOrderFactory.build(portfolio=portfolio, pdf=pdf) unsigned_to = TaskOrderFactory.build(portfolio=portfolio, pdf=pdf)