#!/usr/bin/env python
#
# Copyright (C) 2011 Igalia S.L.
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Library General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library 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
# Library General Public License for more details.
#
# You should have received a copy of the GNU Library General Public License
# along with this library; see the file COPYING.LIB.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.

from webkitpy.common.system.executive import Executive
import subprocess
import os, sys

class TestRunner:

    TEST_DIRS = [ "unittests", "WebKit2APITests" ]
    SKIPPED = [ "unittests/testdownload" ]

    def __init__(self):
        self._executive = Executive()

        # FIXME: webkit-build-directory --configuration always returns
        # Release because we never call set-webkit-configuration.
        #build_directory_script = os.path.join(os.path.dirname(__file__), "webkit-build-directory")
        #build_directory = self._executive.run_command([build_directory_script, "--configuration"]).rstrip()

        def is_valid_build_directory(build_dir):
            return os.path.exists(os.path.join(build_dir, ".libs"))

        script_dir = os.path.dirname(__file__)
        top_level = os.path.normpath(os.path.join(script_dir, "..", ".."))
        build_directory = os.path.join(top_level, 'WebKitBuild', 'Release')
        if not is_valid_build_directory(build_directory):
            build_directory = os.path.join(top_level, 'WebKitBuild', 'Debug')

        self._programs_path = os.path.join(build_directory, "Programs")
        self._tests = []
        for test_dir in self.TEST_DIRS:
            for test_file in os.listdir(os.path.join(self._programs_path, test_dir)):
                test_relative_path = os.path.join(test_dir, test_file)
                if test_relative_path in self.SKIPPED:
                    sys.stdout.write("Skipping test %s\n" % (test_relative_path))
                    sys.stdout.flush()
                    continue

                test_path = os.path.join(self._programs_path, test_relative_path)
                if os.path.isfile(test_path) and os.access(test_path, os.X_OK):
                    self._tests.append(test_path)

    def run(self):
        if not self._tests:
            sys.stderr.write("ERROR: tests not found in %s.\n" % (self._programs_path))
            sys.stderr.flush()
            return 1

        test_env = os.environ
        test_env["DISPLAY"] = ":31"

        exit_status = [0]
        def _error_handler(error):
            exit_status[0] = error.exit_code

        for test in self._tests:
            out = self._executive.run_command(['gtester', test], env=test_env,
                                              error_handler=_error_handler)
            sys.stdout.write(out)
            sys.stdout.flush()

        if exit_status[0]:
            sys.stdout.write("Tests failed\n")
            sys.stdout.flush()

        return exit_status[0]

if __name__ == "__main__":
    try:
        xvfb = Executive().popen(["Xvfb", ":31", "-screen", "0", "800x600x24", "-nolisten", "tcp"],
                                 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    except:
        sys.stderr.write("Failed to run Xvfb\n")
        sys.stderr.flush()
        sys.exit(1)

    try:
        sys.exit(TestRunner().run())
    finally:
        xvfb.kill()