Implemen TaskOrder.is_completed

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

View File

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

View File

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

View File

@ -17,36 +17,42 @@ from tests.factories import (
from tests.mocks import PDF_FILENAME from tests.mocks import PDF_FILENAME
class TestPeriodOfPerformance: def test_period_of_performance_is_first_to_last_clin():
def test_period_of_performance_is_first_to_last_clin(self): start_date = date(2019, 6, 6)
start_date = date(2019, 6, 6) end_date = date(2020, 6, 6)
end_date = date(2020, 6, 6)
intermediate_start_date = date(2019, 7, 1) intermediate_start_date = date(2019, 7, 1)
intermediate_end_date = date(2020, 3, 1) intermediate_end_date = date(2020, 3, 1)
task_order = TaskOrderFactory.create( task_order = TaskOrderFactory.create(
clins=[ clins=[
CLINFactory.create( CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date start_date=intermediate_start_date, end_date=intermediate_end_date
), ),
CLINFactory.create( CLINFactory.create(
start_date=start_date, end_date=intermediate_end_date start_date=start_date, end_date=intermediate_end_date
), ),
CLINFactory.create( CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date start_date=intermediate_start_date, end_date=intermediate_end_date
), ),
CLINFactory.create( CLINFactory.create(
start_date=intermediate_start_date, end_date=end_date start_date=intermediate_start_date, end_date=end_date
), ),
CLINFactory.create( CLINFactory.create(
start_date=intermediate_start_date, end_date=intermediate_end_date start_date=intermediate_start_date, end_date=intermediate_end_date
), ),
] ]
) )
assert task_order.start_date == start_date assert task_order.start_date == start_date
assert task_order.end_date == end_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: class TestTaskOrderStatus: