import asyncio
import os
import shutil
import tempfile
from pathlib import Path
import pytest
class _DummyPipe:
def __init__(self) -> None: # pragma: no cover - trivial
self._buf = b""
def write(self, data: bytes) -> None: # pragma: no cover - trivial
self._buf += data
async def drain(self) -> None: # pragma: no cover - trivial
return None
def close(self) -> None: # pragma: no cover - trivial
return None
class _DummyProc:
def __init__(self, rc: int = 0, out: bytes = b"OK", err: bytes = b"") -> None:
self.returncode = rc
self._out = out
self._err = err
self.stdin = _DummyPipe()
async def communicate(self): # pragma: no cover - exercised indirectly
return self._out, self._err
def kill(self) -> None: # pragma: no cover - trivial
return None
async def wait(self) -> None: # pragma: no cover - trivial
return None
@pytest.mark.asyncio
async def test_codex_overlay_cleanup(monkeypatch, tmp_path: Path):
"""Ensure Codex provider cleans up overlay CODEX_HOME after exec.
We stub out the subprocess call and force a known overlay path using
`_build_overlay_home()`. After `_run_exec()` returns, the overlay should
be removed regardless of success.
"""
from chunkhound.providers.llm.codex_cli_provider import CodexCLIProvider
# Force provider to consider Codex available
monkeypatch.setattr(CodexCLIProvider, "_codex_available", lambda self: True, raising=True)
# Create a deterministic overlay directory
overlay_dir = tmp_path / "overlay-home"
overlay_dir.mkdir(parents=True, exist_ok=True)
# Monkeypatch build to return our overlay path
requested_model = {}
def _fake_overlay_home(self, model_override=None):
requested_model["value"] = model_override
return str(overlay_dir)
monkeypatch.setattr(CodexCLIProvider, "_build_overlay_home", _fake_overlay_home, raising=True)
# Stub out subprocess creation to avoid calling real codex
async def _fake_create_subprocess_exec(*args, **kwargs): # noqa: ANN001
return _DummyProc(rc=0, out=b"OK", err=b"")
monkeypatch.setattr(asyncio, "create_subprocess_exec", _fake_create_subprocess_exec, raising=True)
prov = CodexCLIProvider(model="codex")
# Sanity: overlay exists before run
assert overlay_dir.exists()
# Run via argv path (short content) and ensure success
out = await prov._run_exec("ping", cwd=None, max_tokens=16, timeout=10, model="codex") # type: ignore[attr-defined]
assert out.strip() == "OK"
# Overlay should be cleaned up by provider
assert not overlay_dir.exists(), "overlay CODEX_HOME was not cleaned up"
# "codex" alias should resolve to the default Codex model
assert requested_model.get("value") == "gpt-5.1-codex"
