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

Merge pull request #895 from dod-ccpo/seeds-changes

Browse Source 
More realistic seeds
This commit is contained in:
George Drummond 2019-06-12 15:43:34 -04:00 committed by GitHub
commit 32022b054c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
script/seed_sample.py
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)
@ -164,18 +165,32 @@ def add_task_orders_to_portfolio(portfolio):
today = date.today() today = date.today()
future = today + timedelta(days=100) future = today + timedelta(days=100)
yesterday = today - timedelta(days=1) yesterday = today - timedelta(days=1)
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(task_order=unsigned_to, start_date=today, end_date=today),
CLINFactory.build(task_order=upcoming_to, start_date=future, end_date=future),
CLINFactory.build( CLINFactory.build(
task_order=expired_to, start_date=yesterday, end_date=yesterday task_order=unsigned_to, start_date=(today - five_days), end_date=today
),
CLINFactory.build(
task_order=upcoming_to, start_date=future, end_date=(today + five_days)
),
CLINFactory.build(
task_order=expired_to, start_date=(today - five_days), end_date=yesterday
), ),
CLINFactory.build(task_order=active_to, start_date=yesterday, end_date=future), CLINFactory.build(task_order=active_to, start_date=yesterday, end_date=future),
] ]

10
tests/factories.py
View File

@ -30,7 +30,7 @@ def random_phone_number():
def random_task_order_number(): def random_task_order_number():
return "-".join([str(random.randint(100, 999)) for _ in range(4)]) return "".join(random.choices(string.digits, k=10))
def random_past_date(year_min=1, year_max=5): def random_past_date(year_min=1, year_max=5):
@ -261,7 +261,7 @@ class AttachmentFactory(Base):
class Meta: class Meta:
model = Attachment model = Attachment
filename = factory.Faker("domain_word") filename = factory.Faker("file_name", extension="pdf")
object_name = factory.LazyFunction(lambda *args: uuid4().hex) object_name = factory.LazyFunction(lambda *args: uuid4().hex)
@ -293,8 +293,10 @@ class CLINFactory(Base):
number = factory.LazyFunction(random_task_order_number) number = factory.LazyFunction(random_task_order_number)
start_date = datetime.date.today() start_date = datetime.date.today()
end_date = factory.LazyFunction(random_future_date) end_date = factory.LazyFunction(random_future_date)
obligated_amount = random.randint(100, 999999) obligated_amount = factory.LazyFunction(lambda *args: random.randint(100, 999999))
jedi_clin_type = random.choice(list(clin.JEDICLINType)) jedi_clin_type = factory.LazyFunction(
lambda *args: random.choice(list(clin.JEDICLINType))
)
class NotificationRecipientFactory(Base): class NotificationRecipientFactory(Base):