wehub-resource-sync
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
"""Tests for omnigent.harness_aliases."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from omnigent.harness_aliases import (
|
|
canonicalize_harness,
|
|
is_native_harness,
|
|
native_terminal_name,
|
|
)
|
|
from omnigent.spec._omnigent_compat import OMNIGENT_HARNESSES
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"alias,canonical",
|
|
[
|
|
("claude", "claude-sdk"),
|
|
("native-pi", "pi-native"),
|
|
("native-kiro", "kiro-native"),
|
|
# Docs / runtime-dispatch spelling of the openai-agents harness;
|
|
# specs and OMNIGENT_HARNESSES use "openai-agents".
|
|
("openai-agents-sdk", "openai-agents"),
|
|
# Canonical names pass through unchanged.
|
|
("openai-agents", "openai-agents"),
|
|
("pi", "pi"),
|
|
# Canonical cursor id passes through unchanged (no alias).
|
|
("cursor", "cursor"),
|
|
# Antigravity SDK harness: user-facing spellings → canonical id.
|
|
("agy", "antigravity"),
|
|
("google-antigravity", "antigravity"),
|
|
("antigravity", "antigravity"),
|
|
# Unknown names return unchanged so callers keep their own errors.
|
|
("bogus", "bogus"),
|
|
(None, None),
|
|
],
|
|
)
|
|
def test_canonicalize_harness(alias: str | None, canonical: str | None) -> None:
|
|
"""Alias spellings map to canonical ids; everything else passes through.
|
|
|
|
A missing ``openai-agents-sdk`` mapping breaks the documented
|
|
``omnigent run ... --harness openai-agents-sdk`` invocation at
|
|
``_validate_harness``.
|
|
"""
|
|
assert canonicalize_harness(alias) == canonical
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"harness,expected",
|
|
[
|
|
# Canonical native spellings and their reversed forms.
|
|
("claude-native", True),
|
|
("codex-native", True),
|
|
("native-claude", True),
|
|
("native-codex", True),
|
|
("pi-native", True),
|
|
("native-pi", True),
|
|
("kiro-native", True),
|
|
("native-kiro", True),
|
|
# SDK harnesses are NOT native — they replay the Omnigent
|
|
# transcript and don't own an on-disk runtime transcript. A
|
|
# regression that classified these as native would wrongly route a
|
|
# fork into the native-rebuild path.
|
|
("claude-sdk", False),
|
|
("claude_sdk", False),
|
|
("openai-agents", False),
|
|
("agents_sdk", False),
|
|
("codex", False),
|
|
("kiro", False),
|
|
# The "claude" shorthand canonicalizes to claude-sdk (not native).
|
|
("claude", False),
|
|
# cursor is a headless ACP harness, not a native CLI bridge.
|
|
("cursor", False),
|
|
("some-unknown-harness", False),
|
|
(None, False),
|
|
],
|
|
)
|
|
def test_is_native_harness(harness: str | None, expected: bool) -> None:
|
|
"""``is_native_harness`` flags only the native CLI harnesses.
|
|
|
|
The runner gates terminal-owned turn sequencing and history replay on
|
|
this. Misclassifying either way makes native TUI sessions behave like
|
|
in-process SDK turns, or vice versa.
|
|
"""
|
|
assert is_native_harness(harness) is expected
|
|
|
|
|
|
def test_kiro_native_is_valid_omnigent_harness_but_plain_kiro_is_not() -> None:
|
|
"""Kiro's native identity is canonical; plain ``kiro`` is not a generic harness."""
|
|
assert "kiro-native" in OMNIGENT_HARNESSES
|
|
assert "kiro" not in OMNIGENT_HARNESSES
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("harness", "expected"),
|
|
[
|
|
("claude-native", "claude"),
|
|
("native-claude", "claude"), # reversed alias
|
|
("codex-native", "codex"),
|
|
("cursor-native", "cursor"),
|
|
("goose-native", "goose"),
|
|
("hermes-native", "hermes"),
|
|
("kiro-native", "kiro"),
|
|
("qwen-native", "qwen"),
|
|
("kimi-native", "kimi"),
|
|
("pi-native", "pi"),
|
|
("antigravity-native", "antigravity"),
|
|
("opencode-native", "opencode"),
|
|
("opencode", "opencode"), # alias folds to opencode-native
|
|
# Non-native harnesses (and the SDK shorthands) have no native pane.
|
|
("claude-sdk", None),
|
|
("claude", None),
|
|
("codex", None),
|
|
("cursor", None),
|
|
("agents_sdk", None),
|
|
("some-unknown-harness", None),
|
|
(None, None),
|
|
],
|
|
)
|
|
def test_native_terminal_name(harness: str | None, expected: str | None) -> None:
|
|
"""``native_terminal_name`` maps native harness ids to their tmux pane name.
|
|
|
|
The native-pane idle reaper (#1349) and its turn-path self-heal key panes on
|
|
``(conv, <short-name>, "main")``; a wrong mapping would reap/relaunch the
|
|
wrong pane or silently skip a harness.
|
|
"""
|
|
assert native_terminal_name(harness) == expected
|