From 7411a11f83a46201ce52b6e384adbecc8973d33a Mon Sep 17 00:00:00 2001
From: firestar5683 <168790843+firestar5683@users.noreply.github.com>
Date: Sat, 27 Jun 2026 23:58:18 -0500
Subject: [PATCH] pytest

---
 conftest.py                 | 32 +++++++++++++++++
 scripts/host_tool_runner.sh | 68 ++++++++++++++++++++++++++++++++++++-
 2 files changed, 99 insertions(+), 1 deletion(-)

diff --git a/conftest.py b/conftest.py
index 7e40ec3ed..aeb1423bd 100644
--- a/conftest.py
+++ b/conftest.py
@@ -1,8 +1,40 @@
 import contextlib
 import gc
 import os
+import platform
+import sys
+from pathlib import Path
+
 import pytest
 
+
+def _prepend_host_pytest_runtime() -> None:
+  if platform.system() != "Darwin" or os.getenv("SP_DISABLE_HOST_PYTEST_REDIRECT") == "1":
+    return
+
+  root_dir = Path(__file__).resolve().parent
+  work_dir = root_dir / ".host_runtime" / "darwin" / "worktree"
+  required_extension = work_dir / "msgq_repo" / "msgq" / "ipc_pyx.so"
+  if not required_extension.exists():
+    return
+
+  extra_paths = [work_dir, work_dir / "starpilot" / "third_party"]
+  extra_paths.extend(sorted(work_dir.glob("*_repo")))
+  acados_dir = work_dir / "third_party" / "acados"
+  if acados_dir.is_dir():
+    extra_paths.append(acados_dir)
+
+  existing = set(sys.path)
+  insert_at = 0
+  for path in [str(p) for p in extra_paths if p.exists()]:
+    if path in existing:
+      continue
+    sys.path.insert(insert_at, path)
+    insert_at += 1
+
+
+_prepend_host_pytest_runtime()
+
 from openpilot.common.prefix import OpenpilotPrefix
 from openpilot.system.manager import manager
 from openpilot.system.hardware import TICI, HARDWARE
diff --git a/scripts/host_tool_runner.sh b/scripts/host_tool_runner.sh
index f61905d89..bf9a7243b 100755
--- a/scripts/host_tool_runner.sh
+++ b/scripts/host_tool_runner.sh
@@ -32,6 +32,8 @@ Commands:
   cabana       Build and run cabana from the isolated host cache.
   plotjuggler  Run PlotJuggler helper from the isolated host cache.
   juggle       Alias for plotjuggler.
+  python       Run Python from the isolated host cache with repo paths configured.
+  pytest       Run pytest from the isolated host cache after building host extensions.
   sync         Refresh the isolated host cache only.
   shell        Open a shell inside the isolated host cache.
   help         Show this help text.
@@ -214,6 +216,25 @@ purge_host_python_artifacts() {
     "${WORK_DIR}/msgq_repo/msgq/visionipc/visionipc_pyx.cpp"
 }
 
+purge_host_foreign_platform_artifacts() {
+  rm -rf "${WORK_DIR}/.host_runtime"
+  rm -rf "${WORK_DIR}"/.tmpcapnp*
+  rm -rf "${WORK_DIR}/starpilot/system/the_pond"
+
+  case "$(uname -s)" in
+    Darwin)
+      find "${WORK_DIR}" \
+        \( -path "${WORK_DIR}/.venv" -o -path "${WORK_DIR}/.git" -o -path "${WORK_DIR}/.host_runtime" \) -prune -o \
+        -type f \( -name '*.cpython-*-linux-gnu.so' -o -name '*.cpython-*-linux-musl.so' \) -delete
+      ;;
+    Linux)
+      find "${WORK_DIR}" \
+        \( -path "${WORK_DIR}/.venv" -o -path "${WORK_DIR}/.git" -o -path "${WORK_DIR}/.host_runtime" \) -prune -o \
+        -type f -name '*.cpython-*-darwin.so' -delete
+      ;;
+  esac
+}
+
 purge_host_desktop_ui_artifacts() {
   rm -f \
     "${WORK_DIR}/selfdrive/ui/libqt_widgets.a" \
@@ -251,6 +272,17 @@ ensure_host_python_tools() {
   )
 }
 
+ensure_host_python_extensions() {
+  local jobs
+  jobs="$(default_jobs)"
+
+  run_host_scons "${jobs}" \
+    common/params_pyx.so \
+    common/transformations/transformations.so \
+    msgq_repo/msgq/ipc_pyx.so \
+    msgq_repo/msgq/visionipc/visionipc_pyx.so
+}
+
 sync_host_generated_headers() {
   if ! command -v capnpc >/dev/null 2>&1; then
     return
@@ -272,6 +304,7 @@ sync_worktree() {
   ensure_venv
 
   mkdir -p "${HOST_ROOT}"
+  rm -rf "${WORK_DIR}/starpilot/system/the_pond"
 
   local excludes=(
     ".git/"
@@ -280,9 +313,13 @@ sync_worktree() {
     ".cache/"
     ".comma_sysroot/"
     ".host_runtime/"
+    ".tmpcapnp*/"
     "__pycache__/"
     "*.pyc"
     "*.pyo"
+    "*.cpython-*-linux-gnu.so"
+    "*.cpython-*-linux-musl.so"
+    "*.cpython-*-darwin.so"
     "*.o"
     "*.os"
     ".sconsign.dblite"
@@ -338,6 +375,7 @@ sync_worktree() {
     rm -rf "${SP_SCONS_CACHE_DIR:-${HOST_ROOT}/scons_cache}"
   fi
   purge_host_desktop_ui_artifacts
+  purge_host_foreign_platform_artifacts
   rm -f "${WORK_DIR}/third_party/libjson11.a" "${WORK_DIR}/third_party/libkaitai.a"
   sync_host_generated_headers
   ensure_host_python_tools
@@ -405,6 +443,16 @@ run_python_tool() {
   )
 }
 
+run_host_python() {
+  (
+    cd "${WORK_DIR}"
+    setup_build_env
+    export_workdir_pythonpath
+    source .venv/bin/activate
+    exec "${WORK_DIR}/.venv/bin/python3" "$@"
+  )
+}
+
 launch_c3() {
   local jobs
   jobs="$(default_jobs)"
@@ -508,6 +556,18 @@ launch_plotjuggler() {
   run_python_tool "tools/plotjuggler/juggle.py" "$@"
 }
 
+launch_python() {
+  sync_worktree
+  ensure_host_python_extensions
+  run_host_python "$@"
+}
+
+launch_pytest() {
+  sync_worktree
+  ensure_host_python_extensions
+  run_host_python -m pytest "$@"
+}
+
 open_shell() {
   sync_worktree
   (
@@ -530,7 +590,7 @@ main() {
     help|-h|--help)
       usage
       ;;
-    c3|c4|raybig|onroad|replay|shell)
+    c3|c4|raybig|onroad|replay|shell|python|pytest)
       set_host_bucket "shared"
       acquire_host_lock "${command} $*"
       ;;
@@ -589,6 +649,12 @@ main() {
     shell)
       open_shell
       ;;
+    python)
+      launch_python "$@"
+      ;;
+    pytest)
+      launch_pytest "$@"
+      ;;
   esac
 }