File size: 2,418 Bytes
a2afe2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
"""Custom pytest configuration for the extensions."""
from typing import Dict, List
import pytest
from openbb_core.app.router import CommandMap
from extensions.tests.utils.helpers import list_openbb_extensions
cm = CommandMap()
commands = list(cm.map.keys())
# ruff: noqa: SIM114
def parametrize(argnames: str, argvalues: List, **kwargs):
"""Apply a custom parametrize decorator that filters test cases based on the environment."""
routers, providers, obbject_ext = list_openbb_extensions()
def decorator(function):
"""Patch the pytest.mark.parametrize decorator."""
filtered_argvalues: List[Dict] = []
name = function.__name__.split("_")[1]
# This is a patch to handle the charting extension name
extension_name = "openbb_" + name if name == "charting" else name
function_name = "/" + "/".join(function.__name__.split("_")[1:])
# this handles edge cases where the function name has an underscore
function_name_v2 = (
function_name.rsplit("/", 1)[0] + "_" + function_name.rsplit("/", 1)[1]
)
function_name_v3 = (
function_name.rsplit("/", 2)[0]
+ "_"
+ function_name.rsplit("/", 2)[1]
+ "_"
+ function_name.rsplit("/", 1)[1].replace("/", "_")
)
if extension_name in routers | obbject_ext:
for args in argvalues:
if "provider" in args and args["provider"] in providers:
filtered_argvalues.append(args)
elif "provider" not in args and function_name in commands:
# Run the standard test case
filtered_argvalues.append(args)
elif "provider" not in args and function_name_v2 in commands:
# Handle edge case
filtered_argvalues.append(args)
elif "provider" not in args and function_name_v3 in commands:
# Handle edge case
filtered_argvalues.append(args)
elif extension_name in obbject_ext:
filtered_argvalues.append(args)
# If filtered_argvalues is empty, pytest will skip the test!
return pytest.mark.parametrize(argnames, filtered_argvalues, **kwargs)(
function
)
return pytest.mark.skip(function)
return decorator
|