|
|
|
|
|
from __future__ import annotations |
|
|
"""Streamlit Test Dashboard for AI E-Consult V2. |
|
|
|
|
|
Runs the internal smoke and integration tests (tests/smoke_v2.py, tests/integration_v2.py) |
|
|
and displays their console output live within the UI. |
|
|
""" |
|
|
|
|
|
import io |
|
|
import sys |
|
|
import contextlib |
|
|
import runpy |
|
|
from pathlib import Path |
|
|
|
|
|
import streamlit as st |
|
|
|
|
|
st.title("🧪 Test Dashboard — AI E-Consult V2") |
|
|
st.caption("Run smoke and integration tests directly within the Hugging Face Space.") |
|
|
|
|
|
tests_dir = Path(__file__).resolve().parents[1] / "tests" |
|
|
smoke_path = tests_dir / "smoke_v2.py" |
|
|
integ_path = tests_dir / "integration_v2.py" |
|
|
|
|
|
if not smoke_path.exists() or not integ_path.exists(): |
|
|
st.error("❌ Test files not found. Please ensure tests/smoke_v2.py and tests/integration_v2.py exist.") |
|
|
st.stop() |
|
|
|
|
|
st.markdown("**Available tests:**") |
|
|
st.markdown("- `smoke_v2.py` — basic schema and module checks") |
|
|
st.markdown("- `integration_v2.py` — module wiring and export determinism") |
|
|
|
|
|
run_btn = st.button("▶️ Run All Tests", use_container_width=True) |
|
|
|
|
|
output_buf = io.StringIO() |
|
|
|
|
|
def _run_script(path: Path): |
|
|
print(f"\n=== Running {path.name} ===") |
|
|
try: |
|
|
runpy.run_path(str(path), run_name="__main__") |
|
|
except SystemExit as e: |
|
|
print(f"Exited with code {getattr(e, 'code', '?')}") |
|
|
except Exception as e: |
|
|
print(f"⚠️ Error running {path.name}: {e}") |
|
|
|
|
|
if run_btn: |
|
|
with contextlib.redirect_stdout(output_buf), contextlib.redirect_stderr(output_buf): |
|
|
print("🔍 Starting internal test suite...\n") |
|
|
_run_script(smoke_path) |
|
|
_run_script(integ_path) |
|
|
print("\n✅ Test run completed.") |
|
|
|
|
|
st.success("Test run completed.") |
|
|
st.markdown("**Console Output:**") |
|
|
st.code(output_buf.getvalue(), language="bash") |
|
|
|
|
|
|
|
|
st.download_button( |
|
|
"💾 Download Results", |
|
|
data=output_buf.getvalue(), |
|
|
file_name="test_results_v2.txt", |
|
|
mime="text/plain", |
|
|
use_container_width=True, |
|
|
) |
|
|
else: |
|
|
st.info("Click 'Run All Tests' to execute internal test scripts and view results here.") |
|
|
|
