scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

test(integration): tidy xdist collection bookkeeping

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-19 00:54:45 +00:00
parent 6eb67a8423
commit a15b0fa6d2
1 changed files with 10 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/integration/conftest.py
View file

@ -1,21 +1,20 @@
from __future__ import annotations
import json
import hashlib
import json
import os
from collections.abc import Sequence
from collections.abc import Iterator
from collections.abc import Iterator, Sequence
from importlib.metadata import version
from pathlib import Path
from typing import Final
import pytest
import httpx
import pytest
from redis import Redis
from tests.integration._support.client import Gateway, eventually, gateway_from_environment
from tests.integration._support.manifest import OWNED_DIRECTORIES, contracts
from tests.integration._support.generation import LIFECYCLE_SETTINGS
from tests.integration._support.manifest import OWNED_DIRECTORIES, contracts
COLLECTED: Final = pytest.StashKey[tuple[str, ...]]()
REPORTS: Final = pytest.StashKey[list[pytest.TestReport]]()
@ -41,14 +40,12 @@ class IntegrationReportPlugin:
@pytest.hookimpl(optionalhook=True)
def pytest_xdist_node_collection_finished(self, node: object, ids: Sequence[str]) -> None:
owned_prefix: Final = "tests/integration/"
self.config.stash[COLLECTED] = tuple(
nodeid
for nodeid in ids
if nodeid.split("::", 1)[0].startswith(owned_prefix)
and len(Path(nodeid.split("::", 1)[0]).parts) > 2
and Path(nodeid.split("::", 1)[0]).parts[2] in OWNED_DIRECTORIES
)
self.config.stash[COLLECTED] = tuple(nodeid for nodeid in ids if _owned(nodeid))
def _owned(nodeid: str) -> bool:
parts: Final = Path(nodeid.split("::", 1)[0]).parts
return parts[:2] == ("tests", "integration") and len(parts) > 3 and parts[2] in OWNED_DIRECTORIES
def pytest_collection_modifyitems(config: pytest.Config, items: list[pytest.Item]) -> None: