100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
# This file is registered as a pytest plugin, meaning it will automatically loaded.
|
|
# All fixtures in this file will be available without manual loading.
|
|
|
|
import os
|
|
import re
|
|
from imaplib import IMAP4_SSL
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from dotenv import dotenv_values
|
|
from playwright.sync_api import BrowserContext, expect
|
|
from pytest import Parser
|
|
|
|
from pytest_abra.dir_manager import DirManager
|
|
from pytest_abra.env_manager import EnvFile
|
|
from pytest_abra.utils import BaseUrl
|
|
|
|
|
|
def pytest_addoption(parser: Parser):
|
|
parser.addoption("--runner_index", action="store", type=int)
|
|
parser.addoption("--output_dir", action="store", type=Path)
|
|
parser.addoption("--session_id", action="store", type=str)
|
|
parser.addoption("--timeout", action="store", type=int, default=20_000)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def set_expect_timeout(request):
|
|
TIMEOUT = request.config.getoption("--timeout")
|
|
expect.set_options(timeout=TIMEOUT)
|
|
|
|
|
|
@pytest.fixture
|
|
def context(context: BrowserContext, request) -> BrowserContext:
|
|
# note: because this has the existing context fixture as an argument, it is ensured
|
|
# that the original fixture is called first and then overwritten by this custom one.
|
|
|
|
TIMEOUT = request.config.getoption("--timeout")
|
|
LOCALE = {"Accept-Language": "de_DE"}
|
|
|
|
context.set_default_timeout(TIMEOUT)
|
|
context.set_extra_http_headers(LOCALE)
|
|
return context
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def DIR(request) -> DirManager:
|
|
"""Fixture holding test directories
|
|
|
|
DIR.OUTPUT
|
|
DIR.SESSION
|
|
DIR.RECORDS
|
|
DIR.STATES
|
|
DIR.RESULTS"""
|
|
|
|
output_dir = request.config.getoption("--output_dir")
|
|
assert output_dir, "pytest argument --output_dir not set"
|
|
session_id = request.config.getoption("--session_id")
|
|
assert session_id, "pytest argument --session_id not set"
|
|
dirmanager = DirManager(output_dir=output_dir, session_id=session_id)
|
|
dirmanager.create_all_dirs()
|
|
return dirmanager
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def ENV_FILES(DIR: DirManager) -> dict[int, EnvFile]:
|
|
out: dict[int, EnvFile] = dict()
|
|
for env_path in DIR.ENV_FILES.glob("*.env"):
|
|
config: dict[str, str] = dotenv_values(env_path) # type: ignore
|
|
env_type = config["TYPE"]
|
|
result = re.search(r"(\d+)-*", env_path.name)
|
|
assert result
|
|
runner_index = int(result[1])
|
|
out[runner_index] = EnvFile(env_path=env_path, config=config, env_type=env_type)
|
|
return out
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def dotenv_config(request, ENV_FILES: dict[int, EnvFile]) -> dict[str, str]:
|
|
runner_index = request.config.getoption("--runner_index")
|
|
return ENV_FILES[runner_index].config
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def URL(dotenv_config: dict[str, str]) -> BaseUrl:
|
|
return BaseUrl(netloc=dotenv_config["DOMAIN"])
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def imap_ssl_email_client() -> None:
|
|
assert os.environ["IMAP_HOST"]
|
|
assert os.environ["IMAP_PORT"]
|
|
assert os.environ["IMAP_USER"]
|
|
assert os.environ["IMAP_PASS"]
|
|
port = int(os.environ["IMAP_PORT"])
|
|
imap_client = IMAP4_SSL(host=os.environ["IMAP_HOST"], port=port)
|
|
imap_client.login(os.environ["IMAP_USER"], os.environ["IMAP_PASS"])
|
|
imap_client.select("INBOX")
|
|
yield imap_client
|
|
imap_client.close()
|
|
imap_client.logout()
|