rival_regions_wrapper/tests/conftest.py

"""Test configuration"""

import os

import pytest
from dotenv import load_dotenv

from rival_regions_wrapper import LocalAuthentication, ApiWrapper


load_dotenv()


class MissingAuthenticationError(Exception):
    """Error for missing authentication"""


@pytest.fixture(scope='module')
def vcr(vcr):
    """Set parameters vor VCR"""
    vcr.ignore_localhost = True
    return vcr


@pytest.fixture(scope="module")
def api_wrapper():
    """Set up wrapper before test"""
    username = os.environ.get('RR_USERNAME', None)
    password = os.environ.get('RR_PASSWORD', None)
    login_method = os.environ.get('RR_LOGIN_METHOD', None)
    if None in (username, password, login_method):
        raise MissingAuthenticationError(
            'Load the following variables in your user environment: '
            'username, password, login_method'
        )
    authentication = LocalAuthentication(username, password, login_method)
    return ApiWrapper(authentication)