mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
Adam Perry @ OpenAI
99da697e4c
## Why `codex-extension-api` defines contracts shared by extension crates and their hosts, but it had no direct test suite. Host and feature tests cover downstream behavior, while regressions in the API crate's own typed state, registry ordering, and capability adapters could go unnoticed. ## What - Add public-surface integration tests for `ExtensionData`, including concurrent initialization and poison recovery. - Cover contributor registration order, approval short-circuiting, event sink retention, no-op response injection, and closure-based agent spawning. - Add the test-only dependencies used by the suite. ## Validation - `just test -p codex-extension-api` - `just argument-comment-lint -p codex-extension-api` - `just bazel-lock-check`
57 lines
1.8 KiB
Rust
57 lines
1.8 KiB
Rust
use std::sync::Arc;
|
|
use std::sync::Mutex;
|
|
|
|
use codex_extension_api::AgentSpawnFuture;
|
|
use codex_extension_api::AgentSpawner;
|
|
use codex_extension_api::NoopResponseItemInjector;
|
|
use codex_extension_api::ResponseItemInjector;
|
|
use codex_protocol::ThreadId;
|
|
use codex_protocol::models::ContentItem;
|
|
use codex_protocol::models::ResponseInputItem;
|
|
use pretty_assertions::assert_eq;
|
|
|
|
#[tokio::test]
|
|
async fn noop_response_item_injector_returns_original_items() {
|
|
let items = vec![ResponseInputItem::Message {
|
|
role: "user".to_string(),
|
|
content: vec![ContentItem::InputText {
|
|
text: "keep this input".to_string(),
|
|
}],
|
|
phase: None,
|
|
}];
|
|
|
|
let returned_items = NoopResponseItemInjector
|
|
.inject_response_items(items.clone())
|
|
.await
|
|
.expect_err("noop injector should reject same-turn injection");
|
|
|
|
assert_eq!(returned_items, items);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn closure_agent_spawner_forwards_arguments_and_result() {
|
|
let calls = Arc::new(Mutex::new(Vec::new()));
|
|
let recorded_calls = Arc::clone(&calls);
|
|
let spawner = move |thread_id: ThreadId,
|
|
request: String|
|
|
-> AgentSpawnFuture<'static, usize, &'static str> {
|
|
recorded_calls
|
|
.lock()
|
|
.expect("agent spawn calls lock")
|
|
.push((thread_id, request.clone()));
|
|
Box::pin(async move { Ok(request.len()) })
|
|
};
|
|
let thread_id =
|
|
ThreadId::from_string("11111111-1111-4111-8111-111111111111").expect("valid thread id");
|
|
|
|
let spawned = spawner
|
|
.spawn_subagent(thread_id, "delegate this".to_string())
|
|
.await;
|
|
|
|
assert_eq!(spawned, Ok(13));
|
|
assert_eq!(
|
|
calls.lock().expect("agent spawn calls lock").as_slice(),
|
|
[(thread_id, "delegate this".to_string())]
|
|
);
|
|
}
|