# Copyright (c) 2016 Red Hat, Inc
#
# This file is part of ttrun
#
# ttrun 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.
#
# ttrun 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 Ansible.  If not, see <http://www.gnu.org/licenses/>.

import argparse
import subprocess
import sys

import testtools


def parse_arguments():
    parser = argparse.ArgumentParser(
        description='Simple CLI to run tests with testtools')
    parser.add_argument(
        '-e', dest='environment', help='tox environment to use')
    parser.add_argument('tests', nargs='*', help='Tests to run')
    return parser.parse_args()


def main():
    args = parse_arguments()

    if args.environment:
        return subprocess.call([
            '.tox/{environment}/bin/python'.format(
                environment=args.environment),
            '-m',
            'testtools.run'] + args.tests)
    else:
        return testtools.run.main([sys.argv[0]] + args.tests, sys.stdout)


if __name__ == '__main__':
    sys.exit(main())