Fix determination of fixture parameters

- Added ability to pass any parameters to `params` keyword in `pytest.fixture`, except:
  1) `params=<callable object>`, but `params=[<callable object>, ...]` is allowed
  2) `params=[*<any collection>]`

- Analysis of `pytest_generate_tests` is almost equivalent to regular fixtures, except:
  1) `metafunc.parametrize(*<any collection>, **<any dict>)`
  2) `metafunc.parametrize("any", <callable object>)`
  3) `metafunc.parametrize("any1, any2", [(<callable object>, [*<any collection>])]`

Signed-off-by: Kirill Sosnovskikh <k.sosnovskikh@yadro.com>
This commit is contained in:
k.sosnovskikh 2024-11-15 15:30:43 +03:00
parent cd41eb5d03
commit d050a94272
9 changed files with 191 additions and 28 deletions

View file

@ -52,12 +52,12 @@ class TestIngoredClass: # noqa: allure-validator
### Options ### Options
The linter supports the following flags: The linter supports the following flags:
- `--plugins` - regex patterns that plugins must match (paths are compiled according to the template: `plugin / src / plugin`). - `--plugins` - regex patterns that plugins must match (paths are compiled according to the template: `plugin_name / src / plugin_name`).
- `--files` - paths to files with fixtures. - `--files` - paths to files with fixtures.
Configuration options can also be provided using special command line arguments, for example: Configuration options can also be provided using special command line arguments, for example:
```shell ```shell
allure-validator pytest_tests/ --plugins some[-_]plugin --files etc/workspace/some_plugin/src/some_plugin/fixtures.py allure-validator <tests folder> --plugins <plugin regex>, ... --files <path to file with fixtures>, ...
``` ```
## Work example ## Work example

View file

@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project] [project]
name = "allure-validator" name = "allure-validator"
version = "1.0.0" version = "1.1.1"
description = "Linter for allure.title validation" description = "Linter for allure.title validation"
readme = "README.md" readme = "README.md"
authors = [{ name = "YADRO", email = "info@yadro.com" }] authors = [{ name = "YADRO", email = "info@yadro.com" }]
@ -21,7 +21,7 @@ dependencies = [
"pytest>=7.1.2", "pytest>=7.1.2",
"pytest-lazy-fixture>=0.6.3" "pytest-lazy-fixture>=0.6.3"
] ]
requires-python = "==3.10.*" requires-python = ">=3.10"
[project.optional-dependencies] [project.optional-dependencies]
dev = ["black", "pylint", "isort", "pre-commit"] dev = ["black", "pylint", "isort", "pre-commit"]

View file

@ -187,16 +187,23 @@ def fixtures(ast_fixtures: list[ast.FunctionDef], ast_hooks: list[ast.FunctionDe
if parse.decorator(node.value) != PYTEST_METAFUNC_PARAMETRIZE: if parse.decorator(node.value) != PYTEST_METAFUNC_PARAMETRIZE:
continue continue
fixture_names = parse.param_names(node.value.args[0]) if len(node.value.args) < 2:
raise SyntaxError(f"{path}:{ast_hook.lineno}:{ast_hook.col_offset}: Unexpected syntax in {PYTEST_METAFUNC_PARAMETRIZE}")
ast_names, ast_params, *_ = node.value.args
fixture_names = parse.param_names(ast_names)
for name in fixture_names: for name in fixture_names:
# Dynamic fake fixture params = parse.metafunc_params(ast_params)
# Dynamic fixture
fixture = Fixture( fixture = Fixture(
name, name,
path, path,
node.lineno, node.lineno,
node.col_offset, node.col_offset,
args=[], args=[],
params=["parametrized"], params=params,
) )
fixtures[fixture] = fixture fixtures[fixture] = fixture

View file

@ -3,8 +3,10 @@ REPORTER_TITLE = "reporter.title"
PYTEST_FIXTURE = "pytest.fixture" PYTEST_FIXTURE = "pytest.fixture"
PYTEST_PARAMETRIZE = "pytest.mark.parametrize" PYTEST_PARAMETRIZE = "pytest.mark.parametrize"
PYTEST_PARAM = "pytest.param"
PYTEST_GENERATE_TESTS = "pytest_generate_tests" PYTEST_GENERATE_TESTS = "pytest_generate_tests"
PYTEST_METAFUNC_PARAMETRIZE = "metafunc.parametrize" PYTEST_METAFUNC_PARAMETRIZE = "metafunc.parametrize"
PYTEST_FORCED_PARAMETRIZATION = ["forced", "parametrized", "fixture"]
ATTR_PATH = "__func_path" ATTR_PATH = "__func_path"
ATTR_CLASS_NAME = "__class_name" ATTR_CLASS_NAME = "__class_name"
@ -14,5 +16,6 @@ VALIDATE_ERROR = 1
DIRECTORY_ERROR = 2 DIRECTORY_ERROR = 2
PLUGIN_ERROR = 3 PLUGIN_ERROR = 3
FILE_ERROR = 4 FILE_ERROR = 4
SYNTAX_ERROR = 5
ALLURE_VALIDATOR_IGNORE = "noqa: allure-validator" ALLURE_VALIDATOR_IGNORE = "noqa: allure-validator"

View file

@ -1,8 +1,6 @@
import sys
from allure_validator import collect, validate from allure_validator import collect, validate
from allure_validator.arguments import args from allure_validator.arguments import args
from allure_validator.common import DIRECTORY_ERROR, FILE_ERROR, PLUGIN_ERROR, VALIDATE_DONE, VALIDATE_ERROR from allure_validator.common import DIRECTORY_ERROR, FILE_ERROR, PLUGIN_ERROR, SYNTAX_ERROR, VALIDATE_DONE, VALIDATE_ERROR
def main(): def main():
@ -18,22 +16,31 @@ def main():
paths.extend(files) paths.extend(files)
except NotADirectoryError as e: except NotADirectoryError as e:
print(e) print(e)
sys.exit(DIRECTORY_ERROR) exit(DIRECTORY_ERROR)
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
print(e) print(e)
sys.exit(PLUGIN_ERROR) exit(PLUGIN_ERROR)
except FileNotFoundError as e: except FileNotFoundError as e:
print(e) print(e)
sys.exit(FILE_ERROR) exit(FILE_ERROR)
except SyntaxError as e:
print(e)
exit(SYNTAX_ERROR)
ast_items = collect.ast_pytest_items(paths) ast_items = collect.ast_pytest_items(paths)
files = collect.fixtures(ast_items.fixtures, ast_items.hooks, plugins) fixtures = collect.fixtures(ast_items.fixtures, ast_items.hooks, plugins)
tests = collect.tests(ast_items.tests, files, plugins) tests = collect.tests(ast_items.tests, fixtures, plugins)
validate_result = validate.titles(tests) validate_result = validate.titles(tests)
if validate_result.errors == 0: if validate_result.errors == 0:
print(f"Done🎉\nFiles analyzed: {len(paths)}\nTests analyzed: {len(tests)}") print(f"Done🎉\nFiles analyzed: {len(paths)}\nTests analyzed: {len(tests)}")
sys.exit(VALIDATE_DONE)
if plugins:
patterns = ", ".join(args.plugins)
print(f"Found plugins with patterns: {patterns}")
print(*(f"-> {plugin}" for plugin in plugins), sep="\n")
exit(VALIDATE_DONE)
for title, tests in validate_result.not_unique.items(): for title, tests in validate_result.not_unique.items():
print(f"NOT UNIQUE TITLE: {title}") print(f"NOT UNIQUE TITLE: {title}")
@ -49,4 +56,9 @@ def main():
params_str = " OR ".join(params) if isinstance(params, list) else params params_str = " OR ".join(params) if isinstance(params, list) else params
print(f"{test.link}: MISSING PARAMS: Parameters are missing from title: {params_str}") print(f"{test.link}: MISSING PARAMS: Parameters are missing from title: {params_str}")
sys.exit(VALIDATE_ERROR) if plugins:
patterns = ", ".join(args.plugins)
print(f"Found plugins with patterns: {patterns}")
print(*(f"-> {plugin}" for plugin in plugins), sep="\n")
exit(VALIDATE_ERROR)

View file

@ -2,7 +2,15 @@ import ast
import re import re
from pathlib import Path from pathlib import Path
from allure_validator.common import ALLURE_TITLE, ALLURE_VALIDATOR_IGNORE, PYTEST_FIXTURE, PYTEST_PARAMETRIZE, REPORTER_TITLE from allure_validator.common import (
ALLURE_TITLE,
ALLURE_VALIDATOR_IGNORE,
PYTEST_FIXTURE,
PYTEST_FORCED_PARAMETRIZATION,
PYTEST_PARAM,
PYTEST_PARAMETRIZE,
REPORTER_TITLE,
)
from allure_validator.pytest_items import TestParams from allure_validator.pytest_items import TestParams
@ -130,25 +138,67 @@ def fixture_params(func: ast.FunctionDef) -> list[str]:
""" """
ast_params = [] ast_params = []
for deco in func.decorator_list: for deco in func.decorator_list:
if not isinstance(deco, ast.Call): if not isinstance(deco, ast.Call):
continue continue
for keyword in deco.keywords: for keyword in deco.keywords:
if keyword.arg == "params": if keyword.arg == "params":
ast_params.extend(keyword.value.elts) if isinstance(keyword.value, ast.List | ast.Tuple):
ast_params.extend(keyword.value.elts)
else:
return PYTEST_FORCED_PARAMETRIZATION
params = [] params = []
for ast_param in ast_params: for ast_param in ast_params:
for arg in ast_param.args: if isinstance(ast_param, ast.Call) and decorator(ast_param) == PYTEST_PARAM:
params.append(ast.unparse(arg)) params.extend(ast.unparse(arg) for arg in ast_param.args)
continue
if not isinstance(ast_param, ast.Starred):
params.append(ast.unparse(ast_param))
continue
return PYTEST_FORCED_PARAMETRIZATION
return params return params
def param_names(ast_names: ast.Constant | ast.List) -> list[str]: def param_names(ast_names: ast.Constant | ast.List | ast.Tuple) -> list[str]:
"""Generates a list of parameter names from `pytest.parametrize`.""" """Generates a list of parameter names from `pytest.parametrize`."""
if isinstance(ast_names, ast.Constant): if isinstance(ast_names, ast.Constant):
return ast_names.value.replace(",", " ").split() return ast_names.value.replace(",", " ").split()
return [elt.value for elt in ast_names.elts] return [elt.value for elt in ast_names.elts]
def metafunc_params(ast_params: ast.List | ast.Tuple) -> list:
"""
Extracts parameters for the fixture from the `params` argument of the `metafunc.parametrize` function.
Args:
ast_params: Node from which parameters should be extracted.
Returns:
list: Extracted parameters.
"""
if not isinstance(ast_params, ast.List | ast.Tuple):
return PYTEST_FORCED_PARAMETRIZATION
if any(isinstance(elt, ast.Starred) for elt in ast_params.elts):
return PYTEST_FORCED_PARAMETRIZATION
params = []
for ast_param in ast_params.elts:
if isinstance(ast_param, ast.Constant):
params.append(ast_param.value)
elif isinstance(ast_param, ast.Name | ast.Attribute):
params.append(ast.unparse(ast_param))
else:
SyntaxError(ast.unparse(ast_param))
return params

View file

@ -39,7 +39,7 @@ class Fixture(Item):
@property @property
def parametrized(self) -> bool: def parametrized(self) -> bool:
return bool(self.params) return len(self.params) > 1
@dataclass(eq=False) @dataclass(eq=False)

View file

@ -2,6 +2,10 @@ import allure
import pytest import pytest
def parametrize_fixture() -> list[int]:
return [1, 2, 3]
@pytest.fixture( @pytest.fixture(
scope="function", scope="function",
params=[ params=[
@ -34,3 +38,12 @@ def object_fixture(file_path):
) )
def parametrized_fixture(request: pytest.FixtureRequest): def parametrized_fixture(request: pytest.FixtureRequest):
return request.param return request.param
@allure.title("[Session] Optional parametrized fixture")
@pytest.fixture(
scope="session",
params=[parametrize_fixture()],
)
def optional_parametrized_fixture(request: pytest.FixtureRequest):
return request.param

View file

@ -1,18 +1,91 @@
import allure import allure
import pytest import pytest
from frostfs_testlib.s3.aws_cli_client import AwsCliClient
from frostfs_testlib.s3.boto3_client import Boto3ClientWrapper
class Clients:
AttributeClient = ...
def parametrize_fixture() -> list[int]:
return [1, 2, 3]
# ============== DYNAMIC FIXTURE ============== # # ============== DYNAMIC FIXTURE ============== #
def pytest_generate_tests(metafunc: pytest.Metafunc): def pytest_generate_tests(metafunc: pytest.Metafunc):
if "runtime_fixture" not in metafunc.fixturenames: # Required dynamic fixture
return if "required_runtime_fixture" in metafunc.fixturenames:
metafunc.parametrize("required_runtime_fixture", [AwsCliClient, Boto3ClientWrapper, Clients.AttributeClient])
metafunc.parametrize("runtime_fixture", [1, 2, 3]) # Optional dynamic fixture 1
if "optional_runtime_fixture_1" in metafunc.fixturenames:
metafunc.parametrize("optional_runtime_fixture_1", [True])
# Optional dynamic fixture 2
if "optional_runtime_fixture_2" in metafunc.fixturenames:
metafunc.parametrize("optional_runtime_fixture_2", [parametrize_fixture()])
# Required dynamic fixtures
if "required_runtime_1" in metafunc.fixturenames and "required_runtime_2" in metafunc.fixturenames:
metafunc.parametrize("required_runtime_1,required_runtime_2", [(1, 2), (3, 4)])
# Optional dynamic fixtures
if "optional_runtime_1" in metafunc.fixturenames and "optional_runtime_2" in metafunc.fixturenames:
metafunc.parametrize(["optional_runtime_1", "optional_runtime_2"], [("optional", [1, 2, 3])])
# Required dynamic fixture with implicit parameterization 1
if "implicitly_required_runtime_fixture_1" in metafunc.fixturenames:
metafunc.parametrize("implicitly_required_runtime_fixture_1", parametrize_fixture())
# Required dynamic fixture with implicit parameterization 2
if "implicitly_required_runtime_fixture_2" in metafunc.fixturenames:
metafunc.parametrize("implicitly_required_runtime_fixture_2", [*parametrize_fixture()])
# Required dynamic fixture with implicit parameterization 3
if "implicitly_required_runtime_fixture_3" in metafunc.fixturenames:
metafunc.parametrize("implicitly_required_runtime_fixture_3", [0, parametrize_fixture(), *parametrize_fixture()])
@allure.title("Test runtime fixture") @allure.title("Test required runtime fixture")
def test_runtime_fixture(runtime_fixture): def test_required_runtime_fixture(required_runtime_fixture):
assert True
@allure.title("Test optional runtime fixture 1")
def test_optional_runtime_fixture_1(optional_runtime_fixture_1):
assert True
@allure.title("Test optional runtime fixture 2")
def test_optional_runtime_fixture_2(optional_runtime_fixture_2):
assert True
@allure.title("Test required runtime fixtures")
def test_required_runtime_fixtures(required_runtime_1, required_runtime_2):
assert True
@allure.title("Test optional runtime fixtures")
def test_optional_runtime_fixtures(optional_runtime_1, optional_runtime_2):
assert True
@allure.title("Test implicitly required runtime fixture 1")
def test_implicitly_required_runtime_fixture_1(implicitly_required_runtime_fixture_1):
assert True
@allure.title("Test implicitly required runtime fixture 2")
def test_implicitly_required_runtime_fixture_2(implicitly_required_runtime_fixture_2):
assert True
@allure.title("Test implicitly required runtime fixture 3")
def test_implicitly_required_runtime_fixture_3(implicitly_required_runtime_fixture_3):
assert True assert True
@ -117,6 +190,11 @@ def test_three_multiple_indirect(object_fixture: str):
assert True assert True
@allure.title("Test optional fixture")
def test_optional_fixture(optional_parametrized_fixture):
assert True
# ============== TESTS/FIXTURES IN CLASS ============== # # ============== TESTS/FIXTURES IN CLASS ============== #