chore: import upstream snapshot with attribution

tests/test_codex_native_state.py

@@ -0,0 +1,143 @@
+"""Tests for the codex-native client-side launch-state store."""
+
+from __future__ import annotations
+
+import hashlib
+import logging
+from pathlib import Path
+
+import pytest
+
+from omnigent.codex_native_state import (
+    read_launch_state,
+    write_launch_state,
+)
+
+
+def test_write_and_read_launch_state_round_trips(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    """
+    Launch state persists the cwd needed by resume-time alignment.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :returns: None.
+    """
+    state_root = tmp_path / "state"
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(state_root))
+
+    write_launch_state("conv_abc", "/repo")
+
+    state = read_launch_state("conv_abc")
+    assert state is not None
+    assert state.working_directory == "/repo"
+
+
+def test_launch_state_path_hashes_conversation_id(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    """
+    Conversation ids never land in the filesystem path directly.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :returns: None.
+    """
+    state_root = tmp_path / "state"
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(state_root))
+    conversation_id = "../../../etc/passwd"
+    digest = hashlib.sha256(conversation_id.encode("utf-8")).hexdigest()[:32]
+
+    write_launch_state(conversation_id, "/repo")
+
+    assert (state_root / digest / "launch.json").is_file()
+    assert not (tmp_path / "etc").exists()
+
+
+def test_conflicting_launch_state_write_keeps_original(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+    caplog: pytest.LogCaptureFixture,
+) -> None:
+    """
+    A second write for the same conversation cannot silently move cwd.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :param caplog: Pytest log capture fixture.
+    :returns: None.
+    """
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(tmp_path / "state"))
+    # Defensive: sibling CLI/logging tests can leave the package
+    # logger with propagation disabled in this xdist worker. The
+    # warning is emitted by ``omnigent.codex_native_state`` and
+    # caplog's handler is attached at root.
+    logging.getLogger("omnigent").propagate = True
+    write_launch_state("conv_abc", "/original")
+
+    with caplog.at_level(logging.WARNING):
+        write_launch_state("conv_abc", "/other")
+
+    state = read_launch_state("conv_abc")
+    assert state is not None
+    assert state.working_directory == "/original"
+    assert any("codex-native launch state mismatch" in record.message for record in caplog.records)
+
+
+def test_missing_or_malformed_launch_state_returns_none(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    """
+    Bad state files behave like legacy unrecorded sessions.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :returns: None.
+    """
+    state_root = tmp_path / "state"
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(state_root))
+    digest = hashlib.sha256(b"conv_bad").hexdigest()[:32]
+    state_dir = state_root / digest
+    state_dir.mkdir(parents=True)
+    (state_dir / "launch.json").write_text("{not-json\n", encoding="utf-8")
+
+    assert read_launch_state("conv_missing") is None
+    assert read_launch_state("conv_bad") is None
+
+
+def test_empty_working_directory_rejected(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    """
+    Empty cwd would make every later resume comparison meaningless.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :returns: None.
+    """
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(tmp_path / "state"))
+
+    with pytest.raises(ValueError, match="working_directory"):
+        write_launch_state("conv_abc", "")
+
+
+def test_relative_working_directory_rejected(
+    monkeypatch: pytest.MonkeyPatch,
+    tmp_path: Path,
+) -> None:
+    """
+    Launch state records canonical absolute cwd values only.
+
+    :param monkeypatch: Pytest monkeypatch fixture.
+    :param tmp_path: Temporary directory for isolated state.
+    :returns: None.
+    """
+    monkeypatch.setenv("OMNIGENT_CODEX_NATIVE_STATE_DIR", str(tmp_path / "state"))
+
+    with pytest.raises(ValueError, match="absolute path"):
+        write_launch_state("conv_abc", "relative/repo")