104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
from pathlib import Path
|
|
from typing import Callable, Optional, TypedDict
|
|
|
|
import pytest
|
|
from dotenv import dotenv_values
|
|
from icecream import ic
|
|
|
|
from src.dirmanager import DirManager
|
|
|
|
|
|
class SubTest(TypedDict):
|
|
condition: Callable[[Path], bool]
|
|
test_file: str
|
|
|
|
|
|
class Runner:
|
|
name: Optional[str] = None
|
|
test_dir_name: Optional[str] = None
|
|
main_setup_name: Optional[str] = None
|
|
main_test_name: Optional[str] = None
|
|
sub_tests: list[SubTest] = []
|
|
dependencies: list[str] = []
|
|
|
|
def __init__(self, dotenv_path: Path, output_dir: Path, session_id: str):
|
|
self.dotenv_path = dotenv_path
|
|
self.config: dict[str, str] = dotenv_values(dotenv_path) # type: ignore
|
|
self.output_dir = output_dir
|
|
self.session_id = session_id
|
|
self.dir_manager = DirManager(output_dir, session_id)
|
|
|
|
ic(f"creating instance of {self.__class__.__name__}")
|
|
assert self.test_dir_name is not None
|
|
self.root_dir = Path(__file__).parent
|
|
|
|
def _run_main_test(self):
|
|
if isinstance(self.main_setup_name, str):
|
|
full_path = self.root_dir / self.test_dir_name / self.main_setup_name
|
|
self._run_pytest(full_path)
|
|
if isinstance(self.main_test_name, str):
|
|
full_path = self.root_dir / self.test_dir_name / self.main_test_name
|
|
self._run_pytest(full_path)
|
|
|
|
def _run_pytest(self, full_test_path: Path):
|
|
"""runs pytest programmatically
|
|
|
|
will run all tests in the file at full_test_path with some command line arguments"""
|
|
|
|
ic(f"running test: {full_test_path}")
|
|
|
|
command_arguments = []
|
|
|
|
# command_arguments.append("-v")
|
|
# command_arguments.append("-rx")
|
|
command_arguments.append(str(full_test_path))
|
|
|
|
command_arguments.append("--env_file")
|
|
command_arguments.append(str(self.dotenv_path))
|
|
|
|
command_arguments.append("--output_dir")
|
|
command_arguments.append(str(self.output_dir))
|
|
|
|
command_arguments.append("--session_id")
|
|
command_arguments.append(self.session_id)
|
|
|
|
# artifacts dir
|
|
output = self.dir_manager.RESULTS / full_test_path.stem
|
|
command_arguments.append("--output")
|
|
command_arguments.append(str(output))
|
|
|
|
# tracing
|
|
command_arguments.append("--tracing")
|
|
# command_arguments.append("retain-on-failure")
|
|
command_arguments.append("on")
|
|
# warning: https://github.com/microsoft/playwright-pytest/issues/111
|
|
# --output only works with the given context and page fixture
|
|
|
|
# Disable capturing. With -s set, prints will go to console as if pytest is not there.
|
|
# command_arguments.append("-s")
|
|
|
|
# headed
|
|
# command_arguments.append("--headed")
|
|
|
|
pytest.main(command_arguments)
|
|
|
|
def run_tests(self):
|
|
self._check_dependencies()
|
|
self._run_main_test()
|
|
for sub_test in self.sub_tests:
|
|
condition_function = sub_test["condition"]
|
|
if condition_function(self.dotenv_path):
|
|
test_name = sub_test["test_file"]
|
|
full_test_path = self.root_dir / self.test_dir_name / test_name
|
|
self._run_pytest(full_test_path)
|
|
self._create_complete_file()
|
|
|
|
def _create_complete_file(self):
|
|
file_path = self.dir_manager.RESULTS / self.name
|
|
with open(file_path, "w") as _:
|
|
pass # pass does nothing, so the file remains empty
|
|
|
|
def _check_dependencies(self):
|
|
for dependencie in self.dependencies:
|
|
ic(f"checking {dependencie}")
|
|
assert dependencie in self.dir_manager.RESULTS.glob("*")
|