Implement comprehensive Runpod deployment with S3 volume mount architecture for FP32 ML model training on Tesla V100 GPUs. ## Infrastructure Components ### Deployment Scripts (scripts/) - runpod_deploy.sh: Master deployment orchestrator (8-step workflow) - runpod_upload.sh: S3 upload for binaries and test data - upload_env_to_runpod.sh: Secure .env credentials upload - runpod_deploy_test.sh: Prerequisites validation ### Docker Configuration - Dockerfile.runpod: Multi-stage CUDA 12.1 runtime (~2GB, no binaries) - entrypoint.sh: Volume verification and training execution - Architecture: Volume mount (NO S3 downloads in pods) ### S3 Configuration - Bucket: se3zdnb5o4 (Iceland region: eur-is-1) - Endpoint: https://s3api-eur-is-1.runpod.io - Structure: binaries/, test_data/, models/, .env ### OpenTofu Infrastructure (terraform/runpod/) - main.tf: Pod and volume resources - variables.tf: Configuration variables - outputs.tf: Pod connection info - Security: NO credentials in state (uses volume .env) ## Deployment Assets Uploaded ### Training Binaries (77MB) - train_tft_parquet (23M) - TFT-225 features - train_mamba2_parquet (22M) - MAMBA-2 state space - train_dqn (22M) - Deep Q-Network - train_ppo (13M) - Proximal Policy Optimization ### Test Data (13.8 MB) - 9 Parquet files: ES.FUT, NQ.FUT, 6E.FUT, ZN.FUT (180-day datasets) ### Credentials - .env file (1.5 KB, private access, chmod 600) ## Documentation ### Deployment Guides - RUNPOD_DEPLOYMENT_READY_SUMMARY.md: Complete deployment status - RUNPOD_VOLUME_DEPLOYMENT_GUIDE.md: Step-by-step guide (42KB) - RUNPOD_DEPLOYMENT_QUICK_START.md: Quick reference - RUNPOD_UPLOAD_GUIDE.md: S3 upload instructions - RUNPOD_VOLUME_CONFIGURATION_COMPLETE.md: S3 setup report - RUNPOD_S3_PARQUET_UPLOAD_REPORT.md: Data upload verification ### Architecture Documentation - RUNPOD_VOLUME_MOUNT_ARCHITECTURE.md: Volume mount design - RUNPOD_S3_ARCHITECTURE_DIAGRAM.txt: S3 API vs filesystem access - DOCKERFILE_RUNPOD_FINAL_SUMMARY.md: Docker image specification ### Decision Documentation - RUNPOD_DEPLOYMENT_CHECKLIST.md: Go/no-go decision matrix (27KB) - RUNPOD_DEPLOYMENT_DECISION_TREE.md: Decision workflow - FP32_RUNPOD_DEPLOYMENT_READY.md: FP32 deployment readiness ## QAT Enhancements ### Core QAT Infrastructure - ml/src/memory_optimization/qat.rs: Enhanced QAT observer (+226 lines) - ml/src/memory_optimization/auto_batch_size.rs: OOM recovery (+84 lines) - ml/src/tft/qat_tft.rs: QAT TFT wrapper (+154 lines) - ml/src/trainers/tft.rs: QAT training integration (+433 lines) - ml/src/qat_metrics_exporter.rs: NEW - QAT metrics export ### QAT Testing - ml/tests/qat_integration_tests.rs: NEW - Integration test suite - ml/tests/qat_gradient_clipping_test.rs: NEW - Gradient clipping tests - ml/tests/qat_device_consistency_test.rs: Device mismatch tests (+205 lines) - ml/tests/qat_accuracy_validation_test.rs: Accuracy validation - ml/tests/qat_tft_integration_test.rs: TFT QAT integration ### QAT Documentation - ml/docs/QAT_GUIDE.md: Comprehensive QAT guide (+616 lines) - ml/docs/QAT_GRADIENT_CHECKPOINTING_WORKAROUND.md: NEW - Workaround guide - QAT_BLOCKERS_ROOT_CAUSE_ANALYSIS.md: P0 blocker analysis (44KB) - QAT_ACCURACY_VALIDATION_REPORT.md: Accuracy comparison - QAT_GRADIENT_CLIPPING_VALIDATION_REPORT.md: Clipping validation ### QAT Monitoring - config/grafana/dashboards/qat-training-metrics.json: NEW - Grafana dashboard ## AWS CLI Configuration ### Credentials Setup - ~/.aws/credentials: Runpod profile configured - Access Key: user_2xxA3XcIFj16yfL3aBon9niiSpr - Secret Key: (from RUNPOD_S3_SECRET) - ~/.aws/config: Iceland region (eur-is-1) ## Production Readiness ### FP32 Models: ✅ READY FOR DEPLOYMENT - DQN: 15-20s training, ~6MB GPU memory - PPO: 7-10s training, ~145MB GPU memory - MAMBA-2: 2-3 min training, ~164MB GPU memory - TFT-225: 3-5 min training, ~500MB GPU memory - Total GPU Budget: 815MB (fits on 4GB+ Tesla V100) ### QAT Models: 🔴 BLOCKED - 24 tests implemented but DO NOT COMPILE (11 errors) - 3 P0 blockers: device mismatch, gradient checkpointing, OOM recovery - Timeline: 1-2 weeks to fix (13h P0 fixes + validation) ### Wave D Features: ✅ OPERATIONAL - 225 features fully integrated - Feature extraction: 5.10μs/bar (196x faster than target) - Wave D backtest: Sharpe 2.00, Win Rate 60%, Drawdown 15% - Database migration 045: Applied cleanly, zero conflicts ## Cost Analysis ### One-Time Setup - Network Volume: $4/month (50GB SSD) - Upload costs: FREE (S3 API included) ### Per Training Run (TFT-225) - GPU: Tesla V100-PCIE-16GB @ $0.29/hr - Training Time: ~4 hours - Cost per run: $1.16 ### Monthly (20 Training Runs) - Storage: $4.00/month - Training: $23.20/month (20 runs × $1.16) - Total: $27.20/month ## Security ### Credentials Management - ✅ NO credentials in Docker image - ✅ NO credentials in Terraform state - ✅ .env gitignored and not committed - ✅ .env file private on S3 (HTTP 401 on public access) - ✅ Docker Hub repository PRIVATE (jgrusewski/foxhunt) ### Access Control - S3 API: Local client uploads only - Volume mount: Pod filesystem access only - Authentication: AWS CLI with Runpod profile required ## Next Steps 1. ✅ COMPLETE: Build Docker image 2. ⏳ PENDING: Push to Docker Hub 3. ⏳ PENDING: Deploy pod via Runpod console 4. ⏳ PENDING: Validate training on Tesla V100 ## Performance Targets - Build time: 5-10 min - Upload time: ~20 sec (90MB total) - Pod startup: ~30 sec - Training time: 3-5 min (TFT-225) - Total deployment: ~40 min from start to first training run ## Test Status - FP32 tests: 597/608 passing (98.2%) - QAT tests: 0/24 passing (compilation errors) - Overall: 2,062/2,086 passing (98.8% excluding QAT) 🤖 Generated with Claude Code (https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
364 lines
11 KiB
Rust
364 lines
11 KiB
Rust
//! Error Conversion and Retry Logic Tests for Storage Crate
|
|
//!
|
|
//! This module provides comprehensive tests for:
|
|
//! - StorageError to CommonError conversion (all 15 variants)
|
|
//! - StorageError::retry_delay_ms (retryable vs non-retryable errors)
|
|
//! - std::io::Error to StorageError conversion (all ErrorKind mappings)
|
|
|
|
use common::error::{CommonError, ErrorCategory};
|
|
use std::io::ErrorKind;
|
|
use storage::error::StorageError;
|
|
|
|
// =============================================================================
|
|
// STORAGE ERROR TO COMMON ERROR CONVERSION
|
|
// =============================================================================
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_io() {
|
|
let storage_err = StorageError::IoError {
|
|
message: "disk full".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_network() {
|
|
let storage_err = StorageError::NetworkError {
|
|
message: "host unreachable".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::Network);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_auth() {
|
|
let storage_err = StorageError::AuthError {
|
|
message: "bad credentials".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::Security);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_config() {
|
|
let storage_err = StorageError::ConfigError {
|
|
message: "invalid path".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::Configuration);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_not_found() {
|
|
let storage_err = StorageError::NotFound {
|
|
path: "/data/model.bin".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_operation_failed() {
|
|
use std::sync::Arc;
|
|
let storage_err = StorageError::OperationFailed {
|
|
operation: "write".to_string(),
|
|
path: "/data/checkpoint.bin".to_string(),
|
|
source: Arc::new(CommonError::Network("network failed".to_string())),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_permission_denied() {
|
|
let storage_err = StorageError::PermissionDenied {
|
|
path: "/etc/secrets".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::Security);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_timeout() {
|
|
let storage_err = StorageError::Timeout { timeout_ms: 5000 };
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_rate_limited() {
|
|
let storage_err = StorageError::RateLimited {
|
|
retry_after_ms: 1000,
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_quota_exceeded() {
|
|
let storage_err = StorageError::QuotaExceeded {
|
|
used: 1000,
|
|
limit: 500,
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_serialization() {
|
|
let storage_err = StorageError::SerializationError {
|
|
message: "invalid JSON".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_compression() {
|
|
let storage_err = StorageError::CompressionError {
|
|
message: "gzip failed".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_integrity_error() {
|
|
let storage_err = StorageError::IntegrityError {
|
|
path: "/data/model.bin".to_string(),
|
|
expected: "abc123".to_string(),
|
|
actual: "def456".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_to_common_error_generic() {
|
|
let storage_err = StorageError::Generic {
|
|
message: "unknown error".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[cfg(feature = "s3")]
|
|
#[test]
|
|
fn test_storage_error_to_common_error_s3() {
|
|
let storage_err = StorageError::S3Error {
|
|
message: "S3 operation failed".to_string(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::Network);
|
|
}
|
|
|
|
// =============================================================================
|
|
// RETRY DELAY TESTS
|
|
// =============================================================================
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_network_error() {
|
|
let err = StorageError::NetworkError {
|
|
message: "connection failed".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), Some(100));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_timeout() {
|
|
let err = StorageError::Timeout { timeout_ms: 500 };
|
|
assert_eq!(err.retry_delay_ms(), Some(200));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_rate_limited() {
|
|
let err = StorageError::RateLimited {
|
|
retry_after_ms: 1500,
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), Some(1500));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_generic() {
|
|
let err = StorageError::Generic {
|
|
message: "generic error".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), Some(100));
|
|
}
|
|
|
|
#[cfg(feature = "s3")]
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_s3() {
|
|
let err = StorageError::S3Error {
|
|
message: "S3 error".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), Some(100));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_not_found() {
|
|
let err = StorageError::NotFound {
|
|
path: "/data/missing.bin".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_quota_exceeded() {
|
|
let err = StorageError::QuotaExceeded {
|
|
used: 1000,
|
|
limit: 500,
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_permission_denied() {
|
|
let err = StorageError::PermissionDenied {
|
|
path: "/etc/forbidden".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_serialization_error() {
|
|
let err = StorageError::SerializationError {
|
|
message: "bad JSON".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_retry_delay_ms_integrity_error() {
|
|
let err = StorageError::IntegrityError {
|
|
path: "/data/file.bin".to_string(),
|
|
expected: "abc".to_string(),
|
|
actual: "def".to_string(),
|
|
};
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|
|
|
|
// =============================================================================
|
|
// IO ERROR TO STORAGE ERROR CONVERSION
|
|
// =============================================================================
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_not_found() {
|
|
let io_err = std::io::Error::new(ErrorKind::NotFound, "file not found");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(storage_err, StorageError::NotFound { .. }));
|
|
if let StorageError::NotFound { path } = storage_err {
|
|
assert_eq!(path, "unknown"); // Default value from conversion
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_permission_denied() {
|
|
let io_err = std::io::Error::new(ErrorKind::PermissionDenied, "access denied");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(storage_err, StorageError::PermissionDenied { .. }));
|
|
if let StorageError::PermissionDenied { path } = storage_err {
|
|
assert_eq!(path, "unknown"); // Default value from conversion
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_already_exists() {
|
|
let io_err = std::io::Error::new(ErrorKind::AlreadyExists, "file exists");
|
|
let storage_err: StorageError = io_err.into();
|
|
// AlreadyExists maps to IoError in current implementation
|
|
assert!(matches!(storage_err, StorageError::IoError { .. }));
|
|
if let StorageError::IoError { message } = storage_err {
|
|
assert!(message.contains("file exists"));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_timed_out() {
|
|
let io_err = std::io::Error::new(ErrorKind::TimedOut, "operation timed out");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(
|
|
storage_err,
|
|
StorageError::Timeout { timeout_ms: 5000 }
|
|
));
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_broken_pipe() {
|
|
let io_err = std::io::Error::new(ErrorKind::BrokenPipe, "pipe broken");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(storage_err, StorageError::IoError { .. }));
|
|
if let StorageError::IoError { message } = storage_err {
|
|
assert!(message.contains("pipe broken"));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_connection_reset() {
|
|
let io_err = std::io::Error::new(ErrorKind::ConnectionReset, "connection reset");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(storage_err, StorageError::IoError { .. }));
|
|
if let StorageError::IoError { message } = storage_err {
|
|
assert!(message.contains("connection reset"));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_io_error_to_storage_error_other() {
|
|
let io_err = std::io::Error::other("unknown error");
|
|
let storage_err: StorageError = io_err.into();
|
|
assert!(matches!(storage_err, StorageError::IoError { .. }));
|
|
if let StorageError::IoError { message } = storage_err {
|
|
assert!(message.contains("unknown error"));
|
|
}
|
|
}
|
|
|
|
// =============================================================================
|
|
// EDGE CASES AND BOUNDARY CONDITIONS
|
|
// =============================================================================
|
|
|
|
#[test]
|
|
fn test_storage_error_empty_message() {
|
|
let storage_err = StorageError::Generic {
|
|
message: String::new(),
|
|
};
|
|
let common_err: CommonError = storage_err.into();
|
|
assert_eq!(common_err.category(), ErrorCategory::System);
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_long_path() {
|
|
let long_path = "a".repeat(1000);
|
|
let storage_err = StorageError::NotFound {
|
|
path: long_path.clone(),
|
|
};
|
|
assert!(matches!(storage_err, StorageError::NotFound { .. }));
|
|
if let StorageError::NotFound { path } = storage_err {
|
|
assert_eq!(path.len(), 1000);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_rate_limited_zero_retry() {
|
|
let err = StorageError::RateLimited { retry_after_ms: 0 };
|
|
assert_eq!(err.retry_delay_ms(), Some(0));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_timeout_zero() {
|
|
let err = StorageError::Timeout { timeout_ms: 0 };
|
|
assert_eq!(err.retry_delay_ms(), Some(200));
|
|
}
|
|
|
|
#[test]
|
|
fn test_storage_error_integrity_error_same_values() {
|
|
let err = StorageError::IntegrityError {
|
|
path: "/data/file.bin".to_string(),
|
|
expected: "abc123".to_string(),
|
|
actual: "abc123".to_string(),
|
|
};
|
|
// Even with matching values, it's still an integrity error (logic error in caller)
|
|
assert_eq!(err.retry_delay_ms(), None);
|
|
}
|