test_standards_and_policies.py

import json

import jsonschema
from compendium_v2.routes.standards_and_policies import SECURITY_CONTROLS_RESPONSE_SCHEMA, \
    CRISIS_EXERCISES_RESPONSE_SCHEMA, STANDARDS_RESPONSE_SCHEMA


def test_standards_response(client, test_standards_data):
    rv = client.get(
        '/api/standards-and-policies/standards',
        headers={'Accept': ['application/json']})
    assert rv.status_code == 200
    result = json.loads(rv.data.decode('utf-8'))
    jsonschema.validate(result, STANDARDS_RESPONSE_SCHEMA)
    assert result


def test_crisis_exercises_response(client, test_crisis_exercises_data):
    rv = client.get(
        '/api/standards-and-policies/crisis-exercises',
        headers={'Accept': ['application/json']})
    assert rv.status_code == 200
    result = json.loads(rv.data.decode('utf-8'))
    jsonschema.validate(result, CRISIS_EXERCISES_RESPONSE_SCHEMA)
    assert result


def test_security_controls_response(client, test_security_controls_data):
    rv = client.get(
        '/api/standards-and-policies/security-controls',
        headers={'Accept': ['application/json']})
    assert rv.status_code == 200
    result = json.loads(rv.data.decode('utf-8'))
    jsonschema.validate(result, SECURITY_CONTROLS_RESPONSE_SCHEMA)
    assert result