from __future__ import print_function, division, absolute_import

import os
import pathlib
import sys
from typing import Callable, List

# Hijack sys.path, so we don't have to use 'PYTHONPATH=src/' prefix
rootdir = pathlib.Path(__file__).parent.parent
sys.path.insert(0, str(rootdir / "syspurpose" / "src"))
sys.path.insert(0, str(rootdir / "src"))

from subscription_manager import ga_loader
ga_loader.init_ga()
from . import rhsm_display
rhsm_display.set_display()

import pytest

subman_marker_functional = pytest.mark.functional
subman_marker_zypper = pytest.mark.zypper
subman_marker_slow = pytest.mark.slow
# This allows us to set higher timeout limit for tests that are known to be slow
subman_marker_slow_timeout = pytest.mark.timeout(40)


def subman_marker_dbus(func: Callable) -> Callable:
    """
    Fork each D-Bus unit test into it's own process, because we use threads
    and mocking in these tests and such tests are not reliable without forking.
    :param func: function of D-Bus unit test
    :return: function of unit test marked as forked
    """
    return pytest.mark.dbus(pytest.mark.forked(func))


def subman_marker_needs_envvars(*envvars: List[str]) -> Callable:
    """Skip test if one or more environment variables are missing."""
    missing_vars: List[str] = [v for v in envvars if v not in os.environ]
    skip_func: Callable = pytest.mark.skipif(
        missing_vars,
        reason=f"Missing environment variables {', '.join(missing_vars)}."
    )
    return skip_func
