database-mcp 0.6.2

A single-binary MCP server for MySQL, MariaDB, PostgreSQL, and SQLite
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! Shared helpers for approval tests.
//!
//! Provides duplex transport setup and client lifecycle management
//! used by tool schema snapshot tests.

use database_mcp_server::Server;
use rmcp::service::{Peer, RunningService, ServiceExt};

/// Connects a [`Server`] over a duplex transport, runs a closure, then cleans up.
///
/// Handles the full client lifecycle: duplex creation, server spawn, closure
/// execution, client cancellation, and server join.
pub async fn run_with_client<F, Fut>(server: Server, f: F)
where
    F: FnOnce(Peer<rmcp::RoleClient>) -> Fut,
    Fut: Future<Output = ()>,
{
    let (server_transport, client_transport) = tokio::io::duplex(4096);

    let server_handle = tokio::spawn(async move {
        let running = server.serve(server_transport).await.expect("server serve failed");
        running.waiting().await.ok();
    });

    let client: RunningService<rmcp::RoleClient, ()> = ().serve(client_transport).await.expect("client serve failed");

    f(client.peer().clone()).await;

    client.cancel().await.expect("client cancel failed");
    server_handle.await.expect("server task failed");
}