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