foxhunt/tests/test_common/mod.rs

//! Common test utilities for Foxhunt HFT System
//!
//! This module provides shared testing infrastructure to eliminate duplication
//! across the 80+ test files in the project.
//!
//! # Usage
//! ```rust
//! use common::types::*;
//! use common::types::test_config::*;
//! use common::types::mock_data::*;
//! ```

pub mod database_helper;

// Test Configuration Module
pub mod test_config {
    use std::time::Duration;

    /// Unified test configuration for all test types
    #[derive(Debug, Clone)]
    pub struct UnifiedTestConfig {
        pub environment_name: String,
        pub docker_compose_file: Option<String>,
        pub cleanup_on_exit: bool,
        pub persist_data: bool,
        pub log_level: String,
        pub test_database_url: String,
        pub test_redis_url: String,
        pub test_influxdb_url: String,
        pub parallel_tests: bool,
        pub timeout_seconds: u64,
        pub max_retries: u32,
    }

    impl Default for UnifiedTestConfig {
        fn default() -> Self {
            Self {
                environment_name: "test".to_string(),
                docker_compose_file: Some("docker-compose.test.yml".to_string()),
                cleanup_on_exit: true,
                persist_data: false,
                log_level: "debug".to_string(),
                test_database_url: std::env::var("TEST_DATABASE_URL").unwrap_or_else(|_| {
                    std::env::var("FOXHUNT_TEST_POSTGRES_URL").unwrap_or_else(|_| {
                        let db_host = std::env::var("DATABASE_HOST")
                            .or_else(|_| std::env::var("POSTGRES_HOST"))
                            .unwrap_or_else(|_| "localhost".to_string());
                        format!("postgresql://{}:5432/hft_testing", db_host)
                    })
                }),
                test_redis_url: std::env::var("TEST_REDIS_URL").unwrap_or_else(|_| {
                    std::env::var("FOXHUNT_TEST_REDIS_URL").unwrap_or_else(|_| {
                        let redis_host =
                            std::env::var("REDIS_HOST").unwrap_or_else(|_| "localhost".to_string());
                        format!(
                            "redis://:{}@{}:6379/0",
                            std::env::var("REDIS_TEST_PASSWORD")
                                .unwrap_or_else(|_| "test_password".to_string()),
                            redis_host
                        )
                    })
                }),
                test_influxdb_url: std::env::var("TEST_INFLUXDB_URL").unwrap_or_else(|_| {
                    let influx_host =
                        std::env::var("INFLUXDB_HOST").unwrap_or_else(|_| "localhost".to_string());
                    format!("http://{}:8086", influx_host)
                }),
                parallel_tests: true,
                timeout_seconds: 30,
                max_retries: 3,
            }
        }
    }
}

// Mock Data Generation Module
pub mod mock_data {
    use uuid::Uuid;

    /// Generate mock order data for testing
    pub fn create_mock_order() -> MockOrder {
        MockOrder {
            id: Uuid::new_v4().to_string(),
            symbol: "BTCUSD".to_string(),
            side: "Buy".to_string(),
            quantity: 1.0,
            price: 50000.0,
            status: "Pending".to_string(),
        }
    }

    /// Generate mock market data
    pub fn create_mock_market_tick(symbol: &str) -> MockMarketTick {
        MockMarketTick {
            symbol: symbol.to_string(),
            price: 50000.0,
            volume: 100.0,
            timestamp: chrono::Utc::now().timestamp_millis(),
        }
    }

    /// Mock order structure for tests
    #[derive(Debug, Clone)]
    pub struct MockOrder {
        pub id: String,
        pub symbol: String,
        pub side: String,
        pub quantity: f64,
        pub price: f64,
        pub status: String,
    }

    /// Mock market tick for tests
    #[derive(Debug, Clone)]
    pub struct MockMarketTick {
        pub symbol: String,
        pub price: f64,
        pub volume: f64,
        pub timestamp: i64,
    }
}

// Test Utilities Module
pub mod test_utils {
    use std::time::Duration;
    use tokio::time::timeout;

    /// Async test helper with timeout
    pub async fn run_with_timeout<F, T>(future: F, timeout_secs: u64) -> Result<T, &'static str>
    where
        F: std::future::Future<Output = T>,
    {
        timeout(Duration::from_secs(timeout_secs), future)
            .await
            .map_err(|_| "Test timed out")
    }

    /// Setup tracing for tests
    pub fn setup_test_tracing() {
        use tracing_subscriber::{EnvFilter, FmtSubscriber};

        let _ = tracing_subscriber::fmt()
            .with_test_writer()
            .with_env_filter(EnvFilter::from_default_env())
            .try_init();
    }

    /// Common test assertions
    pub mod assertions {
        use std::time::Duration;

        /// Assert that a value is within a percentage tolerance
        pub fn assert_within_percent(actual: f64, expected: f64, percent: f64) {
            let tolerance = expected * (percent / 100.0);
            let diff = (actual - expected).abs();
            assert!(
                diff <= tolerance,
                "Value {} is not within {}% of expected {}, difference: {}",
                actual,
                percent,
                expected,
                diff
            );
        }

        /// Assert that latency is within HFT requirements
        pub fn assert_hft_latency(duration: Duration, max_microseconds: u64) {
            let micros = duration.as_micros() as u64;
            assert!(
                micros <= max_microseconds,
                "Latency {}μs exceeds HFT requirement of {}μs",
                micros,
                max_microseconds
            );
        }
    }
}

// Async Test Patterns Module
pub mod async_patterns {
    use tokio::sync::broadcast;

    /// Proper broadcast receiver pattern for tests
    pub struct TestBroadcastReceiver<T> {
        receiver: broadcast::Receiver<T>,
    }

    impl<T> TestBroadcastReceiver<T>
    where
        T: Clone + Send + 'static,
    {
        pub fn new(receiver: broadcast::Receiver<T>) -> Self {
            Self { receiver }
        }

        pub async fn wait_for_shutdown(mut self) -> Result<(), broadcast::error::RecvError> {
            loop {
                tokio::select! {
                    msg = self.receiver.recv() => {
                        match msg {
                            Ok(_) => return Ok(()),
                            Err(e) => return Err(e),
                        }
                    }
                }
            }
        }
    }
}

// Re-export commonly used items for convenience
pub use database_helper::{
    benchmark_database_operations, cleanup_all_test_data, create_test_execution, create_test_order,
    create_test_position, create_test_user, get_test_database_pool,
    get_test_database_pool_with_config, setup_test_database, teardown_test_database,
    DatabaseBenchmarkResult, DatabaseTestConfig, DatabaseTestPool,
};

pub use async_patterns::TestBroadcastReceiver;
pub use mock_data::{create_mock_market_tick, create_mock_order, MockMarketTick, MockOrder};
pub use test_config::UnifiedTestConfig;
pub use test_utils::{assertions, run_with_timeout, setup_test_tracing};