rig_core/test_utils/
mod.rs

1//! Test utilities for deterministic completion-model tests.
2
3mod completion;
4mod embeddings;
5mod http;
6#[cfg(test)]
7pub(crate) mod internal_streaming_profiles;
8mod memory;
9mod model_listing;
10mod streaming;
11mod tools;
12mod tracing_isolation;
13
14pub use completion::{MockCompletionModel, MockError, MockTurn};
15pub use embeddings::{MockEmbeddingModel, MockMultiTextDocument, MockTextDocument};
16pub use http::{
17    CapturedHttpRequest, HttpErrorStreamingClient, MockHttpResponse, MockStreamingClient,
18    RecordingHttpClient, SequencedStreamingHttpClient,
19};
20pub use memory::{AppendFailingMemory, CountingMemory, FailingMemory};
21pub use model_listing::MockModelLister;
22pub use streaming::{MockResponse, MockStreamEvent};
23pub use tools::{
24    BarrierMockToolIndex, MockAddTool, MockBarrierTool, MockControlledTool, MockDeniedTool,
25    MockExampleTool, MockExtensionsProbeTool, MockFailingTool, MockFailure, MockHandledFailureTool,
26    MockImageGeneratorTool, MockImageOutputTool, MockMetadataTool, MockObjectOutputTool,
27    MockOperationArgs, MockRequestId, MockStringOutputTool, MockSubtractTool, MockToolError,
28    MockToolIndex, SessionId, mock_math_toolset,
29};
30pub use tracing_isolation::{
31    scoped_tracing_subscriber_guard, scoped_tracing_subscriber_guard_blocking,
32};
rig_core/test_utils/mod.rs

rig_core/test_utils/
mod.rs
