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

Implemen TaskOrder.is_completed

Browse Source
This commit is contained in:
richard-dds 2019-06-06 16:21:02 -04:00
parent 8ecf112c48
commit 0f4d17a94a
3 changed files with 35 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
atst/models/task_order.py
View File

@ -67,7 +67,7 @@ class TaskOrder(Base, mixins.TimestampsMixin):
@property
def is_completed(self):
return True
return all([self.pdf, self.number, len(self.clins)])
@property
def is_signed(self):

1
tests/factories.py
View File

@ -272,6 +272,7 @@ class TaskOrderFactory(Base):
portfolio = factory.SubFactory(PortfolioFactory)
number = factory.LazyFunction(random_task_order_number)
creator = factory.SubFactory(UserFactory)
_pdf = factory.SubFactory(AttachmentFactory)
@classmethod
def _create(cls, model_class, *args, **kwargs):

60
tests/models/test_task_order.py
View File

@ -17,36 +17,42 @@ from tests.factories import (
from tests.mocks import PDF_FILENAME
class TestPeriodOfPerformance:
def test_period_of_performance_is_first_to_last_clin(self):
start_date = date(2019, 6, 6)
end_date = date(2020, 6, 6)
def test_period_of_performance_is_first_to_last_clin():
start_date = date(2019, 6, 6)
end_date = date(2020, 6, 6)
intermediate_start_date = date(2019, 7, 1)
intermediate_end_date = date(2020, 3, 1)
intermediate_start_date = date(2019, 7, 1)
intermediate_end_date = date(2020, 3, 1)
task_order = TaskOrderFactory.create(
clins=[
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
]
)
task_order = TaskOrderFactory.create(
clins=[
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=end_date
),
CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date
),
]
)
assert task_order.start_date == start_date
assert task_order.end_date == end_date
assert task_order.start_date == start_date
assert task_order.end_date == end_date
def test_task_order_completed():
assert TaskOrderFactory.create(clins=[CLINFactory.create()]).is_completed
assert not TaskOrderFactory.create().is_completed
assert not TaskOrderFactory.create(clins=[]).is_completed
assert not TaskOrderFactory.create(number=None).is_completed
class TestTaskOrderStatus: