# Run with something like
#
#   env GN2_PROFILE=/home/wrk/opt/gn-latest ./bin/genenetwork2 ./etc/default_settings.py -c ../test/requests/test-website.py http://localhost:5003
#
# Mostly to pick up the Guix GN2_PROFILE and python modules

import argparse
from link_checker import check_links
from link_checker import check_packaged_js_files
from mapping_tests import check_mapping
from navigation_tests import check_navigation
from correlation_tests import check_correlations
from main_web_functionality import check_main_web_functionality
import link_checker
import sys

# Imports for integration tests
from unittest import TestSuite, TextTestRunner, TestLoader

print("Mechanical Rob firing up...")

def host_is_online(host):
    import time
    import requests
    for count in range(1, 5):
        try:
            time.sleep(count)
            requests.get(host)
            return True
        except Exception as cre:
            print(f"Retrying in {count + 1} seconds ...")

    return False

def run_all(args_obj, parser):
    print("")
    print("Running all tests.")
    print(args_obj)
    assert host_is_online(args_obj.host), f"Could not connect to {host}"
    link_checker.DO_FAIL = args_obj.fail
    check_main_web_functionality(args_obj, parser)
    check_links(args_obj, parser)
    check_packaged_js_files(args_obj, parser)
    check_mapping(args_obj, parser)
    check_correlations(args_obj, parser)
    # TODO: Add other functions as they are created.


def print_help(args_obj, parser):
    print(parser.format_help())


def dummy(args_obj, parser):
    print("Not implemented yet.")


def initTest(klass, gn2_url, es_url):
    loader = TestLoader()
    methodNames = loader.getTestCaseNames(klass)
    return [klass(mname, gn2_url, es_url) for mname in methodNames]


desc = """
This is Mechanical-Rob - an automated web server tester for
                         Genenetwork.org
"""
parser = argparse.ArgumentParser(description=desc)

parser.add_argument("--fail", help="Fail and stop on any error", action="store_true")

parser.add_argument(
    "-d",
    "--database",
    metavar="DB",
    type=str,
    default="db_webqtl_s",
    help="Use database (default db_webqtl_s)",
)

parser.add_argument(
    "host",
    metavar="HOST",
    type=str,
    default="http://localhost:5003",
    help="The url to the web server",
)

parser.add_argument(
    "-a",
    "--all",
    dest="accumulate",
    action="store_const",
    const=run_all,
    default=print_help,
    help="Runs all tests.",
)

parser.add_argument(
    "-l",
    "--link-checker",
    dest="accumulate",
    action="store_const",
    const=check_links,
    default=print_help,
    help="Checks for dead links.",
)

parser.add_argument(
    "-f",
    "--main-functionality",
    dest="accumulate",
    action="store_const",
    const=check_main_web_functionality,
    default=print_help,
    help="Checks for main web functionality.",
)

parser.add_argument(
    "-m",
    "--mapping",
    dest="accumulate",
    action="store_const",
    const=check_mapping,
    default=print_help,
    help="Checks for mapping.",
)


args = parser.parse_args()


args.accumulate(args, parser)

@@ -3099,16 +3099,227 @@
 g.%22,%0A)%0A%0A
+parser.add_argument(%0A    %22-c%22,%0A    %22--correlations%22,%0A    dest=%22accumulate%22,%0A    action=%22store_const%22,%0A    const=check_correlations,%0A    default=print_help,%0A    help=%22Checks that correlations run correctly.%22,%0A)%0A%0A
 %0Aargs = 

Enable running correlation tests alone.