#!/usr/bin/python3

# Authors:
#   Petr Viktorin <pviktori@redhat.com>
#   Jason Gerard DeRose <jderose@redhat.com>
#
# Copyright (C) 2008-2013  Red Hat
# see file 'COPYING' for use and warranty information
#
# 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 3 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, see <http://www.gnu.org/licenses/>.

"""Pytest wrapper for running an installed (not in-tree) IPA test suite

Any command-line arguments are passed directly to py.test.
The current directory is changed to the locaition of the ipatests package,
so any relative paths given will be based on the ipatests module's path
"""

import os
import copy
import sys

import pytest

import ipatests

# This is set to store --with-xunit report in an accessible place:
os.environ['IPATEST_XUNIT_PATH'] = os.path.join(os.getcwd(), 'nosetests.xml')

HERE = os.path.dirname(os.path.abspath(ipatests.__file__))


class ArgsManglePlugin(object):
    """Modify pytest arguments

    * Add confcutdir if hasn't been set already
    * Mangle paths to support tests both relative to basedir and ipatests/
    * Default to ipatests/ if no tests are specified
    """
    def pytest_load_initial_conftests(self, early_config, parser, args):
        # During initial loading, parser supports only basic options
        ns = early_config.known_args_namespace
        if not ns.confcutdir:
            # add config cut directory to only load fixtures from ipatests/
            args.insert(0, '--confcutdir={}'.format(HERE))

        if not ns.file_or_dir:
            # No file or directory found, run all tests
            args.append(HERE)
        else:
            for name in ns.file_or_dir:
                idx = args.index(name)
                # split on pytest separator
                # ipatests/test_ipaplatform/test_importhook.py::test_override
                filename, sep, suffix = name.partition('::')
                # a file or directory relative to cwd or already absolute
                if os.path.exists(filename):
                    continue
                # a file or directory relative to ipatests package
                args[idx] = sep.join((os.path.join(HERE, filename), suffix))

        # replace ignores, e.g. "--ignore test_integration" is changed to
        # "--ignore path/to/ipatests/test_integration"
        if ns.ignore:
            for ignore in ns.ignore:
                idx = args.index(ignore)
                if os.path.exists(ignore):
                    continue
                args[idx] = os.path.join(HERE, ignore)

        # rebuild early_config's known args with new args. The known args
        # are used for initial conftest.py from ipatests, which adds
        # additional arguments.
        early_config.known_args_namespace = parser.parse_known_args(
            args, namespace=copy.copy(early_config.option))


sys.exit(pytest.main(plugins=[ArgsManglePlugin()]))
