Attach PDFs to task orders

This commit is contained in:
George Drummond
2019-06-12 15:02:57 -04:00
parent 14b3e3f933
commit c719999cdb

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(