# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.

import argparse
import os.path
import sys

import pytest


TEST_ROOT = os.path.dirname(__file__)
SRC_ROOT = os.path.dirname(TEST_ROOT)
PROJECT_ROOT = os.path.dirname(SRC_ROOT)
IPYTHON_ROOT = os.path.join(SRC_ROOT, 'ipython')
TESTING_TOOLS_ROOT = os.path.join(SRC_ROOT, 'testing_tools')


def parse_args():
    parser = argparse.ArgumentParser()
    # To mark a test as functional:  (decorator) @pytest.mark.functional
    parser.add_argument('--functional', dest='markers',
                        action='append_const', const='functional')
    parser.add_argument('--no-functional', dest='markers',
                        action='append_const', const='not functional')
    args, remainder = parser.parse_known_args()

    ns = vars(args)

    return ns, remainder


def main(pytestargs, markers=None):
    sys.path.insert(1, IPYTHON_ROOT)
    sys.path.insert(1, TESTING_TOOLS_ROOT)

    pytestargs = [
        '--rootdir', SRC_ROOT,
        TEST_ROOT,
        ] + pytestargs
    for marker in reversed(markers or ()):
        pytestargs.insert(0, marker)
        pytestargs.insert(0, '-m')

    ec = pytest.main(pytestargs)
    return ec


if __name__ == '__main__':
    mainkwargs, pytestargs = parse_args()
    ec = main(pytestargs, **mainkwargs)
    sys.exit(ec)