sdk/python/tests/installed_sdk_smoke.py

"""Exercise a built SDK's default runtime in an otherwise isolated environment."""

from dataclasses import replace
from importlib.metadata import distribution, version
from pathlib import Path
from tempfile import TemporaryDirectory

from app_server_harness import AppServerHarness

import openai_codex
from openai_codex import Codex


def main() -> None:
    installed_root = Path(distribution("openai-codex").locate_file("")).resolve()
    assert Path(openai_codex.__file__).resolve().is_relative_to(installed_root), (
        "The smoke test must import the installed SDK, not the source checkout"
    )

    with TemporaryDirectory() as directory, AppServerHarness(Path(directory)) as harness:
        harness.responses.enqueue_assistant_message("Installed SDK works")
        config = replace(harness.app_server_config(), codex_bin=None)
        with Codex(config=config) as codex:
            thread = codex.thread_start()
            result = thread.run(
                "Check the installed SDK", turn_service_tier="default", source="automation"
            )
            codex.thread_resume(thread.id, include_turns=False)
            codex.thread_fork(thread.id, include_turns=True)
        assert result.final_response == "Installed SDK works"
        assert harness.responses.single_request().message_input_texts("user")[-1:] == [
            "Check the installed SDK"
        ]

    print(f"Installed SDK passed with CLI runtime {version('openai-codex-cli-bin')}")


if __name__ == "__main__":
    main()