From 2f2ec5a5f06c334a4a3a77b781488f8945a6106f Mon Sep 17 00:00:00 2001
From: Ahmed Ibrahim <aibrahim@openai.com>
Date: Thu, 4 Dec 2025 11:36:38 -0800
Subject: [PATCH] wire

---
 codex-rs/core/tests/models_manager.rs | 62 +++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 codex-rs/core/tests/models_manager.rs

diff --git a/codex-rs/core/tests/models_manager.rs b/codex-rs/core/tests/models_manager.rs
new file mode 100644
index 0000000000..29523fb285
--- /dev/null
+++ b/codex-rs/core/tests/models_manager.rs
@@ -0,0 +1,62 @@
+use codex_core::WireApi;
+use codex_core::create_oss_provider_with_base_url;
+use codex_core::openai_models::models_manager::ModelsManager;
+use codex_protocol::openai_models::ClientVersion;
+use codex_protocol::openai_models::ModelInfo;
+use codex_protocol::openai_models::ModelVisibility;
+use codex_protocol::openai_models::ModelsResponse;
+use codex_protocol::openai_models::ReasoningLevel;
+use codex_protocol::openai_models::ShellType;
+use core_test_support::responses::mount_models_once;
+use core_test_support::responses::start_mock_server;
+use pretty_assertions::assert_eq;
+
+#[tokio::test]
+async fn fetches_models_via_models_endpoint() {
+    let server = start_mock_server().await;
+    let body = ModelsResponse {
+        models: vec![ModelInfo {
+            slug: "gpt-test".to_string(),
+            display_name: "gpt-test".to_string(),
+            description: Some("desc".to_string()),
+            default_reasoning_level: ReasoningLevel::Medium,
+            supported_reasoning_levels: vec![
+                ReasoningLevel::Low,
+                ReasoningLevel::Medium,
+                ReasoningLevel::High,
+            ],
+            shell_type: ShellType::ShellCommand,
+            visibility: ModelVisibility::List,
+            minimal_client_version: ClientVersion(0, 99, 0),
+            supported_in_api: true,
+            priority: 1,
+        }],
+    };
+    let models_mock = mount_models_once(&server, body.clone()).await;
+
+    let base_url = format!("{}/api/codex", server.uri());
+    let provider = create_oss_provider_with_base_url(&base_url, WireApi::Responses);
+    let manager = ModelsManager::new(None);
+
+    let models = manager
+        .fetch_models_from_api(&provider, None)
+        .await
+        .expect("fetch models");
+
+    assert_eq!(models, body.models);
+
+    let request = models_mock
+        .requests()
+        .into_iter()
+        .next()
+        .expect("models request captured");
+    assert_eq!(request.url.path(), "/api/codex/models");
+
+    let client_version = request
+        .url
+        .query_pairs()
+        .find(|(k, _)| k == "client_version")
+        .map(|(_, v)| v.to_string())
+        .expect("client_version query param");
+    assert_eq!(client_version, env!("CARGO_PKG_VERSION"));
+}