tests.py

#!/usr/bin/env python3
"""Checks that the peering details in this repo conform to a few basic rules."""
import json
import os
import sys

REQUIRED_FIELDS = ['publicKey', 'password', 'contact']
RECOMMENDED_FIELDS = ['gpg']

RED = '\x1b[01;31m'
GREEN = '\x1b[01;32m'
YELLOW = '\x1b[01;33m'
END = '\x1b[0m'


def validate(path):
    """Test a single set of peering creds."""
    print("Validating %s" % path)
    try:
        creds = open(path).read()
        peers = json.loads("{%s}" % creds)
        hosts = peers.keys()
        warning = False
        for host in hosts:
            for field in REQUIRED_FIELDS:
                if field not in peers[host]:
                    print("    %sHost %s is missing the required field %s%s" % (RED, host,
                                                                                field, END))
                    return False
            for field in RECOMMENDED_FIELDS:
                if field not in peers[host]:
                    warning = True
                    print("    %sHost %s is missing the recommended field %s%s" % (YELLOW, host,
                                                                                   field, END))
        if warning:
            print("    %sSuccess, but missing recommended fields%s" % (YELLOW, END))
        else:
            print("    %sSuccess!%s" % (YELLOW, END))
        return True
    except ValueError:
        print("    %sInvalid JSON!%s" % (RED, END))
        return False

if __name__ == "__main__":
    success = True
    if len(sys.argv) == 2:
        success = validate(sys.argv[1])
    else:
        for directory, subdirs, files in os.walk('.'):
            if len(files) > 0:
                for f in files:
                    if f.endswith('.k'):
                        result = validate("%s/%s" % (directory, f))
                        if not result:
                            success = False
    if not success:
        sys.exit(1)