diff --git a/app/models.py b/app/models.py index 9d16c4968..f209dbb13 100644 --- a/app/models.py +++ b/app/models.py @@ -645,6 +645,14 @@ class TemplateHistory(db.Model): nullable=False, default=NORMAL) + def get_link(self): + return url_for( + "v2_template.get_template_by_id", + template_id=self.id, + version=self.version, + _external=True + ) + def _as_utils_template(self): return Template._as_utils_template(self) diff --git a/tests/app/v2/notifications/test_get_notifications.py b/tests/app/v2/notifications/test_get_notifications.py index d4306ba93..fc6b091c9 100644 --- a/tests/app/v2/notifications/test_get_notifications.py +++ b/tests/app/v2/notifications/test_get_notifications.py @@ -335,7 +335,7 @@ def test_get_all_notifications_filter_by_template_type(client): assert json_response['notifications'][0]['status'] == "created" assert json_response['notifications'][0]['template'] == { 'id': str(email_template.id), - 'uri': email_template.get_link(), + 'uri': notification.template.get_link(), 'version': 1 } assert json_response['notifications'][0]['email_address'] == "don.draper@scdp.biz"