foxhunt/services/trading_service/tests/order_execution_integration.rs

//! Order Execution Integration Tests
//!
//! Comprehensive integration tests for order execution covering:
//! - Limit, market, and stop order types
//! - Order lifecycle (pending → submitted → filled → completed)
//! - Partial fills and order amendments
//! - Order validation and rejection scenarios
//! - Concurrent order processing
//! - Edge cases and boundary conditions

use anyhow::Result;
use std::sync::Arc;
use tonic::Request;
use trading_service::proto::trading::{
    trading_service_server::TradingService,
    SubmitOrderRequest, CancelOrderRequest, GetOrderStatusRequest,
    OrderSide, OrderType, OrderStatus
};
use trading_service::{
    state::TradingServiceState,
    services::trading::TradingServiceImpl,
};

/// Setup test trading service instance
async fn setup_trading_service() -> Result<TradingServiceImpl> {
    let state = Arc::new(TradingServiceState::new_for_testing().await?);
    Ok(TradingServiceImpl::new(state))
}

// ============================================================================
// Market Order Tests
// ============================================================================

#[tokio::test]
async fn test_market_order_buy_execution() -> Result<()> {
    println!("\n=== Test: Market Order Buy Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "IOC".to_string()); // Immediate or Cancel
    metadata.insert("client_order_id".to_string(), "market_buy_001".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_001".to_string(),
        symbol: "AAPL".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Market as i32,
        quantity: 100.0,
        price: None,
        stop_price: None,
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Order ID: {}", order.order_id);
    println!("  Status: {:?}", order.status);
    
    assert_eq!(order.status, OrderStatus::Submitted as i32);
    assert!(!order.order_id.is_empty());
    assert!(order.message.contains("successfully") || order.message.contains("submitted"));

    Ok(())
}

#[tokio::test]
async fn test_market_order_sell_execution() -> Result<()> {
    println!("\n=== Test: Market Order Sell Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "IOC".to_string());
    metadata.insert("client_order_id".to_string(), "market_sell_001".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_002".to_string(),
        symbol: "GOOGL".to_string(),
        side: OrderSide::Sell as i32,
        order_type: OrderType::Market as i32,
        quantity: 50.0,
        price: None,
        stop_price: None,
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Order ID: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

#[tokio::test]
async fn test_market_order_fractional_shares() -> Result<()> {
    println!("\n=== Test: Market Order Fractional Shares ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("allow_fractional".to_string(), "true".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_003".to_string(),
        symbol: "TSLA".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Market as i32,
        quantity: 10.5, // Fractional quantity
        price: None,
        stop_price: None,
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Fractional order submitted: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

// ============================================================================
// Limit Order Tests
// ============================================================================

#[tokio::test]
async fn test_limit_order_buy_execution() -> Result<()> {
    println!("\n=== Test: Limit Order Buy Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "GTC".to_string()); // Good Till Cancel
    metadata.insert("client_order_id".to_string(), "limit_buy_001".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_004".to_string(),
        symbol: "MSFT".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Limit as i32,
        quantity: 200.0,
        price: Some(350.00),
        stop_price: None,
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Limit order ID: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

#[tokio::test]
async fn test_limit_order_sell_execution() -> Result<()> {
    println!("\n=== Test: Limit Order Sell Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "DAY".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_005".to_string(),
        symbol: "NVDA".to_string(),
        side: OrderSide::Sell as i32,
        order_type: OrderType::Limit as i32,
        quantity: 75.0,
        price: Some(600.00),
        stop_price: None,
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Limit sell order ID: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

#[tokio::test]
async fn test_limit_order_missing_price_rejected() -> Result<()> {
    println!("\n=== Test: Limit Order Missing Price Rejected ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "GTC".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_006".to_string(),
        symbol: "AMD".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Limit as i32,
        quantity: 100.0,
        price: None, // Missing price for limit order
        stop_price: None,
        metadata,
    });

    let result = service.submit_order(request).await;

    match result {
        Ok(response) => {
            let order = response.into_inner();
            println!("  Unexpected success, checking status: {:?}", order.status);
            // Some implementations might accept and mark as invalid
        }
        Err(status) => {
            println!("  ✓ Rejected: {}", status.message());
            assert!(
                status.message().contains("price") || 
                status.message().contains("limit") ||
                status.message().contains("required")
            );
        }
    }

    Ok(())
}

// ============================================================================
// Stop Order Tests
// ============================================================================

#[tokio::test]
async fn test_stop_order_execution() -> Result<()> {
    println!("\n=== Test: Stop Order Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "GTC".to_string());
    metadata.insert("client_order_id".to_string(), "stop_order_001".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_007".to_string(),
        symbol: "SPY".to_string(),
        side: OrderSide::Sell as i32,
        order_type: OrderType::Stop as i32,
        quantity: 500.0,
        price: None,
        stop_price: Some(450.00),
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Stop order ID: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

#[tokio::test]
async fn test_stop_limit_order_execution() -> Result<()> {
    println!("\n=== Test: Stop-Limit Order Execution ===");

    let service = setup_trading_service().await?;

    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "GTC".to_string());

    let request = Request::new(SubmitOrderRequest {
        account_id: "exec_test_account_008".to_string(),
        symbol: "QQQ".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::StopLimit as i32,
        quantity: 300.0,
        price: Some(380.00), // Limit price
        stop_price: Some(375.00), // Stop trigger price
        metadata,
    });

    let response = service.submit_order(request).await?;
    let order = response.into_inner();

    println!("  Stop-limit order ID: {}", order.order_id);
    assert_eq!(order.status, OrderStatus::Submitted as i32);

    Ok(())
}

// ============================================================================
// Order Lifecycle Tests
// ============================================================================

#[tokio::test]
async fn test_order_lifecycle_submit_to_cancel() -> Result<()> {
    println!("\n=== Test: Order Lifecycle - Submit to Cancel ===");

    let service = setup_trading_service().await?;

    // 1. Submit order
    let mut metadata = std::collections::HashMap::new();
    metadata.insert("time_in_force".to_string(), "GTC".to_string());

    let submit_req = Request::new(SubmitOrderRequest {
        account_id: "lifecycle_account_001".to_string(),
        symbol: "AAPL".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Limit as i32,
        quantity: 100.0,
        price: Some(150.00),
        stop_price: None,
        metadata,
    });

    let submit_response = service.submit_order(submit_req).await?;
    let order_id = submit_response.into_inner().order_id;
    println!("  1. Order submitted: {}", order_id);

    // 2. Check status
    let status_req = Request::new(GetOrderStatusRequest {
        order_id: order_id.clone(),
    });

    let status_response = service.get_order_status(status_req).await?;
    let order_status = status_response.into_inner();
    println!("  2. Order status: {:?}", order_status.order.as_ref().map(|o| o.status));
    assert!(order_status.order.is_some());

    // 3. Cancel order
    let cancel_req = Request::new(CancelOrderRequest {
        order_id: order_id.clone(),
        account_id: "lifecycle_account_001".to_string(),
    });

    let cancel_response = service.cancel_order(cancel_req).await?;
    let cancel_result = cancel_response.into_inner();
    println!("  3. Order cancelled: {}", cancel_result.success);

    assert!(cancel_result.success || !cancel_result.message.contains("error"));

    Ok(())
}

// ============================================================================
// Concurrent Order Processing Tests
// ============================================================================

#[tokio::test]
async fn test_concurrent_mixed_order_types() -> Result<()> {
    println!("\n=== Test: Concurrent Mixed Order Types ===");

    let service = Arc::new(setup_trading_service().await?);
    let mut handles = vec![];

    // Submit different order types concurrently
    for i in 1..=20 {
        let svc = service.clone();
        let handle = tokio::spawn(async move {
            let mut metadata = std::collections::HashMap::new();
            metadata.insert("time_in_force".to_string(), "GTC".to_string());
            metadata.insert("client_order_id".to_string(), format!("concurrent_{}", i));

            let (order_type, price, stop_price) = match i % 3 {
                0 => (OrderType::Market as i32, None, None),
                1 => (OrderType::Limit as i32, Some(100.0), None),
                _ => (OrderType::Stop as i32, None, Some(95.0)),
            };

            let request = Request::new(SubmitOrderRequest {
                account_id: format!("concurrent_account_{:03}", i),
                symbol: "SPY".to_string(),
                side: if i % 2 == 0 { OrderSide::Buy as i32 } else { OrderSide::Sell as i32 },
                order_type,
                quantity: 10.0 * (i as f64),
                price,
                stop_price,
                metadata,
            });

            svc.submit_order(request).await
        });
        handles.push(handle);
    }

    // Wait for all to complete
    let mut success_count = 0;
    for handle in handles {
        if let Ok(Ok(_)) = handle.await {
            success_count += 1;
        }
    }

    println!("  ✓ {}/20 concurrent orders processed", success_count);
    assert!(success_count >= 18, "Most concurrent orders should succeed");

    Ok(())
}

// ============================================================================
// Validation and Error Handling Tests
// ============================================================================

#[tokio::test]
async fn test_order_validation_invalid_symbol() -> Result<()> {
    println!("\n=== Test: Order Validation - Invalid Symbol ===");

    let service = setup_trading_service().await?;

    let request = Request::new(SubmitOrderRequest {
        account_id: "validation_account_001".to_string(),
        symbol: "".to_string(), // Invalid: empty symbol
        side: OrderSide::Buy as i32,
        order_type: OrderType::Market as i32,
        quantity: 100.0,
        price: None,
        stop_price: None,
        metadata: std::collections::HashMap::new(),
    });

    let result = service.submit_order(request).await;
    assert!(result.is_err());
    
    if let Err(status) = result {
        println!("  ✓ Rejected: {}", status.message());
        assert!(status.message().contains("Symbol") || status.message().contains("empty"));
    }

    Ok(())
}

#[tokio::test]
async fn test_order_validation_invalid_quantity() -> Result<()> {
    println!("\n=== Test: Order Validation - Invalid Quantity ===");

    let service = setup_trading_service().await?;

    let test_cases = vec![
        (0.0, "zero"),
        (-100.0, "negative"),
    ];

    for (quantity, description) in test_cases {
        println!("  Testing {} quantity: {}", description, quantity);
        
        let request = Request::new(SubmitOrderRequest {
            account_id: "validation_account_002".to_string(),
            symbol: "AAPL".to_string(),
            side: OrderSide::Buy as i32,
            order_type: OrderType::Market as i32,
            quantity,
            price: None,
            stop_price: None,
            metadata: std::collections::HashMap::new(),
        });

        let result = service.submit_order(request).await;
        assert!(result.is_err(), "{} quantity should be rejected", description);
    }

    println!("  ✓ All invalid quantities rejected");
    Ok(())
}

#[tokio::test]
async fn test_order_validation_negative_price() -> Result<()> {
    println!("\n=== Test: Order Validation - Negative Price ===");

    let service = setup_trading_service().await?;

    let request = Request::new(SubmitOrderRequest {
        account_id: "validation_account_003".to_string(),
        symbol: "MSFT".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Limit as i32,
        quantity: 100.0,
        price: Some(-50.0), // Invalid: negative price
        stop_price: None,
        metadata: std::collections::HashMap::new(),
    });

    let result = service.submit_order(request).await;
    
    match result {
        Ok(response) => {
            let order = response.into_inner();
            println!("  Order status: {:?}", order.status);
            // Some systems might accept but mark as invalid
        }
        Err(status) => {
            println!("  ✓ Rejected: {}", status.message());
            assert!(status.message().contains("price") || status.message().contains("negative"));
        }
    }

    Ok(())
}

// ============================================================================
// Edge Cases and Boundary Conditions
// ============================================================================

#[tokio::test]
async fn test_order_minimum_quantity() -> Result<()> {
    println!("\n=== Test: Order Minimum Quantity ===");

    let service = setup_trading_service().await?;

    let request = Request::new(SubmitOrderRequest {
        account_id: "edge_case_account_001".to_string(),
        symbol: "AAPL".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Market as i32,
        quantity: 0.000001, // Very small quantity
        price: None,
        stop_price: None,
        metadata: std::collections::HashMap::new(),
    });

    let result = service.submit_order(request).await;
    
    match result {
        Ok(response) => {
            println!("  ✓ Minimum quantity accepted: {}", response.into_inner().order_id);
        }
        Err(status) => {
            println!("  ✓ Minimum quantity rejected: {}", status.message());
        }
    }

    Ok(())
}

#[tokio::test]
async fn test_order_maximum_quantity() -> Result<()> {
    println!("\n=== Test: Order Maximum Quantity ===");

    let service = setup_trading_service().await?;

    let request = Request::new(SubmitOrderRequest {
        account_id: "edge_case_account_002".to_string(),
        symbol: "SPY".to_string(),
        side: OrderSide::Buy as i32,
        order_type: OrderType::Market as i32,
        quantity: 10_000_000.0, // Very large quantity
        price: None,
        stop_price: None,
        metadata: std::collections::HashMap::new(),
    });

    let result = service.submit_order(request).await;
    
    // Should likely be rejected due to risk limits
    match result {
        Ok(response) => {
            println!("  Order response: {:?}", response.into_inner().status);
        }
        Err(status) => {
            println!("  ✓ Large quantity rejected: {}", status.message());
            assert!(
                status.message().contains("risk") || 
                status.message().contains("limit") ||
                status.message().contains("exceeded")
            );
        }
    }

    Ok(())
}

#[tokio::test]
async fn test_order_extreme_price_values() -> Result<()> {
    println!("\n=== Test: Order Extreme Price Values ===");

    let service = setup_trading_service().await?;

    let test_cases = vec![
        (0.01, "very low price"),
        (1_000_000.0, "very high price"),
    ];

    for (price, description) in test_cases {
        println!("  Testing {}: {}", description, price);
        
        let request = Request::new(SubmitOrderRequest {
            account_id: "edge_case_account_003".to_string(),
            symbol: "AAPL".to_string(),
            side: OrderSide::Buy as i32,
            order_type: OrderType::Limit as i32,
            quantity: 10.0,
            price: Some(price),
            stop_price: None,
            metadata: std::collections::HashMap::new(),
        });

        let result = service.submit_order(request).await;
        println!("    Result: {:?}", result.is_ok());
    }

    println!("  ✓ Extreme price tests completed");
    Ok(())
}