50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
# Copyright (c) 2023-2024 Arista Networks, Inc.
|
|
# Use of this source code is governed by the Apache License 2.0
|
|
# that can be found in the LICENSE file.
|
|
"""conftest.py - used to store anta specific fixtures used for tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
# Load fixtures from dedicated file tests/lib/fixture.py
|
|
# As well as pytest_asyncio plugin to test co-routines
|
|
pytest_plugins = [
|
|
"tests.lib.fixture",
|
|
"pytest_asyncio",
|
|
]
|
|
|
|
# Enable nice assert messages
|
|
# https://docs.pytest.org/en/7.1.x/how-to/writing_plugins.html#assertion-rewriting
|
|
pytest.register_assert_rewrite("tests.lib.anta")
|
|
|
|
# Placeholder to disable logging of some external libs
|
|
for _ in ("asyncio", "httpx"):
|
|
logging.getLogger(_).setLevel(logging.CRITICAL)
|
|
|
|
|
|
def build_test_id(val: dict[str, Any]) -> str:
|
|
"""Build id for a unit test of an AntaTest subclass.
|
|
|
|
{
|
|
"name": "meaniful test name",
|
|
"test": <AntaTest instance>,
|
|
...
|
|
}
|
|
"""
|
|
return f"{val['test'].__module__}.{val['test'].__name__}-{val['name']}"
|
|
|
|
|
|
def pytest_generate_tests(metafunc: pytest.Metafunc) -> None:
|
|
"""Generate ANTA testts unit tests dynamically during test collection.
|
|
|
|
It will parametrize test cases based on the `DATA` data structure defined in `tests.units.anta_tests` modules.
|
|
See `tests/units/anta_tests/README.md` for more information on how to use it.
|
|
Test IDs are generated using the `build_test_id` function above.
|
|
"""
|
|
if "tests.units.anta_tests" in metafunc.module.__package__:
|
|
# This is a unit test for an AntaTest subclass
|
|
metafunc.parametrize("data", metafunc.module.DATA, ids=build_test_id)
|