mirror of
https://github.com/openai/codex.git
synced 2026-03-05 21:45:28 +03:00
bd3ce98190
https://github.com/modelcontextprotocol/rust-sdk/pull/598 in 0.14 broke some MCP oauth (like Linear) and https://github.com/modelcontextprotocol/rust-sdk/pull/641 fixed it in 0.15
80 lines
2.8 KiB
Rust
80 lines
2.8 KiB
Rust
#![allow(clippy::unwrap_used, clippy::expect_used)]
|
|
use std::borrow::Cow;
|
|
use std::fs;
|
|
use std::sync::Arc;
|
|
|
|
use anyhow::Result;
|
|
use exec_server_test_support::create_transport;
|
|
use pretty_assertions::assert_eq;
|
|
use rmcp::ServiceExt;
|
|
use rmcp::model::Tool;
|
|
use rmcp::model::object;
|
|
use serde_json::json;
|
|
use tempfile::TempDir;
|
|
|
|
/// Verify the list_tools call to the MCP server returns the expected response.
|
|
#[tokio::test(flavor = "current_thread")]
|
|
async fn list_tools() -> Result<()> {
|
|
let codex_home = TempDir::new()?;
|
|
let policy_dir = codex_home.path().join("rules");
|
|
fs::create_dir_all(&policy_dir)?;
|
|
fs::write(
|
|
policy_dir.join("default.rules"),
|
|
r#"prefix_rule(pattern=["ls"], decision="prompt")"#,
|
|
)?;
|
|
let dotslash_cache_temp_dir = TempDir::new()?;
|
|
let dotslash_cache = dotslash_cache_temp_dir.path();
|
|
let transport = create_transport(codex_home.path(), dotslash_cache).await?;
|
|
|
|
let service = ().serve(transport).await?;
|
|
let tools = service.list_tools(Default::default()).await?.tools;
|
|
assert_eq!(
|
|
vec![Tool {
|
|
name: Cow::Borrowed("shell"),
|
|
title: None,
|
|
description: Some(Cow::Borrowed(
|
|
"Runs a shell command and returns its output. You MUST provide the workdir as an absolute path."
|
|
)),
|
|
input_schema: Arc::new(object(json!({
|
|
"$schema": "https://json-schema.org/draft/2020-12/schema",
|
|
"properties": {
|
|
"command": {
|
|
"description": "The bash string to execute.",
|
|
"type": "string",
|
|
},
|
|
"login": {
|
|
"description": "Launch Bash with -lc instead of -c: defaults to true.",
|
|
"nullable": true,
|
|
"type": "boolean",
|
|
},
|
|
"timeout_ms": {
|
|
"description": "The timeout for the command in milliseconds.",
|
|
"format": "uint64",
|
|
"minimum": 0,
|
|
"nullable": true,
|
|
"type": "integer",
|
|
},
|
|
"workdir": {
|
|
"description": "The working directory to execute the command in. Must be an absolute path.",
|
|
"type": "string",
|
|
},
|
|
},
|
|
"required": [
|
|
"command",
|
|
"workdir",
|
|
],
|
|
"title": "ExecParams",
|
|
"type": "object",
|
|
}))),
|
|
output_schema: None,
|
|
execution: None,
|
|
annotations: None,
|
|
icons: None,
|
|
meta: None
|
|
}],
|
|
tools
|
|
);
|
|
|
|
Ok(())
|
|
}
|