def test_displayed_status_field_visit(self):
e = Engagement(
status=Engagement.STATUSES.partner_contacted,
date_of_field_visit=datetime.date(2001, 1, 1),
)
self.assertEqual(e.displayed_status,
Engagement.DISPLAY_STATUSES.field_visit)
def test_displayed_status_draft_issues_to_partner(self):
e = Engagement(
status=Engagement.STATUSES.partner_contacted,
date_of_draft_report_to_ip=datetime.date(2001, 1, 1),
)
self.assertEqual(e.displayed_status,
Engagement.DISPLAY_STATUSES.draft_issued_to_partner)
def test_displayed_status_comments_by_partner(self):
e = Engagement(
status=Engagement.STATUSES.partner_contacted,
date_of_comments_by_ip=datetime.date(2001, 1, 1),
)
self.assertEqual(
e.displayed_status,
Engagement.DISPLAY_STATUSES.comments_received_by_partner)
def test_displayed_status_partner_not_contacted(self):
e = Engagement(status=Engagement.STATUSES.final)
self.assertEqual(e.displayed_status, e.status)