mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
Add schemas, endpoints and supporting tests
This commit is contained in:
@@ -110,9 +110,11 @@ def register_blueprint(application):
|
||||
def register_v2_blueprints(application):
|
||||
from app.v2.notifications.post_notifications import notification_blueprint as post_notifications
|
||||
from app.v2.notifications.get_notifications import notification_blueprint as get_notifications
|
||||
from app.v2.template.get_template import template_blueprint
|
||||
|
||||
application.register_blueprint(post_notifications)
|
||||
application.register_blueprint(get_notifications)
|
||||
application.register_blueprint(template_blueprint)
|
||||
|
||||
|
||||
def init_app(app):
|
||||
|
||||
@@ -320,6 +320,21 @@ class Template(db.Model):
|
||||
_external=True
|
||||
)
|
||||
|
||||
def serialize(self):
|
||||
|
||||
serialized = {
|
||||
"id": self.id,
|
||||
"type": self.template_type,
|
||||
"created_at": self.created_at.strftime(DATETIME_FORMAT),
|
||||
"updated_at": self.updated_at.strftime(DATETIME_FORMAT) if self.updated_at else None,
|
||||
"created_by": self.created_by.email_address,
|
||||
"version": self.version,
|
||||
"body": self.content,
|
||||
"subject": self.subject if self.template_type == EMAIL_TYPE else None
|
||||
}
|
||||
|
||||
return serialized
|
||||
|
||||
|
||||
class TemplateHistory(db.Model):
|
||||
__tablename__ = 'templates_history'
|
||||
@@ -343,6 +358,21 @@ class TemplateHistory(db.Model):
|
||||
nullable=False,
|
||||
default=NORMAL)
|
||||
|
||||
def serialize(self):
|
||||
|
||||
serialized = {
|
||||
"id": self.id,
|
||||
"type": self.template_type,
|
||||
"created_at": self.created_at.strftime(DATETIME_FORMAT),
|
||||
"updated_at": self.updated_at.strftime(DATETIME_FORMAT) if self.updated_at else None,
|
||||
"created_by": self.created_by.email_address,
|
||||
"version": self.version,
|
||||
"body": self.content,
|
||||
"subject": self.subject if self.template_type == EMAIL_TYPE else None
|
||||
}
|
||||
|
||||
return serialized
|
||||
|
||||
|
||||
MMG_PROVIDER = "mmg"
|
||||
FIRETEXT_PROVIDER = "firetext"
|
||||
|
||||
7
app/v2/template/__init__.py
Normal file
7
app/v2/template/__init__.py
Normal file
@@ -0,0 +1,7 @@
|
||||
from flask import Blueprint
|
||||
|
||||
from app.v2.errors import register_errors
|
||||
|
||||
template_blueprint = Blueprint("v2_template", __name__, url_prefix='/v2/template')
|
||||
|
||||
register_errors(template_blueprint)
|
||||
31
app/v2/template/get_template.py
Normal file
31
app/v2/template/get_template.py
Normal file
@@ -0,0 +1,31 @@
|
||||
import uuid
|
||||
|
||||
from flask import jsonify, request
|
||||
from werkzeug.exceptions import abort
|
||||
|
||||
from app import api_user
|
||||
from app.dao import templates_dao
|
||||
from app.schema_validation import validate
|
||||
from app.v2.template import template_blueprint
|
||||
from app.v2.template.template_schemas import get_template_by_id_request
|
||||
|
||||
|
||||
@template_blueprint.route("/<template_id>", methods=['GET'])
|
||||
@template_blueprint.route("/<template_id>/version/<version>", methods=['GET'])
|
||||
def get_template_by_id(template_id, version=None):
|
||||
try:
|
||||
casted_id = uuid.UUID(template_id)
|
||||
|
||||
_data = {}
|
||||
_data['id'] = template_id
|
||||
if version:
|
||||
_data['version'] = int(version)
|
||||
|
||||
data = validate(_data, get_template_by_id_request)
|
||||
except ValueError or AttributeError:
|
||||
abort(404)
|
||||
|
||||
template = templates_dao.dao_get_template_by_id_and_service_id(
|
||||
casted_id, api_user.service_id, data.get('version'))
|
||||
|
||||
return jsonify(template.serialize()), 200
|
||||
41
app/v2/template/template_schemas.py
Normal file
41
app/v2/template/template_schemas.py
Normal file
@@ -0,0 +1,41 @@
|
||||
from app.models import TEMPLATE_TYPES
|
||||
from app.schema_validation.definitions import uuid
|
||||
|
||||
|
||||
get_template_by_id_request = {
|
||||
"$schema": "http://json-schema.org/draft-04/schema#",
|
||||
"description": "schema for query parameters allowed when getting list of notifications",
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"id": uuid,
|
||||
"version": {"type": ["integer", "null"], "minimum": 1}
|
||||
},
|
||||
"required": ["id"],
|
||||
"additionalProperties": False,
|
||||
}
|
||||
|
||||
get_template_by_id_response = {
|
||||
"$schema": "http://json-schema.org/draft-04/schema#",
|
||||
"description": "GET template by id schema response",
|
||||
"type": "object",
|
||||
"title": "reponse v2/template",
|
||||
"properties": {
|
||||
"id": uuid,
|
||||
"type": {"enum": TEMPLATE_TYPES},
|
||||
"created_at": {
|
||||
"format": "date-time",
|
||||
"type": "string",
|
||||
"description": "Date+time created"
|
||||
},
|
||||
"updated_at": {
|
||||
"format": "date-time",
|
||||
"type": "string",
|
||||
"description": "Date+time updated"
|
||||
},
|
||||
"created_by": {"type": "string"},
|
||||
"version": {"type": "integer"},
|
||||
"body": {"type": "string"},
|
||||
"subject": {"type": ["string", "null"]}
|
||||
},
|
||||
"required": ["id", "type", "created_at", "updated_at", "version", "created_by", "body"]
|
||||
}
|
||||
0
tests/app/v2/template/__init__.py
Normal file
0
tests/app/v2/template/__init__.py
Normal file
93
tests/app/v2/template/test_get_template.py
Normal file
93
tests/app/v2/template/test_get_template.py
Normal file
@@ -0,0 +1,93 @@
|
||||
import pytest
|
||||
|
||||
from flask import json
|
||||
|
||||
from app import DATETIME_FORMAT
|
||||
from tests import create_authorization_header
|
||||
from tests.app.conftest import sample_template as create_sample_template
|
||||
|
||||
EMAIL_TYPE = 'email'
|
||||
SMS_TYPE = 'sms'
|
||||
LETTER_TYPE = 'letter'
|
||||
|
||||
template_types = [EMAIL_TYPE, SMS_TYPE, LETTER_TYPE]
|
||||
valid_version_params = [None, 1]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("tmp_type", template_types)
|
||||
@pytest.mark.parametrize("version", valid_version_params)
|
||||
def test_get_email_template_by_id_returns_200(client, notify_db, notify_db_session, sample_service, tmp_type, version):
|
||||
template = create_sample_template(notify_db, notify_db_session, template_type=tmp_type)
|
||||
auth_header = create_authorization_header(service_id=sample_service.id)
|
||||
|
||||
version_path = '/version/{}'.format(version) if version else ''
|
||||
|
||||
response = client.get(path='/v2/template/{}{}'.format(template.id, version_path),
|
||||
headers=[('Content-Type', 'application/json'), auth_header])
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.headers['Content-type'] == 'application/json'
|
||||
|
||||
json_response = json.loads(response.get_data(as_text=True))
|
||||
|
||||
expected_response = {
|
||||
'id': '{}'.format(template.id),
|
||||
'type': '{}'.format(template.template_type),
|
||||
'created_at': template.created_at.strftime(DATETIME_FORMAT),
|
||||
'updated_at': None,
|
||||
'version': template.version,
|
||||
'created_by': template.created_by.email_address,
|
||||
'body': template.content,
|
||||
"subject": template.subject if tmp_type == EMAIL_TYPE else None
|
||||
}
|
||||
|
||||
assert json_response == expected_response
|
||||
|
||||
|
||||
def test_get_template_with_invalid_template_id_returns_404(client, sample_service):
|
||||
auth_header = create_authorization_header(service_id=sample_service.id)
|
||||
|
||||
invalid_template_id = 'some_other_id'
|
||||
|
||||
response = client.get(path='/v2/template/{}'.format(invalid_template_id),
|
||||
headers=[('Content-Type', 'application/json'), auth_header])
|
||||
|
||||
assert response.status_code == 404
|
||||
assert response.headers['Content-type'] == 'application/json'
|
||||
|
||||
json_response = json.loads(response.get_data(as_text=True))
|
||||
|
||||
assert json_response == {
|
||||
"message": "The requested URL was not found on the server. "
|
||||
"If you entered the URL manually please check your spelling and try again.",
|
||||
"result": "error"
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize("tmp_type", template_types)
|
||||
def test_get_template_with_invalid_version_returns_404(client, notify_db, notify_db_session, sample_service, tmp_type):
|
||||
template = create_sample_template(
|
||||
notify_db, notify_db_session, template_type=tmp_type)
|
||||
|
||||
auth_header = create_authorization_header(service_id=sample_service.id)
|
||||
|
||||
# test with version number beyond latest version
|
||||
invalid_version = template.version + 1
|
||||
|
||||
response = client.get(path='/v2/template/{}/version/{}'.format(template.id, invalid_version),
|
||||
headers=[('Content-Type', 'application/json'), auth_header])
|
||||
|
||||
assert response.status_code == 404
|
||||
assert response.headers['Content-type'] == 'application/json'
|
||||
|
||||
json_response = json.loads(response.get_data(as_text=True))
|
||||
|
||||
assert json_response == {
|
||||
"errors": [
|
||||
{
|
||||
"error": "NoResultFound",
|
||||
"message": "No result found"
|
||||
}
|
||||
],
|
||||
"status_code": 404
|
||||
}
|
||||
67
tests/app/v2/template/test_template_schemas.py
Normal file
67
tests/app/v2/template/test_template_schemas.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import uuid
|
||||
|
||||
import pytest
|
||||
from flask import json
|
||||
|
||||
from app.v2.template.template_schemas import (
|
||||
get_template_by_id_response,
|
||||
get_template_by_id_request
|
||||
)
|
||||
from app.schema_validation import validate
|
||||
from jsonschema.exceptions import ValidationError
|
||||
|
||||
|
||||
valid_json = {
|
||||
'id': str(uuid.uuid4()),
|
||||
'type': 'email',
|
||||
'created_at': '2017-01-10T18:25:43.511Z',
|
||||
'updated_at': '2017-04-23T18:25:43.511Z',
|
||||
'version': 1,
|
||||
'created_by': 'someone@test.com',
|
||||
'body': "some body"
|
||||
}
|
||||
|
||||
valid_json_with_optionals = {
|
||||
'id': str(uuid.uuid4()),
|
||||
'type': 'email',
|
||||
'created_at': '2017-01-10T18:25:43.511Z',
|
||||
'updated_at': '2017-04-23T18:25:43.511Z',
|
||||
'version': 1,
|
||||
'created_by': 'someone',
|
||||
'body': "some body",
|
||||
'subject': "some subject"
|
||||
}
|
||||
|
||||
valid_request_args = [
|
||||
{"id": str(uuid.uuid4()), "version": 1}, {"id": str(uuid.uuid4())}]
|
||||
|
||||
invalid_request_args = [
|
||||
({"id": str(uuid.uuid4()), "version": "test"}, ["version test is not of type integer, null"]),
|
||||
({"id": str(uuid.uuid4()), "version": 0}, ["version 0 is less than the minimum of 1"]),
|
||||
({"version": 1}, ["id is a required property"]),
|
||||
({"id": "invalid_uuid"}, ["id is not a valid UUID"]),
|
||||
({"id": "invalid_uuid", "version": 0}, ["version 0 is less than the minimum of 1",
|
||||
"id is not a valid UUID"])
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("args", valid_request_args)
|
||||
def test_get_template_request_schema__against_valid_args_is_valid(args):
|
||||
assert validate(args, get_template_by_id_request) == args
|
||||
|
||||
|
||||
@pytest.mark.parametrize("args,error_message", invalid_request_args)
|
||||
def test_get_template_request_schema_against_invalid_args_is_invalid(args, error_message):
|
||||
with pytest.raises(ValidationError) as e:
|
||||
validate(args, get_template_by_id_request)
|
||||
errors = json.loads(str(e.value))
|
||||
|
||||
assert errors['status_code'] == 400
|
||||
|
||||
for error in errors['errors']:
|
||||
assert error['message'] in error_message
|
||||
|
||||
|
||||
@pytest.mark.parametrize("response", [valid_json, valid_json_with_optionals])
|
||||
def test_get_template_response_schema_is_valid(response):
|
||||
assert validate(response, get_template_by_id_response) == response
|
||||
Reference in New Issue
Block a user