conftest.py

import pytest

from src.main.allure_helpers import AllureCatchLogs

pytest_plugins = ['fixtures_browsers', 'fixtures_session']


def pytest_addoption(parser):
    """ Pytest option variables"""
    parser.addoption('--browser',
                     help=u'Which test browser?',
                     choices=['chrome', 'firefox'],
                     default='chrome')
    parser.addoption('--remote',
                     help=u'Is remote webdriver?',
                     default='')


@pytest.fixture(scope='session')
def t_browser(request):
    """  Test browser. Params: [chrome, opera, firefox].  """
    return request.config.getoption('--browser')


@pytest.fixture(scope='session')
def is_remote(request):
    """if browser is remote? Returns: True/False."""
    return bool(request.config.getoption('--remote'))


@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_setup():
    """ Allure hook """
    with AllureCatchLogs():
        yield


@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_call():
    """ Allure hook """
    with AllureCatchLogs():
        yield


@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_teardown():
    """ Allure hook """
    with AllureCatchLogs():
        yield