frontend/health/check_test_health.py

#!/usr/bin/python
#
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse, logging, os, subprocess, sys

THIS_DIR = os.path.dirname(__file__)
UTILS_DIR = os.path.abspath(os.path.join(THIS_DIR, os.pardir, os.pardir,
                                         'utils'))
TEST_IMPORTER = os.path.join(UTILS_DIR, 'test_importer.py')

# The list of scripts are passed as lists as this allows us to add arguments
# as the list is just passed to subprocess.call(). We expect the scripts to
# return 0 on success and a non-zero value on failure.

# Scripts that need to be ran first to do any preperation. E.g. update the
# database.
PREP_SCRIPTS = [
                    [TEST_IMPORTER]
               ]

COMPLETE_FAILURES = os.path.join(THIS_DIR, 'complete_failures.py')
PASSING_EXPERIMENTAL = os.path.join(THIS_DIR, 'passing_experimental.py')
# Scripts ran to do the analysis.
ANALYSIS_SCRIPTS = [
                        [COMPLETE_FAILURES],
                        [PASSING_EXPERIMENTAL]
                   ]


def run_prep_scripts(scripts):
    """
    Run the scripts that are required to be ran before the analysis.

    This stops and returns False at the first script failure.

    @param scripts: A list of lists. Where the inner list is the script name
        and arguments to be called (as subprocess.call() expects).

    @return True if all the scripts succeeded and False otherwise.

    """

    for script in scripts:
        logging.info('Running %s', ' '.join(script))
        return_code = subprocess.call(script)
        if return_code != 0:
            logging.error('\'%s\' failed with return code %d',
                          (' '.join(script), return_code))
            return False

    return True


def run_analysis_scripts(scripts):
    """
    Run the scripts that analyze the database.

    All scripts are ran even if one fails.

    @param scripts: A list of lists, where the inner list is the script name
        and arguments to be called (as subprocess.call() expects).

    @return True if all the scripts succeeded and False otherwise.

    """

    success = True

    for script in scripts:
        logging.info('Running %s', ' '.join(script))
        return_code = subprocess.call(script)
        if return_code != 0:
            logging.error('\'%s\' failed with return code %d',
                          (' '.join(script), return_code))
            success = False

    return success


def parse_options(args):
    """Parse the command line options."""

    description = ('Runs test health and preparation scripts.')
    parser = argparse.ArgumentParser(description=description)
    parser.parse_args(args)


def main(args=None):
    """
    The main function.

    This allows us to test this function by calling it in the unit test file.

    @param args: The command line arguments being passed in.

    @return 0 if everything succeeded and a non-zero integer otherwise.

    """
    args = [] if args is None else args
    parse_options(args)

    logging.getLogger().setLevel(logging.INFO)

    prep_success = run_prep_scripts(PREP_SCRIPTS)
    if not prep_success:
        return 1

    analysis_success = run_analysis_scripts(ANALYSIS_SCRIPTS)
    if not analysis_success:
        return 1

    return 0


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))