darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-07-17 21:20:59 -04:00
Code Issues Packages Projects Releases Wiki Activity

remove default postage from noti history

Browse Source 
also add test for notification_history properly setting
This commit is contained in:
Leo Hemsted
2018-09-20 12:26:58 +01:00
parent 4dc3f829e3
commit 8e19a0742f
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/models.py
View File

@@ -1436,7 +1436,7 @@ class NotificationHistory(db.Model, HistoryModel):
created_by = db.relationship('User')
created_by_id = db.Column(UUID(as_uuid=True), db.ForeignKey('users.id'), nullable=True)
postage = db.Column(db.String, nullable=True, default='second')
postage = db.Column(db.String, nullable=True)
CheckConstraint("notification_type != 'letter' or postage in ('first', 'second')")
__table_args__ = (

6
tests/app/v2/notifications/test_post_letter_notifications.py
View File

@@ -9,6 +9,7 @@ from app.config import TaskNames, QueueNames
from app.models import (
Job,
Notification,
NotificationHistory,
EMAIL_TYPE,
KEY_TYPE_NORMAL,
KEY_TYPE_TEAM,
@@ -483,12 +484,15 @@ def test_post_precompiled_letter_notification_returns_201(client, notify_user, m
s3mock.assert_called_once_with(ANY, b'letter-content', precompiled=True)
notification = Notification.query.first()
notification = Notification.query.one()
assert notification.billable_units == 3
assert notification.status == NOTIFICATION_PENDING_VIRUS_CHECK
assert notification.postage == postage
notification_history = NotificationHistory.query.one()
assert notification_history.postage == postage
resp_json = json.loads(response.get_data(as_text=True))
assert resp_json == {'id': str(notification.id), 'reference': 'letter-reference'}