Skip to main content

rig_agent/test_utils/
mod.rs

1//! Test utilities for the classic runtime and its provider-facing acceptance tests.
2
3mod model_conformance;
4mod tools;
5
6pub use model_conformance::{
7    ConformanceToolError, ScenarioError, ScenarioReport, buffered_streaming_text_parity,
8    cancellation_and_max_turns, complex_tool_arguments, decode_structured_output,
9    hook_rewrites_and_request_patch, invalid_tool_recovery, optional_argument, parallel_tools,
10    sequential_tools, streaming_structured_after_tool, streaming_tool, structured_after_tool,
11    structured_extraction, tool_choice_modes, tool_output_serialization,
12    validate_cancelled_failure, validate_extraction_fields, validate_max_turns_failure,
13    validate_protocol_hygiene, validate_result_redaction, validate_rewritten_arguments,
14    validate_unknown_tool_failure, zero_argument_tool,
15};
16pub use rig_core::test_utils::*;
17pub use tools::{
18    BarrierMockToolIndex, MockAddTool, MockBarrierTool, MockContextProbeTool, MockControlledTool,
19    MockDeniedTool, MockExampleTool, MockFailingTool, MockFailure, MockHandledFailureTool,
20    MockImageGeneratorTool, MockImageOutputTool, MockMetadataTool, MockObjectOutputTool,
21    MockOperationArgs, MockRequestId, MockStringOutputTool, MockSubtractTool, MockToolError,
22    MockToolIndex, SessionId, mock_math_toolset,
23};