phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: only sleep for tests when they pass or fail

Browse source
This commit is contained in:
Ashwin Bharambe 2025-04-25 13:15:52 -07:00
parent 1b2e116a2a
commit b5d8e44e81
1 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/integration/conftest.py
View file

@ -10,6 +10,7 @@ import platform
import textwrap import textwrap
import time import time
import pytest
from dotenv import load_dotenv from dotenv import load_dotenv
from llama_stack.log import get_logger from llama_stack.log import get_logger
@ -19,10 +20,29 @@ from .report import Report
logger = get_logger(__name__, category="tests") logger = get_logger(__name__, category="tests")
@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item, call):
outcome = yield
report = outcome.get_result()
if report.when == "call":
item.execution_outcome = report.outcome
item.was_xfail = getattr(report, "wasxfail", False)
def pytest_runtest_teardown(item): def pytest_runtest_teardown(item):
interval_seconds = os.getenv("LLAMA_STACK_TEST_INTERVAL_SECONDS") # Check if the test actually ran and passed or failed, but was not skipped or an expected failure (xfail)
if interval_seconds: outcome = getattr(item, "execution_outcome", None)
time.sleep(float(interval_seconds)) was_xfail = getattr(item, "was_xfail", False)
name = item.nodeid
if not any(x in name for x in ("inference/", "safety/", "agents/")):
return
logger.debug(f"Test '{item.nodeid}' outcome was '{outcome}' (xfail={was_xfail})")
if outcome in ("passed", "failed") and not was_xfail:
interval_seconds = os.getenv("LLAMA_STACK_TEST_INTERVAL_SECONDS")
if interval_seconds:
time.sleep(float(interval_seconds))
def pytest_configure(config): def pytest_configure(config):