// Compliance Engine Integration Tests for DQN Training
// Tests the integration of ComplianceValidator into DQNTrainer

use common::types::{OrderSide, OrderType, Price, Quantity, Symbol};
use ml::trainers::dqn::DQNTrainer;
use risk::compliance::{
    ClientClassification, ClientType, ComplianceValidator, PositionLimit, RegulatoryReportingConfig,
    RiskTolerance,
};
use risk::risk_types::{ComplianceConfig, OrderInfo, PositionLimits};
use std::collections::HashMap;
use std::sync::Arc;

// Helper to create test compliance config
fn create_test_compliance_config() -> ComplianceConfig {
    ComplianceConfig {
        rules: vec![],
        position_limits: PositionLimits {
            max_position_per_instrument: HashMap::new(),
            max_portfolio_value: Price::from_f64(1_000_000.0).unwrap(),
            max_leverage: 5.0,
            max_concentration_pct: 0.1,
            global_limit: Price::from_f64(10_000_000.0).unwrap(),
        },
        audit_retention_days: 2555,
        market_abuse_threshold: Some(Price::from_f64(100_000.0).unwrap()),
        large_exposure_threshold: Price::from_f64(500_000.0).unwrap(),
    }
}

#[tokio::test]
async fn test_compliance_engine_initialization() {
    // Test that DQNTrainer can be created with compliance engine
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Verify compliance engine is initialized
    assert_eq!(compliance_engine.compliance_rule_count().await, 0);
}

#[tokio::test]
async fn test_position_limit_enforcement() {
    // Test that position limits are enforced during training
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Set strict position limit
    let limit = PositionLimit {
        instrument_id: "ES_FUT".to_string(),
        max_position_size: Price::from_f64(5.0).unwrap(), // Very strict limit
        max_daily_turnover: Price::from_f64(50_000.0).unwrap(),
        concentration_limit: Price::from_f64(0.05).unwrap(),
        regulatory_basis: "Test limit".to_string(),
    };

    compliance_engine
        .set_position_limit("ES_FUT".to_string(), limit)
        .await
        .unwrap();

    // Create order that exceeds limit
    let order = OrderInfo {
        order_id: "test_order_1".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(10.0).unwrap(), // Exceeds 5.0 limit
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Market),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    let result = compliance_engine.validate_order(&order, None).await.unwrap();

    // Should have violations due to position limit
    assert!(!result.is_compliant);
    assert!(!result.violations.is_empty());
}

#[tokio::test]
async fn test_trading_hours_enforcement() {
    // Test that trading hours restrictions are logged
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Compliance engine doesn't directly enforce trading hours,
    // but logs compliance checks
    let order = OrderInfo {
        order_id: "test_order_2".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Sell,
        quantity: Quantity::from_f64(2.0).unwrap(),
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Limit),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    let result = compliance_engine.validate_order(&order, None).await.unwrap();

    // Should pass basic validation (trading hours would be enforced by rule engine)
    assert!(result.is_compliant || !result.warnings.is_empty());
}

#[tokio::test]
async fn test_concentration_risk_warning() {
    // Test concentration risk warnings
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Large order that might trigger concentration warnings
    let order = OrderInfo {
        order_id: "test_order_3".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(1000.0).unwrap(),
        price: Price::from_f64(4500.0).unwrap(), // $4.5M order
        order_type: Some(OrderType::Market),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    let result = compliance_engine.validate_order(&order, None).await.unwrap();

    // May have warnings or flags for large position
    // This is acceptable as long as validation completes
    assert!(result.is_compliant || !result.warnings.is_empty() || !result.regulatory_flags.is_empty());
}

#[tokio::test]
async fn test_compliance_logging() {
    // Test that compliance checks create audit trail
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    let order = OrderInfo {
        order_id: "test_order_4".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(5.0).unwrap(),
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Limit),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    compliance_engine.validate_order(&order, None).await.unwrap();

    // Verify audit trail exists
    let audit_trail = compliance_engine.get_enhanced_audit_trail(Some(10)).await;
    assert!(!audit_trail.is_empty());
}

#[tokio::test]
async fn test_client_suitability() {
    // Test client classification and suitability checks
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Set conservative client classification
    let classification = ClientClassification {
        client_id: "conservative_client".to_string(),
        classification: ClientType::RetailClient,
        leverage_limit: Price::from_f64(3.0).unwrap(),
        risk_tolerance: RiskTolerance::Conservative,
        regulatory_restrictions: vec!["NO_HIGH_RISK_DERIVATIVES".to_string()],
    };

    compliance_engine
        .set_client_classification("conservative_client".to_string(), classification)
        .await
        .unwrap();

    // Large order for conservative client
    let order = OrderInfo {
        order_id: "test_order_5".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(100.0).unwrap(),
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Market),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    let result = compliance_engine
        .validate_order(&order, Some("conservative_client"))
        .await
        .unwrap();

    // Should have suitability warnings
    assert!(!result.warnings.is_empty());
}

#[tokio::test]
async fn test_compliance_metrics() {
    // Test compliance metrics collection
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Execute several compliance checks
    for i in 0..5 {
        let order = OrderInfo {
            order_id: format!("test_order_{}", i),
            symbol: Symbol::from("ES_FUT"),
            instrument_id: "ES_FUT".to_string(),
            side: if i % 2 == 0 { OrderSide::Buy } else { OrderSide::Sell },
            quantity: Quantity::from_f64(5.0).unwrap(),
            price: Price::from_f64(4500.0).unwrap(),
            order_type: Some(OrderType::Limit),
            portfolio_id: Some("test_portfolio".to_string()),
            strategy_id: Some("dqn_strategy".to_string()),
        };

        compliance_engine.validate_order(&order, None).await.unwrap();
    }

    // Get compliance metrics
    let metrics = compliance_engine.get_compliance_metrics().await;
    assert!(metrics.contains_key("total_audit_entries"));
    assert!(metrics.contains_key("compliance_rate"));
}

#[tokio::test]
async fn test_hot_reload_support() {
    // Test that compliance engine supports rule reloading
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    // Initial rule count should be 0
    let initial_count = compliance_engine.compliance_rule_count().await;
    assert_eq!(initial_count, 0);

    // Clear cache (simulating hot-reload)
    compliance_engine.clear_compliance_rules().await;
    let count_after_clear = compliance_engine.compliance_rule_count().await;
    assert_eq!(count_after_clear, 0);
}

#[tokio::test]
async fn test_violation_broadcasting() {
    // Test that violations are broadcast
    let config = create_test_compliance_config();
    let regulatory_config = RegulatoryReportingConfig::default();
    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));

    let mut violation_receiver = compliance_engine.subscribe_to_violations();

    // Set strict limit to trigger violation
    let limit = PositionLimit {
        instrument_id: "ES_FUT".to_string(),
        max_position_size: Price::from_f64(1.0).unwrap(), // Very strict
        max_daily_turnover: Price::from_f64(10_000.0).unwrap(),
        concentration_limit: Price::from_f64(0.01).unwrap(),
        regulatory_basis: "Test".to_string(),
    };

    compliance_engine
        .set_position_limit("ES_FUT".to_string(), limit)
        .await
        .unwrap();

    // Order that violates limit
    let order = OrderInfo {
        order_id: "test_order_violation".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(10.0).unwrap(), // Exceeds 1.0 limit
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Market),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    let result = compliance_engine.validate_order(&order, None).await.unwrap();
    assert!(!result.is_compliant);

    // Check if violations were broadcast
    // (May not receive due to timing, but subscription should work)
    let _ = violation_receiver.try_recv();
}

#[tokio::test]
async fn test_warning_broadcasting() {
    // Test that warnings are broadcast
    let config = create_test_compliance_config();
    let mut regulatory_config = RegulatoryReportingConfig::default();
    regulatory_config.mifid2_enabled = true; // Enable MiFID II for warnings

    let compliance_engine = Arc::new(ComplianceValidator::new(config, regulatory_config));
    let mut warning_receiver = compliance_engine.subscribe_to_warnings();

    let order = OrderInfo {
        order_id: "test_order_warning".to_string(),
        symbol: Symbol::from("ES_FUT"),
        instrument_id: "ES_FUT".to_string(),
        side: OrderSide::Buy,
        quantity: Quantity::from_f64(5.0).unwrap(),
        price: Price::from_f64(4500.0).unwrap(),
        order_type: Some(OrderType::Limit),
        portfolio_id: Some("test_portfolio".to_string()),
        strategy_id: Some("dqn_strategy".to_string()),
    };

    compliance_engine.validate_order(&order, None).await.unwrap();

    // Check if warnings were broadcast (may be empty)
    let _ = warning_receiver.try_recv();
}