Attach PDFs to task orders

This commit is contained in:
George Drummond 2019-06-12 15:02:57 -04:00
parent 14b3e3f933
commit c719999cdb
No known key found for this signature in database
GPG Key ID: 296DD6077123BF17

View File

@ -4,6 +4,7 @@ import sys
from datetime import timedelta, date, timedelta from datetime import timedelta, date, timedelta
import random import random
from faker import Faker from faker import Faker
from werkzeug.datastructures import FileStorage
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)
@ -166,11 +167,20 @@ 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 = 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) unsigned_to = TaskOrderFactory.build(portfolio=portfolio, pdf=pdf)
upcoming_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) upcoming_to = TaskOrderFactory.build(
expired_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) portfolio=portfolio, signed_at=yesterday, pdf=pdf
active_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) )
expired_to = TaskOrderFactory.build(
portfolio=portfolio, signed_at=yesterday, pdf=pdf
)
active_to = TaskOrderFactory.build(
portfolio=portfolio, signed_at=yesterday, pdf=pdf
)
clins = [ clins = [
CLINFactory.build( CLINFactory.build(