def test_get_Templates(notify_api, notify_db, notify_db_session, sample_service):
sample_template = create_sample_template(notify_db,
notify_db_session,
service=sample_service)
assert Template.query.count() == 1
assert len(get_model_templates()) == 1
template_name = "Another Template"
sample_template = create_sample_template(notify_db,
notify_db_session,
template_name=template_name,
service=sample_service)
assert Template.query.count() == 2
assert len(get_model_templates()) == 2
def test_get_Templates(notify_api, notify_db, notify_db_session,
sample_service):
sample_template = create_sample_template(notify_db,
notify_db_session,
service=sample_service)
assert Template.query.count() == 1
assert len(get_model_templates()) == 1
template_name = "Another Template"
sample_template = create_sample_template(notify_db,
notify_db_session,
template_name=template_name,
service=sample_service)
assert Template.query.count() == 2
assert len(get_model_templates()) == 2
def update_template(service_id, template_id):
try:
service = get_model_services(service_id=service_id)
except DataError:
return jsonify(result="error", message="Invalid service id"), 400
except NoResultFound:
return jsonify(result="error", message="Service not found"), 404
try:
template = get_model_templates(template_id=template_id)
except DataError:
return jsonify(result="error", message="Invalid template id"), 400
except NoResultFound:
return jsonify(result="error", message="Template not found"), 404
if request.method == 'DELETE':
status_code = 202
delete_model_template(template)
else:
status_code = 200
# TODO there has got to be a better way to do the next three lines
upd_temp, errors = template_schema.load(request.get_json())
if errors:
return jsonify(result="error", message=errors), 400
upd_temp.service = service
update_dict, errors = template_schema.dump(upd_temp)
# TODO FIX ME
# Remove update_temp model which is added to db.session
db.session.rollback()
try:
save_model_template(template, update_dict=update_dict)
except DAOException as e:
return jsonify(result="error", message=str(e)), 400
return jsonify(data=template_schema.dump(template).data), status_code
def update_template(service_id, template_id):
try:
service = get_model_services(service_id=service_id)
except DataError:
return jsonify(result="error", message="Invalid service id"), 400
except NoResultFound:
return jsonify(result="error", message="Service not found"), 404
try:
template = get_model_templates(template_id=template_id)
except DataError:
return jsonify(result="error", message="Invalid template id"), 400
except NoResultFound:
return jsonify(result="error", message="Template not found"), 404
if request.method == 'DELETE':
status_code = 202
delete_model_template(template)
else:
status_code = 200
# TODO there has got to be a better way to do the next three lines
upd_temp, errors = template_schema.load(request.get_json())
if errors:
return jsonify(result="error", message=errors), 400
upd_temp.service = service
update_dict, errors = template_schema.dump(upd_temp)
# TODO FIX ME
# Remove update_temp model which is added to db.session
db.session.rollback()
try:
save_model_template(template, update_dict=update_dict)
except DAOException as e:
return jsonify(result="error", message=str(e)), 400
return jsonify(data=template_schema.dump(template).data), status_code
def test_get_user_Template(notify_api, notify_db, notify_db_session, sample_service):
assert Template.query.count() == 0
template_name = "Random Template"
sample_template = create_sample_template(notify_db,
notify_db_session,
template_name=template_name,
service=sample_service)
assert get_model_templates(template_id=sample_template.id).name == template_name
assert Template.query.count() == 1
def test_get_user_Template(notify_api, notify_db, notify_db_session,
sample_service):
assert Template.query.count() == 0
template_name = "Random Template"
sample_template = create_sample_template(notify_db,
notify_db_session,
template_name=template_name,
service=sample_service)
assert get_model_templates(
template_id=sample_template.id).name == template_name
assert Template.query.count() == 1
def get_template(template_id=None):
try:
templates = get_model_templates(template_id=template_id)
except DataError:
return jsonify(result="error", message="Invalid template id"), 400
except NoResultFound:
return jsonify(result="error", message="Template not found"), 404
if isinstance(templates, list):
data, errors = templates_schema.dump(templates)
else:
data, errors = template_schema.dump(templates)
if errors:
return jsonify(result="error", message=str(errors))
return jsonify(data=data)
def get_service_template(service_id, template_id=None):
try:
templates = get_model_templates(service_id=service_id,
template_id=template_id)
except DataError:
return jsonify(result="error", message="Invalid template id"), 400
except NoResultFound:
return jsonify(result="error", message="Template not found"), 404
if isinstance(templates, list):
data, errors = templates_schema.dump(templates)
else:
data, errors = template_schema.dump(templates)
if errors:
return jsonify(result="error", message=str(errors))
return jsonify(data=data)
def validate_template(json_body, service_id):
errors = {"template": []}
template_id = json_body.get('template', None)
content = ''
if not template_id:
errors['template'].append('Required data missing')
else:
if service_id == current_app.config.get('ADMIN_CLIENT_USER_NAME'):
content = json_body['template']
else:
try:
template = templates_dao.get_model_templates(
template_id=json_body['template'], service_id=service_id)
content = template.content
except:
errors['template'].append("Unable to load template.")
return content, errors
def validate_template(json_body, service_id):
errors = {"template": []}
template_id = json_body.get('template', None)
content = ''
if not template_id:
errors['template'].append('Required data missing')
else:
if service_id == current_app.config.get('ADMIN_CLIENT_USER_NAME'):
content = json_body['template']
else:
try:
template = templates_dao.get_model_templates(
template_id=json_body['template'],
service_id=service_id)
content = template.content
except:
errors['template'].append("Unable to load template.")
return content, errors
