notifications-api/tests/app/public_contracts/__init__.py

import os

import jsonschema
from flask import json
from jsonschema import Draft7Validator


def return_json_from_response(response):
    return json.loads(response.get_data(as_text=True))


def validate_v0(json_to_validate, schema_filename):
    schema_dir = os.path.join(os.path.dirname(__file__), "schemas/v0")
    resolver = jsonschema.RefResolver("file://" + schema_dir + "/", None)
    with open(os.path.join(schema_dir, schema_filename)) as schema:
        jsonschema.validate(
            json_to_validate,
            json.load(schema),
            format_checker=jsonschema.FormatChecker(),
            resolver=resolver,
        )


def validate(json_to_validate, schema):
    validator = Draft7Validator(schema)
    validator.validate(json_to_validate, schema)