mockforge_data/

mock_generator.rs

//! Enhanced Mock Data Generator for OpenAPI Specifications
//!
//! This module provides comprehensive mock data generation capabilities that go beyond
//! the basic schema generator, offering intelligent data generation based on OpenAPI
//! specifications with type safety and realistic data patterns.

use crate::faker::EnhancedFaker;
use crate::schema::{FieldDefinition, SchemaDefinition};
use crate::{Error, Result};
use serde_json::{json, Value};
use std::collections::HashMap;
use tracing::{debug, info, warn};

/// Configuration for mock data generation
#[derive(Debug, Clone)]
pub struct MockGeneratorConfig {
    /// Whether to use realistic data patterns
    pub realistic_mode: bool,
    /// Default array size for generated arrays
    pub default_array_size: usize,
    /// Maximum array size for generated arrays
    pub max_array_size: usize,
    /// Whether to include optional fields
    pub include_optional_fields: bool,
    /// Custom field mappings for specific field names
    pub field_mappings: HashMap<String, String>,
    /// Whether to validate generated data against schemas
    pub validate_generated_data: bool,
}

impl Default for MockGeneratorConfig {
    fn default() -> Self {
        Self {
            realistic_mode: true,
            default_array_size: 3,
            max_array_size: 10,
            include_optional_fields: true,
            field_mappings: HashMap::new(),
            validate_generated_data: true,
        }
    }
}

impl MockGeneratorConfig {
    /// Create a new configuration with realistic defaults
    pub fn new() -> Self {
        Self::default()
    }

    /// Enable realistic data generation
    pub fn realistic_mode(mut self, enabled: bool) -> Self {
        self.realistic_mode = enabled;
        self
    }

    /// Set default array size
    pub fn default_array_size(mut self, size: usize) -> Self {
        self.default_array_size = size;
        self
    }

    /// Set maximum array size
    pub fn max_array_size(mut self, size: usize) -> Self {
        self.max_array_size = size;
        self
    }

    /// Control whether to include optional fields
    pub fn include_optional_fields(mut self, include: bool) -> Self {
        self.include_optional_fields = include;
        self
    }

    /// Add a custom field mapping
    pub fn field_mapping(mut self, field_name: String, faker_type: String) -> Self {
        self.field_mappings.insert(field_name, faker_type);
        self
    }

    /// Enable/disable data validation
    pub fn validate_generated_data(mut self, validate: bool) -> Self {
        self.validate_generated_data = validate;
        self
    }
}

/// Enhanced mock data generator with intelligent schema analysis
#[derive(Debug)]
pub struct MockDataGenerator {
    /// Configuration for the generator
    config: MockGeneratorConfig,
    /// Enhanced faker instance
    faker: EnhancedFaker,
    /// Schema registry for complex types
    #[allow(dead_code)]
    schema_registry: HashMap<String, SchemaDefinition>,
    /// Field name patterns for intelligent mapping
    field_patterns: HashMap<String, String>,
}

impl MockDataGenerator {
    /// Create a new mock data generator with default configuration
    pub fn new() -> Self {
        Self::with_config(MockGeneratorConfig::new())
    }

    /// Create a new mock data generator with custom configuration
    pub fn with_config(config: MockGeneratorConfig) -> Self {
        let mut generator = Self {
            config,
            faker: EnhancedFaker::new(),
            schema_registry: HashMap::new(),
            field_patterns: Self::create_field_patterns(),
        };

        // Initialize with common schema patterns
        generator.initialize_common_schemas();
        generator
    }

    /// Generate mock data from an OpenAPI specification
    pub fn generate_from_openapi_spec(&mut self, spec: &Value) -> Result<MockDataResult> {
        info!("Generating mock data from OpenAPI specification");

        // Parse the OpenAPI spec
        let openapi_spec = self.parse_openapi_spec(spec)?;

        // Extract all schemas from the spec
        let schemas = self.extract_schemas_from_spec(spec)?;

        // Generate mock data for each schema
        let mut generated_data = HashMap::new();
        let mut warnings = Vec::new();

        for (schema_name, schema_def) in schemas {
            debug!("Generating data for schema: {}", schema_name);

            match self.generate_schema_data(&schema_def) {
                Ok(data) => {
                    generated_data.insert(schema_name, data);
                }
                Err(e) => {
                    let warning =
                        format!("Failed to generate data for schema '{}': {}", schema_name, e);
                    warn!("{}", warning);
                    warnings.push(warning);
                }
            }
        }

        // Generate mock responses for each endpoint
        let mut mock_responses = HashMap::new();
        for (path, path_item) in &openapi_spec.paths {
            for (method, operation) in path_item.operations() {
                let endpoint_key = format!("{} {}", method.to_uppercase(), path);

                // Generate mock response for this endpoint
                if let Some(response_data) = self.generate_endpoint_response(operation)? {
                    mock_responses.insert(endpoint_key, response_data);
                }
            }
        }

        Ok(MockDataResult {
            schemas: generated_data,
            responses: mock_responses,
            warnings,
            spec_info: openapi_spec.info,
        })
    }

    /// Generate mock data from a JSON Schema
    pub fn generate_from_json_schema(&mut self, schema: &Value) -> Result<Value> {
        debug!("Generating mock data from JSON Schema");

        // Convert JSON Schema to our internal schema format
        let schema_def = SchemaDefinition::from_json_schema(schema)?;

        // Generate data using our enhanced generator
        self.generate_schema_data(&schema_def)
    }

    /// Generate mock data for a specific schema definition
    fn generate_schema_data(&mut self, schema: &SchemaDefinition) -> Result<Value> {
        let mut object = serde_json::Map::new();

        for field in &schema.fields {
            // Skip optional fields if configured to do so
            if !field.required && !self.config.include_optional_fields {
                continue;
            }

            // Determine the best faker type for this field
            let faker_type = self.determine_faker_type(field);

            // Generate the value
            let value = self.generate_field_value(field, &faker_type)?;

            // Validate the generated value if configured
            if self.config.validate_generated_data {
                field.validate_value(&value)?;
            }

            object.insert(field.name.clone(), value);
        }

        Ok(Value::Object(object))
    }

    /// Generate mock response for an OpenAPI operation
    fn generate_endpoint_response(
        &mut self,
        operation: &openapiv3::Operation,
    ) -> Result<Option<MockResponse>> {
        // Find the best response to mock (prefer 200, then 201, then any 2xx)
        let response_schema = self.find_best_response_schema(operation)?;

        if let Some(schema) = response_schema {
            let mock_data = self.generate_from_json_schema(&schema)?;

            Ok(Some(MockResponse {
                status: 200, // Default to 200 for successful responses
                headers: HashMap::new(),
                body: mock_data,
            }))
        } else {
            Ok(None)
        }
    }

    /// Find the best response schema from an operation
    fn find_best_response_schema(&self, operation: &openapiv3::Operation) -> Result<Option<Value>> {
        let responses = &operation.responses;

        // Look for 200 response first
        if let Some(response) = responses.responses.get(&openapiv3::StatusCode::Code(200)) {
            if let Some(schema) = self.extract_response_schema(response)? {
                return Ok(Some(schema));
            }
        }

        // Look for 201 response
        if let Some(response) = responses.responses.get(&openapiv3::StatusCode::Code(201)) {
            if let Some(schema) = self.extract_response_schema(response)? {
                return Ok(Some(schema));
            }
        }

        // Look for any 2xx response
        for (code, response) in &responses.responses {
            if let openapiv3::StatusCode::Code(status_code) = code {
                if *status_code >= 200 && *status_code < 300 {
                    if let Some(schema) = self.extract_response_schema(response)? {
                        return Ok(Some(schema));
                    }
                }
            }
        }

        Ok(None)
    }

    /// Extract schema from an OpenAPI response
    fn extract_response_schema(
        &self,
        response: &openapiv3::ReferenceOr<openapiv3::Response>,
    ) -> Result<Option<Value>> {
        match response {
            openapiv3::ReferenceOr::Item(response) => {
                let content = &response.content;
                // Prefer application/json content
                if let Some(json_content) = content.get("application/json") {
                    if let Some(schema) = &json_content.schema {
                        return Ok(Some(serde_json::to_value(schema)?));
                    }
                }

                // Fall back to any content type
                for (_, media_type) in content {
                    if let Some(schema) = &media_type.schema {
                        return Ok(Some(serde_json::to_value(schema)?));
                    }
                }

                Ok(None)
            }
            openapiv3::ReferenceOr::Reference { .. } => {
                // Handle reference responses (could be expanded)
                Ok(None)
            }
        }
    }

    /// Determine the best faker type for a field based on its name and type
    fn determine_faker_type(&self, field: &FieldDefinition) -> String {
        let field_name = field.name.to_lowercase();

        // Check custom field mappings first
        if let Some(mapped_type) = self.config.field_mappings.get(&field_name) {
            return mapped_type.clone();
        }

        // Use field name patterns for intelligent mapping
        for (pattern, faker_type) in &self.field_patterns {
            if field_name.contains(pattern) {
                return faker_type.clone();
            }
        }

        // Fall back to field type
        field.field_type.clone()
    }

    /// Generate a value for a specific field
    fn generate_field_value(&mut self, field: &FieldDefinition, faker_type: &str) -> Result<Value> {
        // Use faker template if provided
        if let Some(template) = &field.faker_template {
            return Ok(self.faker.generate_by_type(template));
        }

        // Generate based on determined faker type
        let value = self.faker.generate_by_type(faker_type);

        // Apply constraints if present
        self.apply_constraints(&value, field)
    }

    /// Apply constraints to a generated value
    fn apply_constraints(&mut self, value: &Value, field: &FieldDefinition) -> Result<Value> {
        let mut constrained_value = value.clone();

        // Apply numeric constraints
        if let Value::Number(num) = value {
            if let Some(minimum) = field.constraints.get("minimum") {
                if let Some(min_val) = minimum.as_f64() {
                    if num.as_f64().unwrap_or(0.0) < min_val {
                        constrained_value = json!(min_val);
                    }
                }
            }

            if let Some(maximum) = field.constraints.get("maximum") {
                if let Some(max_val) = maximum.as_f64() {
                    if num.as_f64().unwrap_or(0.0) > max_val {
                        constrained_value = json!(max_val);
                    }
                }
            }
        }

        // Apply string constraints
        if let Value::String(s) = value {
            let mut constrained_string = s.clone();

            // Apply min/max length constraints
            if let Some(min_length) = field.constraints.get("minLength") {
                if let Some(min_len) = min_length.as_u64() {
                    if constrained_string.len() < min_len as usize {
                        // Pad with random characters
                        let padding_needed = min_len as usize - constrained_string.len();
                        let padding = self.faker.string(padding_needed);
                        constrained_string = format!("{}{}", constrained_string, padding);
                    }
                }
            }

            if let Some(max_length) = field.constraints.get("maxLength") {
                if let Some(max_len) = max_length.as_u64() {
                    if constrained_string.len() > max_len as usize {
                        constrained_string.truncate(max_len as usize);
                    }
                }
            }

            constrained_value = json!(constrained_string);
        }

        // Apply enum constraints
        if let Some(enum_values) = field.constraints.get("enum") {
            if let Some(enum_array) = enum_values.as_array() {
                if !enum_array.is_empty() {
                    if let Some(random_value) = self.faker.random_element(enum_array) {
                        constrained_value = random_value.clone();
                    }
                }
            }
        }

        Ok(constrained_value)
    }

    /// Parse OpenAPI specification
    fn parse_openapi_spec(&self, spec: &Value) -> Result<OpenApiSpec> {
        // This is a simplified parser - in a real implementation,
        // you'd want to use a proper OpenAPI parser
        let spec_obj = spec
            .as_object()
            .ok_or_else(|| Error::generic("Invalid OpenAPI specification"))?;

        let info = spec_obj
            .get("info")
            .ok_or_else(|| Error::generic("Missing 'info' section in OpenAPI spec"))?;

        let title = info.get("title").and_then(|t| t.as_str()).unwrap_or("Unknown API").to_string();

        let version = info.get("version").and_then(|v| v.as_str()).unwrap_or("1.0.0").to_string();

        let description = info.get("description").and_then(|d| d.as_str()).map(|s| s.to_string());

        Ok(OpenApiSpec {
            info: OpenApiInfo {
                title,
                version,
                description,
            },
            paths: HashMap::new(), // Simplified for this example
        })
    }

    /// Extract schemas from OpenAPI specification
    fn extract_schemas_from_spec(
        &mut self,
        spec: &Value,
    ) -> Result<HashMap<String, SchemaDefinition>> {
        let mut schemas = HashMap::new();

        // Extract component schemas
        if let Some(components) = spec.get("components") {
            if let Some(schemas_section) = components.get("schemas") {
                if let Some(schema_obj) = schemas_section.as_object() {
                    for (name, schema_def) in schema_obj {
                        let schema = SchemaDefinition::from_json_schema(schema_def)?;
                        schemas.insert(name.clone(), schema);
                    }
                }
            }
        }

        // Extract schemas from paths
        if let Some(paths) = spec.get("paths") {
            if let Some(paths_obj) = paths.as_object() {
                for (path, path_item) in paths_obj {
                    if let Some(path_obj) = path_item.as_object() {
                        for (method, operation) in path_obj {
                            if let Some(op_obj) = operation.as_object() {
                                // Extract request body schemas
                                if let Some(request_body) = op_obj.get("requestBody") {
                                    if let Some(content) = request_body.get("content") {
                                        if let Some(json_content) = content.get("application/json")
                                        {
                                            if let Some(schema) = json_content.get("schema") {
                                                let schema_name = format!(
                                                    "{}_{}_request",
                                                    path.replace("/", "_").trim_start_matches("_"),
                                                    method
                                                );
                                                let schema_def =
                                                    SchemaDefinition::from_json_schema(schema)?;
                                                schemas.insert(schema_name, schema_def);
                                            }
                                        }
                                    }
                                }

                                // Extract response schemas
                                if let Some(responses) = op_obj.get("responses") {
                                    if let Some(resp_obj) = responses.as_object() {
                                        for (status_code, response) in resp_obj {
                                            if let Some(content) = response.get("content") {
                                                if let Some(json_content) =
                                                    content.get("application/json")
                                                {
                                                    if let Some(schema) = json_content.get("schema")
                                                    {
                                                        let schema_name = format!(
                                                            "{}_{}_response_{}",
                                                            path.replace("/", "_")
                                                                .trim_start_matches("_"),
                                                            method,
                                                            status_code
                                                        );
                                                        let schema_def =
                                                            SchemaDefinition::from_json_schema(
                                                                schema,
                                                            )?;
                                                        schemas.insert(schema_name, schema_def);
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }

        Ok(schemas)
    }

    /// Create field name patterns for intelligent mapping
    fn create_field_patterns() -> HashMap<String, String> {
        let mut patterns = HashMap::new();

        // Email patterns
        patterns.insert("email".to_string(), "email".to_string());
        patterns.insert("mail".to_string(), "email".to_string());

        // Name patterns
        patterns.insert("name".to_string(), "name".to_string());
        patterns.insert("firstname".to_string(), "name".to_string());
        patterns.insert("lastname".to_string(), "name".to_string());
        patterns.insert("username".to_string(), "name".to_string());

        // Phone patterns
        patterns.insert("phone".to_string(), "phone".to_string());
        patterns.insert("mobile".to_string(), "phone".to_string());
        patterns.insert("telephone".to_string(), "phone".to_string());

        // Address patterns
        patterns.insert("address".to_string(), "address".to_string());
        patterns.insert("street".to_string(), "address".to_string());
        patterns.insert("city".to_string(), "string".to_string());
        patterns.insert("state".to_string(), "string".to_string());
        patterns.insert("zip".to_string(), "string".to_string());
        patterns.insert("postal".to_string(), "string".to_string());

        // Company patterns
        patterns.insert("company".to_string(), "company".to_string());
        patterns.insert("organization".to_string(), "company".to_string());
        patterns.insert("corp".to_string(), "company".to_string());

        // URL patterns
        patterns.insert("url".to_string(), "url".to_string());
        patterns.insert("website".to_string(), "url".to_string());
        patterns.insert("link".to_string(), "url".to_string());

        // Date patterns
        patterns.insert("date".to_string(), "date".to_string());
        patterns.insert("created".to_string(), "date".to_string());
        patterns.insert("updated".to_string(), "date".to_string());
        patterns.insert("timestamp".to_string(), "date".to_string());

        // ID patterns
        patterns.insert("id".to_string(), "uuid".to_string());
        patterns.insert("uuid".to_string(), "uuid".to_string());
        patterns.insert("guid".to_string(), "uuid".to_string());

        // IP patterns
        patterns.insert("ip".to_string(), "ip".to_string());
        patterns.insert("ipv4".to_string(), "ip".to_string());
        patterns.insert("ipv6".to_string(), "ip".to_string());

        patterns
    }

    /// Initialize common schemas
    fn initialize_common_schemas(&mut self) {
        // Add common schema patterns here
        // This could include User, Product, Order, etc.
    }
}

impl Default for MockDataGenerator {
    fn default() -> Self {
        Self::new()
    }
}

/// Result of mock data generation
#[derive(Debug, Clone, serde::Serialize)]
pub struct MockDataResult {
    /// Generated data for each schema
    pub schemas: HashMap<String, Value>,
    /// Generated mock responses for each endpoint
    pub responses: HashMap<String, MockResponse>,
    /// Warnings encountered during generation
    pub warnings: Vec<String>,
    /// OpenAPI specification info
    pub spec_info: OpenApiInfo,
}

/// Mock response data
#[derive(Debug, Clone, serde::Serialize)]
pub struct MockResponse {
    /// HTTP status code
    pub status: u16,
    /// Response headers
    pub headers: HashMap<String, String>,
    /// Response body
    pub body: Value,
}

/// OpenAPI specification info
#[derive(Debug, Clone, serde::Serialize)]
pub struct OpenApiInfo {
    /// API title
    pub title: String,
    /// API version
    pub version: String,
    /// API description
    pub description: Option<String>,
}

/// Simplified OpenAPI spec structure
#[derive(Debug)]
struct OpenApiSpec {
    info: OpenApiInfo,
    paths: HashMap<String, PathItem>,
}

/// Simplified path item structure
#[derive(Debug)]
struct PathItem {
    operations: HashMap<String, openapiv3::Operation>,
}

impl PathItem {
    fn operations(&self) -> &HashMap<String, openapiv3::Operation> {
        &self.operations
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_mock_generator_config_default() {
        let config = MockGeneratorConfig::default();

        assert!(config.realistic_mode);
        assert_eq!(config.default_array_size, 3);
        assert_eq!(config.max_array_size, 10);
        assert!(config.include_optional_fields);
        assert!(config.validate_generated_data);
    }

    #[test]
    fn test_mock_generator_config_custom() {
        let config = MockGeneratorConfig::new()
            .realistic_mode(false)
            .default_array_size(5)
            .max_array_size(20)
            .include_optional_fields(false)
            .field_mapping("email".to_string(), "email".to_string())
            .validate_generated_data(false);

        assert!(!config.realistic_mode);
        assert_eq!(config.default_array_size, 5);
        assert_eq!(config.max_array_size, 20);
        assert!(!config.include_optional_fields);
        assert!(!config.validate_generated_data);
        assert!(config.field_mappings.contains_key("email"));
    }

    #[test]
    fn test_mock_data_generator_new() {
        let generator = MockDataGenerator::new();

        assert!(generator.config.realistic_mode);
        assert!(!generator.field_patterns.is_empty());
    }

    #[test]
    fn test_mock_data_generator_with_config() {
        let config = MockGeneratorConfig::new().realistic_mode(false).default_array_size(10);

        let generator = MockDataGenerator::with_config(config);

        assert!(!generator.config.realistic_mode);
        assert_eq!(generator.config.default_array_size, 10);
    }

    #[test]
    fn test_determine_faker_type_custom_mapping() {
        let mut config = MockGeneratorConfig::new();
        config.field_mappings.insert("user_email".to_string(), "email".to_string());

        let generator = MockDataGenerator::with_config(config);

        let field = FieldDefinition::new("user_email".to_string(), "string".to_string());
        let faker_type = generator.determine_faker_type(&field);

        assert_eq!(faker_type, "email");
    }

    #[test]
    fn test_determine_faker_type_pattern_matching() {
        let generator = MockDataGenerator::new();

        let field = FieldDefinition::new("email_address".to_string(), "string".to_string());
        let faker_type = generator.determine_faker_type(&field);

        assert_eq!(faker_type, "email");
    }

    #[test]
    fn test_determine_faker_type_fallback() {
        let generator = MockDataGenerator::new();

        let field = FieldDefinition::new("unknown_field".to_string(), "integer".to_string());
        let faker_type = generator.determine_faker_type(&field);

        assert_eq!(faker_type, "integer");
    }

    #[test]
    fn test_field_patterns_creation() {
        let patterns = MockDataGenerator::create_field_patterns();

        assert!(patterns.contains_key("email"));
        assert!(patterns.contains_key("name"));
        assert!(patterns.contains_key("phone"));
        assert!(patterns.contains_key("address"));
        assert!(patterns.contains_key("company"));
        assert!(patterns.contains_key("url"));
        assert!(patterns.contains_key("date"));
        assert!(patterns.contains_key("id"));
        assert!(patterns.contains_key("ip"));
    }

    #[test]
    fn test_generate_from_json_schema_simple() {
        let mut generator = MockDataGenerator::new();

        let schema = json!({
            "type": "object",
            "properties": {
                "name": { "type": "string" },
                "age": { "type": "integer" },
                "email": { "type": "string" }
            },
            "required": ["name", "age"]
        });

        let result = generator.generate_from_json_schema(&schema).unwrap();

        assert!(result.is_object());
        let obj = result.as_object().unwrap();
        assert!(obj.contains_key("name"));
        assert!(obj.contains_key("age"));
        assert!(obj.contains_key("email"));
    }

    #[test]
    fn test_generate_from_json_schema_with_constraints() {
        let mut generator = MockDataGenerator::new();

        let schema = json!({
            "type": "object",
            "properties": {
                "age": {
                    "type": "integer",
                    "minimum": 18,
                    "maximum": 65
                },
                "name": {
                    "type": "string",
                    "minLength": 5,
                    "maxLength": 20
                }
            }
        });

        let result = generator.generate_from_json_schema(&schema).unwrap();

        assert!(result.is_object());
        let obj = result.as_object().unwrap();

        if let Some(age) = obj.get("age") {
            if let Some(age_num) = age.as_i64() {
                assert!(age_num >= 18);
                assert!(age_num <= 65);
            }
        }
    }

    #[test]
    fn test_generate_from_json_schema_with_enum() {
        let mut generator = MockDataGenerator::new();

        let schema = json!({
            "type": "object",
            "properties": {
                "status": {
                    "type": "string",
                    "enum": ["active", "inactive", "pending"]
                }
            }
        });

        let result = generator.generate_from_json_schema(&schema).unwrap();

        assert!(result.is_object());
        let obj = result.as_object().unwrap();

        if let Some(status) = obj.get("status") {
            if let Some(status_str) = status.as_str() {
                assert!(["active", "inactive", "pending"].contains(&status_str));
            }
        }
    }
}