Struct TestContext 

pub struct TestContext { /* private fields */ }

Test context wrapper providing convenient testing utilities.

Wraps Cx::for_testing() and provides helper methods for:

• Budget/timeout configuration
• Creating McpContext instances
• Running async operations with cleanup

Example

let ctx = TestContext::new();
let mcp_ctx = ctx.mcp_context(1);  // Request ID 1

// With custom budget
let ctx = TestContext::new().with_budget_secs(30);

Implementations

impl TestContext

pub fn new() -> Self

Creates a new test context using Cx::for_testing().

Example

let ctx = TestContext::new();

pub fn with_budget_secs(self, secs: u64) -> Self

Creates a test context with a budget timeout.

Arguments

• secs - Timeout in seconds

Example

let ctx = TestContext::new().with_budget_secs(5);

pub fn with_budget_ms(self, ms: u64) -> Self

Creates a test context with a budget timeout in milliseconds.

Arguments

• ms - Timeout in milliseconds

pub fn with_session_state(self, state: SessionState) -> Self

Creates a test context with shared session state.

Useful for testing state persistence across multiple contexts.

Example

let state = SessionState::new();
let ctx1 = TestContext::new().with_session_state(state.clone());
let ctx2 = TestContext::new().with_session_state(state.clone());
// Both contexts share the same session state

pub fn cx(&self) -> &Cx

Returns the underlying Cx.

pub fn cx_clone(&self) -> Cx

Returns a clone of the underlying Cx.

pub fn budget(&self) -> Option<&Budget>

Returns the budget if configured.

pub fn mcp_context(&self, request_id: u64) -> McpContext

Creates an McpContext for handler testing.

Arguments

• request_id - The request ID for this context

Example

let ctx = TestContext::new();
let mcp_ctx = ctx.mcp_context(1);

// Use in handler testing
let result = my_tool_handler.call(&mcp_ctx, args)?;

pub fn mcp_context_with_state( &self, request_id: u64, state: SessionState, ) -> McpContext

Creates an McpContext with shared session state.

Arguments

• request_id - The request ID
• state - Session state to attach

pub fn is_cancelled(&self) -> bool

Checks if cancellation has been requested.

pub fn checkpoint(&self) -> McpResult<()>

Performs a cancellation checkpoint.

Returns Err(CancelledError) if cancellation was requested.

Trait Implementations

impl Clone for TestContext

fn clone(&self) -> TestContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TestContext

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for TestContext

fn default() -> Self

Returns the “default value” for a type.