~camptocamp/openerp-command/7.0-oca_buildbot_fixes-afe

#! /usr/bin/env python

"""

runbot-job is a small script intended to perform tasks on OpenERP projects,

in particular testing.

"""

import argparse

from contextlib import contextmanager

import json

import os

import re

import subprocess

import time

 

install_base = False

install_all = False

javascript_tests = True

install_base_log_path = 'install-base.log'

install_all_log_path = 'install-all.log'

server_log_path = 'server.log'

javascript_tests_log_path = 'javascript_tests.log'

 

ORIGINAL_PATH = os.environ.get('PATH','').split(':')

 

def oe(env, log_path, command):

    if log_path is None:

        log_file = None

    else:

        log_file = open(log_path, 'w')

    p = subprocess.Popen(['oe'] + command.split(), stdout=log_file, stderr=log_file, close_fds=True, env=env)

    return p.wait()

 

def wait_until_server_is_ready():

    for _ in xrange(20):

        with open(server_log_path, 'r') as f:

            for line in f:

                if re.search('HTTP service (.*) running on', line):

                    return

        time.sleep(1)

 

def check_install_success(log_path):

    with open(log_path, 'r') as f:

        for line in f:

            if re.search('openerp.modules.loading: Modules loaded.', line):

                return True

    return False

 

@contextmanager

def server(args):

    env = env_from_args(args, 'base')

    log_file = open(server_log_path, 'w')

    cmd = 'python2 %s/openerp-server --addons-path %s --pidfile server-pid.txt' % \

        (args.server_path, env['OPENERP_ADDONS'].replace(':', ','))

    p = subprocess.Popen(cmd.split(), stdout=log_file, stderr=log_file, close_fds=True)

    server_pid = p.pid

    wait_until_server_is_ready() # TODO will blow if the server failed to start

    yield p

    try:

        os.kill(server_pid, 9)

    except OSError:

        pass

 

def env_from_args(args, database_suffix):

    paths = [args.server_path, args.command_path]

    addons = [args.addons_path, os.path.join(args.web_path, 'addons'), os.path.join(args.server_path, 'openerp/tests/addons')]

    e = {

        'PYTHONPATH': ':'.join(paths),

        'PATH': ':'.join(paths + ORIGINAL_PATH),

        'OPENERP_DATABASE': '%s_%s' % (args.prefix, database_suffix),

        'OPENERP_ADDONS': ':'.join(addons),

        'OPENERP_HOST': '127.0.0.1',

        'OPENERP_PORT': args.test_port,

    }

    return e

 

def run(args):

    results = {}

 

    if install_base:

        env = env_from_args(args, 'base')

        oe(env, os.path.join(args.prefix, install_base_log_path), 'initialize --tests')

        r = check_install_success(os.path.join(args.prefix, install_base_log_path))

        results['install_base'] = 0 if r else 1

 

    if install_all:

        env = env_from_args(args, 'all')

        oe(env, os.path.join(args.prefix, install_all_log_path), 'initialize --all-modules --exclude auth_ldap --tests')

        r = check_install_success(os.path.join(args.prefix, install_all_log_path))

        results['install_all'] = 0 if r else 1

 

    if javascript_tests:

        env = env_from_args(args, 'all')

        with server(args) as p:

            r = oe(env, os.path.join(args.prefix, javascript_tests_log_path), 'run-tests -d %s -m web.test_js' % env['OPENERP_DATABASE'])

            results['javascript_tests'] = r

 

    print json.dumps(results)

 

if __name__ == '__main__':

 

    parser = argparse.ArgumentParser(

        description=__doc__)

 

    parser.add_argument('--prefix', metavar='PREFIX',

        required=True,

        help='the prefix is used to create a working directory and to name databases')

    parser.add_argument('--server-path', metavar='PATH',

        required=True,

        help='path to the server library and executable')

    parser.add_argument('--addons-path', metavar='PATH',

        required=True,

        help='path to the addons')

    parser.add_argument('--web-path', metavar='PATH',

        required=True,

        help='path to the web addons')

    parser.add_argument('--command-path', metavar='PATH',

        required=True,

        help='path to the command library and executable')

    parser.add_argument('--test-port', metavar='PORT',

        required=True,

        help='port used to run the javascript tests')

 

    parser.set_defaults(run=run)

    args = parser.parse_args()

    args.run(args)