Skip to main content

mockforge_bench/
lib.rs

1//! Load and performance testing module for MockForge
2//!
3//! This module provides functionality to run load tests against real API endpoints
4//! using OpenAPI specifications to generate realistic traffic patterns.
5
6pub mod chunked_bench;
7pub mod cloud_api;
8pub mod command;
9pub mod conformance;
10pub mod crud_flow;
11pub mod data_driven;
12pub mod dynamic_params;
13pub mod error;
14pub mod executor;
15pub mod invalid_data;
16pub mod k6_gen;
17pub mod mock_integration;
18pub mod owasp_api;
19pub mod parallel_executor;
20pub mod parallel_requests;
21pub mod param_overrides;
22pub mod preflight;
23pub mod reporter;
24pub mod request_gen;
25pub mod scenarios;
26pub mod security_payloads;
27pub mod spec_dependencies;
28pub mod spec_parser;
29pub mod ssrf;
30pub mod target_parser;
31pub mod wafbench;
32
33pub use cloud_api::{
34    run_bench, run_conformance, run_crud_flow, run_owasp, run_security, run_wafbench,
35    CloudBenchInputs, CloudConformanceInputs, CloudCrudFlowInputs, CloudOwaspInputs,
36    CloudRunArtifacts, CloudSecurityInputs, CloudWafBenchInputs, SpecFormat,
37};
38pub use command::BenchCommand;
39pub use crud_flow::{CrudFlow, CrudFlowConfig, CrudFlowDetector, FlowStep};
40pub use data_driven::{
41    DataDistribution, DataDrivenConfig, DataDrivenGenerator, DataMapping, PerUriColumns,
42};
43pub use error::{BenchError, Result};
44pub use invalid_data::{InvalidDataConfig, InvalidDataGenerator, InvalidDataType};
45pub use mock_integration::{MockIntegrationConfig, MockIntegrationGenerator, MockServerDetector};
46pub use parallel_executor::{AggregatedResults, TargetResult};
47pub use parallel_requests::{ParallelConfig, ParallelRequestGenerator};
48pub use param_overrides::{OperationOverrides, ParameterOverrides};
49pub use scenarios::LoadScenario;
50pub use security_payloads::{SecurityCategory, SecurityPayloads, SecurityTestConfig};
51pub use spec_dependencies::{
52    DependencyDetector, ExtractedValues, SpecDependency, SpecDependencyConfig, SpecGroup,
53};
54pub use ssrf::{validate_target_url, Policy as SsrfPolicy, SsrfError};
55pub use target_parser::{parse_targets_file, TargetConfig};
56pub use wafbench::{WafBenchLoader, WafBenchStats, WafBenchTestCase};
57
58// OpenAPI 3.0.0 Conformance Testing exports
59pub use conformance::{
60    generate_custom_yaml_from_har, ConformanceConfig, ConformanceFeature, ConformanceGenerator,
61    ConformanceReport, ConformanceSarifReport, HarToCustomOptions, SchemaValidatorGenerator,
62    SpecDrivenConformanceGenerator,
63};
64
65// OWASP API Security Top 10 exports
66pub use owasp_api::{
67    AuthToken, DiscoveryConfig, OwaspApiConfig, OwaspApiGenerator, OwaspCategory, OwaspFinding,
68    OwaspPayload, OwaspPayloadGenerator, OwaspReport, OwaspScanInfo, OwaspSummary, OwaspValidator,
69    RateLimitConfig, ReportFormat, Severity, SsrfConfig, ValidationResult,
70};