diff --git a/script/seed_sample.py b/script/seed_sample.py index 2ed03d90..7bc9cb9d 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -4,6 +4,7 @@ import sys from datetime import timedelta, date, timedelta import random from faker import Faker +from werkzeug.datastructures import FileStorage parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) sys.path.append(parent_dir) @@ -166,11 +167,20 @@ def add_task_orders_to_portfolio(portfolio): yesterday = today - timedelta(days=1) five_days = timedelta(days=5) - draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None) - unsigned_to = TaskOrderFactory.build(portfolio=portfolio) - upcoming_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) - expired_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) - active_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) + with open("tests/fixtures/sample.pdf", "rb") as fp: + pdf = FileStorage(fp, content_type="application/pdf") + + draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None) + unsigned_to = TaskOrderFactory.build(portfolio=portfolio, pdf=pdf) + upcoming_to = TaskOrderFactory.build( + portfolio=portfolio, signed_at=yesterday, pdf=pdf + ) + expired_to = TaskOrderFactory.build( + portfolio=portfolio, signed_at=yesterday, pdf=pdf + ) + active_to = TaskOrderFactory.build( + portfolio=portfolio, signed_at=yesterday, pdf=pdf + ) clins = [ CLINFactory.build(