Cleaning up tests.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-01-16 14:46:17 -05:00
parent 9523cc1d97
commit 8c6046b03b
33 changed files with 268 additions and 315 deletions

View File

@@ -1581,7 +1581,7 @@ class Notification(db.Model):
self._personalisation = encryption.encrypt(personalisation or {})
def completed_at(self):
if self.status in NotificationStatus.completed:
if self.status in NotificationStatus.completed_types:
return self.updated_at.strftime(DATETIME_FORMAT)
return None
@@ -1621,7 +1621,7 @@ class Notification(db.Model):
def _substitute_status_str(_status):
return (
NotificationStatus.failed
NotificationStatus.failed_types
if _status == NotificationStatus.FAILED
else [_status]
)