hurl/integration/check_json_output.py

#!/usr/bin/env python3
import json
import sys


def check(expected, actual):
    expected_test = json.loads(expected)
    actual_test = json.loads(actual)
    check_output(expected_test, actual_test)


def check_output(expected_test, actual_test):
    expected_entries = expected_test["entries"]
    actual_entries = actual_test["entries"]
    if len(expected_entries) != len(actual_entries):
        print("expected entries: %d" % len(expected_entries))
        print("actual entries: %d" % len(actual_entries))
        sys.exit(1)
    for i in range(len(expected_entries)):
        check_entry(i, expected_entries[i], actual_entries[i])


def check_entry(entry_index, expected_entry, actual_entry):
    check_request(entry_index, expected_entry["request"], actual_entry["request"])
    check_response(entry_index, expected_entry["response"], actual_entry["response"])
    if expected_entry.get("asserts") is not None:
        check_asserts(entry_index, expected_entry["asserts"], actual_entry["asserts"])


def check_request(entry_index, expected_request, actual_request):
    check_method(entry_index, expected_request["method"], actual_request["method"])
    check_url(entry_index, expected_request["url"], actual_request["url"])


def check_response(entry_index, expected_response, actual_response):
    check_status(entry_index, expected_response["status"], actual_response["status"])


def check_method(entry_index, expected, actual):
    if expected != actual:
        print("Invalid entry %d" % entry_index)
        print("expected method: %s" % expected)
        print("actual method  : %s" % actual)
        sys.exit(1)


def check_url(entry_index, expected, actual):
    if expected != actual:
        print("Invalid entry %d" % entry_index)
        print("expected url: %s" % expected)
        print("actual url  : %s" % actual)
        sys.exit(1)


def check_status(entry_index, expected, actual):
    if expected != actual:
        print("Invalid entry %d" % entry_index)
        print("expected status: %s" % expected)
        print("actual status  : %s" % actual)
        sys.exit(1)


def check_asserts(entry_index, expected, actual):
    if expected != actual:
        print("Invalid entry %d" % entry_index)
        print("expected asserts:\n%s" % json.dumps(expected, indent=2))
        print("actual asserts:\n%s" % json.dumps(actual, indent=2))
        sys.exit(1)


def main():
    if len(sys.argv) < 3:
        print("usage: check_output.py expected_file actual_file")
        sys.exit(1)
    expected_file = sys.argv[1]
    actual_file = sys.argv[2]
    expected = open(expected_file).read()
    actual = open(actual_file).read()
    check_file(expected, actual)


if __name__ == "__main__":
    main()