mirror of
https://github.com/openai/codex.git
synced 2026-09-13 11:47:17 +00:00
10cc57c95c
## What changed - Add typed helpers for initialized app-server instances, JSON-RPC requests, responses, notifications, and thread startup. - Add a composable `MockResponsesConfig` builder while preserving the existing mock config writers. - Migrate the app-server integration suite to the shared helpers, removing repeated request ID management, deserialization, initialization, and config assembly. ## Testing - Add unit coverage for composing mock provider, feature, and extra config, and for compatibility of the existing mock config writer. GitOrigin-RevId: 62da177c1d6e8299400de64bd6766567bf1ba3d8
103 lines
3.2 KiB
Rust
103 lines
3.2 KiB
Rust
use anyhow::Result;
|
|
use app_test_support::MockResponsesConfig;
|
|
use app_test_support::TestAppServer;
|
|
use app_test_support::create_final_assistant_message_sse_response;
|
|
use codex_app_server_protocol::ClientRequest;
|
|
use codex_app_server_protocol::CurrentTimeReadResponse;
|
|
use codex_app_server_protocol::ServerRequest;
|
|
use codex_app_server_protocol::ThreadStartParams;
|
|
use codex_app_server_protocol::ThreadStartResponse;
|
|
use codex_app_server_protocol::TurnStartParams;
|
|
use codex_app_server_protocol::TurnStartResponse;
|
|
use codex_app_server_protocol::UserInput;
|
|
use core_test_support::responses;
|
|
use core_test_support::skip_if_no_network;
|
|
use pretty_assertions::assert_eq;
|
|
use tempfile::TempDir;
|
|
use tokio::time::Duration;
|
|
use tokio::time::timeout;
|
|
|
|
#[cfg(windows)]
|
|
const DEFAULT_READ_TIMEOUT: Duration = Duration::from_secs(25);
|
|
#[cfg(not(windows))]
|
|
const DEFAULT_READ_TIMEOUT: Duration = Duration::from_secs(10);
|
|
const CURRENT_TIME_AT: i64 = 1_781_717_655;
|
|
const CURRENT_TIME_REMINDER: &str = "It is 2026-06-17 17:34:15 UTC.";
|
|
|
|
#[tokio::test]
|
|
async fn current_time_read_round_trip_adds_reminder_to_model_input() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = responses::start_mock_server().await;
|
|
let response_mock = responses::mount_sse_once(
|
|
&server,
|
|
create_final_assistant_message_sse_response("Done")?,
|
|
)
|
|
.await;
|
|
let codex_home = TempDir::new()?;
|
|
MockResponsesConfig::new(&server.uri())
|
|
.with_extra_config(
|
|
r#"[features.current_time_reminder]
|
|
enabled = true
|
|
reminder_interval_seconds = 1
|
|
clock_source = "external"
|
|
"#,
|
|
)
|
|
.write(codex_home.path())?;
|
|
|
|
let mut app_server = TestAppServer::builder()
|
|
.with_codex_home(codex_home.path())
|
|
.build_initialized()
|
|
.await?;
|
|
|
|
let ThreadStartResponse { thread, .. } = app_server
|
|
.start_thread(ThreadStartParams::default())
|
|
.await?;
|
|
|
|
let _: TurnStartResponse = app_server
|
|
.request(|request_id| ClientRequest::TurnStart {
|
|
request_id,
|
|
params: TurnStartParams {
|
|
thread_id: thread.id.clone(),
|
|
input: vec![UserInput::Text {
|
|
text: "What time is it?".to_string(),
|
|
text_elements: Vec::new(),
|
|
}],
|
|
..Default::default()
|
|
},
|
|
})
|
|
.await?;
|
|
|
|
let server_request = timeout(
|
|
DEFAULT_READ_TIMEOUT,
|
|
app_server.read_stream_until_request_message(),
|
|
)
|
|
.await??;
|
|
let ServerRequest::CurrentTimeRead { request_id, params } = server_request else {
|
|
panic!("expected CurrentTimeRead request, got: {server_request:?}");
|
|
};
|
|
assert_eq!(params.thread_id, thread.id);
|
|
app_server
|
|
.send_response(
|
|
request_id,
|
|
serde_json::to_value(CurrentTimeReadResponse {
|
|
current_time_at: CURRENT_TIME_AT,
|
|
})?,
|
|
)
|
|
.await?;
|
|
timeout(
|
|
DEFAULT_READ_TIMEOUT,
|
|
app_server.read_stream_until_notification_message("turn/completed"),
|
|
)
|
|
.await??;
|
|
|
|
assert!(
|
|
response_mock
|
|
.single_request()
|
|
.message_input_texts("developer")
|
|
.iter()
|
|
.any(|text| text == CURRENT_TIME_REMINDER)
|
|
);
|
|
Ok(())
|
|
}
|