kewpie.py

#! /usr/bin/env python
# -*- mode: python; indent-tabs-mode: nil; -*-
# vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
#
# Copyright (C) 2010 Patrick Crews
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA


""" kewpie.py

(DataBase) Quality Platform - system for executing various
testing systems and the helper code 

Designed to be a modular test-runner.  Different testing tools
and databases may be plugged into the system via hacking the
appropriate modules

Currently geared towards Drizzle and MySQL systems
Runs: Drizzle, MySQL, Percona Server, Galera dbms's
"""

# imports
import os
import imp
import sys

import lib.opts.test_run_options as test_run_options
from lib.opts.defaults import get_defaults
from lib.modes.test_mode import handle_mode
from lib.server_mgmt.server_management import serverManager
from lib.sys_mgmt.system_management import systemManager
from lib.test_mgmt.execution_management import executionManager
from lib.opts.matrix_manager import matrixManager
from lib.util.xb_manager import xtrabackupManager

# functions
def handle_sys_config(input_args, defaults):
    """ Look for / update defaults based on sys_config file
        if specified

    """
    key_string = '--sys-config'
    module_file = None
    for input_arg in input_args:
        if input_arg.startswith(key_string):
            module_file = input_arg.split(key_string)[1].replace('=','').strip()
            module_file = os.path.abspath(module_file)
            break
    if module_file:
        module_parent = os.path.dirname(module_file)
        sys.path.append(module_parent) 
        module_name = os.path.basename(module_file).replace('.py','')
        project = imp.load_source(module_name, module_file)
        defaults = project.get_project_defaults(module_file, defaults)
    return defaults

# main
# We base / look for a lot of things based on the location of
# the kewpie.py file
qp_rootdir = os.path.dirname(os.path.abspath(sys.argv[0]))
#project_name = 'percona-xtradb-cluster'
#project_name = 'xtrabackup'
project_name = None
defaults = get_defaults(qp_rootdir,project_name)
defaults = handle_sys_config(sys.argv, defaults)
variables = test_run_options.parse_qp_options(defaults)
variables['qp_root'] = qp_rootdir
system_manager = None
server_manager = None
test_manager = None
test_executor = None
execution_manager = None

try:
        # Instantinate option matrix
        matrix_manager = matrixManager(variables)
        # Some system-level work is constant regardless
        # of the test to be run
        system_manager = systemManager(variables)

        # Create our server_manager
        server_manager = serverManager(system_manager, variables)

        # Create XtraBackup manager
        xtrabackup_manager = xtrabackupManager(system_manager, server_manager, variables)
        # Get our mode-specific test_manager and test_executor
        (test_manager,test_executor) = handle_mode(variables, system_manager)

        # Gather our tests for execution
        test_manager.gather_tests()

        # Initialize test execution manager
        execution_manager = executionManager(server_manager, system_manager
                                        , test_manager, test_executor
                                        , variables, matrix_manager, xtrabackup_manager)

        # Execute our tests!
        execution_manager.execute_tests()
    
except Exception, e:
       print Exception, e

except KeyboardInterrupt:
      print "\n\nDetected <Ctrl>+c, shutting down and cleaning up..."

finally:
# TODO - make a more robust cleanup
# At the moment, runaway servers are our biggest concern
    if server_manager and not variables['startandexit']:
        if variables['gdb']:
            server_manager.cleanup_all_servers()
        else:
            server_manager.cleanup()
    if not variables['startandexit']:
        if test_manager:
            fail_count = test_manager.has_failing_tests()
            sys.exit(test_manager.has_failing_tests())
        else:
            # return 1 as we likely have a problem if we don't have a
            # test_manager
            sys.exit(1)