mockforge_intelligence/behavioral_cloning/
mod.rs

1//! Behavioral Cloning of Backends
2//!
3//! This module provides functionality to learn from recorded traffic and create
4//! realistic mock behavior that captures the "personality" of the real backend.
5//!
6//! # Features
7//!
8//! - **Sequence Learning**: Discover and model multi-step flows from real traffic
9//! - **Probabilistic Outcomes**: Model probabilities of errors, latency, and edge cases per endpoint
10//! - **Rare Edge Amplification**: Option to increase rare error frequency for testing
11//!
12//! # Example Usage
13//!
14//! ```rust,ignore
15//! use mockforge_core::behavioral_cloning::{
16//!     SequenceLearner, ProbabilisticModel, EdgeAmplifier,
17//! };
18//!
19//! async fn example(database: &impl mockforge_core::behavioral_cloning::TraceQueryProvider) -> mockforge_core::Result<()> {
20//!     // Learn sequences from recorded traffic
21//!     let sequences = SequenceLearner::discover_sequences_from_traces(database).await?;
22//!
23//!     // Build probability models for endpoints
24//!     let model = ProbabilisticModel::build_probability_model(
25//!         database,
26//!         "/api/users",
27//!         "GET"
28//!     ).await?;
29//!
30//!     // Sample a status code based on learned distribution
31//!     let status_code = model.sample_status_code();
32//!
33//!     // Amplify rare errors for testing
34//!     let amplifier = EdgeAmplifier::new();
35//!     amplifier.apply_amplification(&mut model.clone(), 0.5)?; // 50% frequency
36//!     Ok(())
37//! }
38//! ```
39
40pub mod edge_amplifier;
41pub mod probabilistic_model;
42pub mod sequence_learner;
43pub mod types;
44
45pub use edge_amplifier::EdgeAmplifier;
46pub use probabilistic_model::ProbabilisticModel;
47pub use sequence_learner::{SequenceLearner, TraceQueryProvider, TraceRequest};
48pub use types::{
49    AmplificationScope, BehavioralSequence, EdgeAmplificationConfig, EndpointProbabilityModel,
50    ErrorPattern, LatencyDistribution, PayloadVariation, SequenceStep,
51};
mockforge_intelligence/behavioral_cloning/mod.rs

mockforge_intelligence/behavioral_cloning/
mod.rs
