Skip to main content

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;
11#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
12pub mod streaming_conformance;
13#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
14mod streaming_conformance_suite;
15mod tracing_isolation;
16
17pub use completion::{MockCompletionModel, MockError, MockTurn};
18pub use embeddings::{MockEmbeddingModel, MockMultiTextDocument, MockTextDocument};
19pub use http::{
20    CapturedHttpRequest, HttpErrorStreamingClient, MockHttpResponse, MockStreamingClient,
21    RecordingHttpClient, SequencedHttpClient, SequencedStreamingHttpClient,
22};
23pub use memory::{AppendFailingMemory, CountingMemory, FailingMemory};
24pub use model_listing::MockModelLister;
25pub use streaming::{MOCK_PROVIDER, MockStreamEvent, mock_final, mock_final_with_total_tokens};
26pub use tracing_isolation::{
27    scoped_tracing_subscriber_guard, scoped_tracing_subscriber_guard_blocking,
28};