CodeSource
/
Readur
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(ci): resolve issues in stress tests

This commit is contained in:
perf3ct 2025-09-18 21:07:01 +00:00
parent e43e7af372
commit 32262db3dd
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/webdav_stress_tests.rs
View File

@ -169,7 +169,7 @@ impl WebDAVResourceManager {
/// Acquire resources for a WebDAV operation /// Acquire resources for a WebDAV operation
pub async fn acquire_operation_permit(&self) -> anyhow::Result<OperationPermit> { pub async fn acquire_operation_permit(&self) -> anyhow::Result<OperationPermit> {
// Wait for semaphore permit // Wait for semaphore permit
let permit = self.operation_semaphore.acquire().await let permit = self.operation_semaphore.clone().acquire_owned().await
.map_err(|e| anyhow::anyhow!("Failed to acquire operation permit: {}", e))?; .map_err(|e| anyhow::anyhow!("Failed to acquire operation permit: {}", e))?;
// Rate limiting // Rate limiting
@ -420,7 +420,7 @@ impl LoopDetectionMonitor {
// Store the task handle // Store the task handle
if let Ok(mut handle) = cleanup_interval.try_lock() { if let Ok(mut handle) = cleanup_interval.try_lock() {
*handle = Some(task); *handle = Some(task);
} };
} }
/// Record a directory access for loop detection with circuit breaker protection /// Record a directory access for loop detection with circuit breaker protection
@ -523,10 +523,9 @@ impl LoopDetectionMonitor {
self.monitoring_active.store(false, Ordering::Relaxed); self.monitoring_active.store(false, Ordering::Relaxed);
// Stop cleanup task // Stop cleanup task
if let Ok(mut handle) = self.cleanup_interval.lock().await { let mut handle = self.cleanup_interval.lock().await;
if let Some(task) = handle.take() { if let Some(task) = handle.take() {
task.abort(); task.abort();
}
} }
// Clear all data to free memory // Clear all data to free memory
@ -604,7 +603,7 @@ fn create_stress_test_webdav_service(config: &StressTestConfig) -> Result<WebDAV
file_extensions: vec![], file_extensions: vec![],
}; };
Ok(WebDAVService::new(webdav_config)) WebDAVService::new(webdav_config)
} }
/// Get stress test configuration from environment variables /// Get stress test configuration from environment variables
@ -724,6 +723,7 @@ async fn perform_loop_detection_test(
// Perform concurrent WebDAV operations // Perform concurrent WebDAV operations
for i in 0..operation_count { for i in 0..operation_count {
let path = test_paths[i % test_paths.len()].to_string(); let path = test_paths[i % test_paths.len()].to_string();
let path_for_check = path.clone();
let service = webdav_service.clone(); let service = webdav_service.clone();
let monitor = loop_monitor.clone(); let monitor = loop_monitor.clone();
@ -740,7 +740,8 @@ async fn perform_loop_detection_test(
}; };
// Acquire directory lock to prevent race conditions // Acquire directory lock to prevent race conditions
let _dir_lock = resource_mgr.acquire_directory_lock(&path).await.lock().await; let dir_lock_arc = resource_mgr.acquire_directory_lock(&path).await;
let _dir_lock = dir_lock_arc.lock().await;
// Record directory access for loop detection // Record directory access for loop detection
monitor.record_directory_access(&path).await; monitor.record_directory_access(&path).await;
@ -778,8 +779,8 @@ async fn perform_loop_detection_test(
// Check for suspected loops periodically // Check for suspected loops periodically
if i % 10 == 0 { if i % 10 == 0 {
if loop_monitor.is_suspected_loop(&path).await { if loop_monitor.is_suspected_loop(&path_for_check).await {
warn!("Suspected loop detected for path: {} - continuing test to gather data", path); warn!("Suspected loop detected for path: {} - continuing test to gather data", path_for_check);
} }
} }
@ -1019,7 +1020,8 @@ async fn test_wide_directory_scanning(
}; };
// Acquire directory lock // Acquire directory lock
let _dir_lock = resource_mgr.acquire_directory_lock(&dir_path).await.lock().await; let dir_lock_arc = resource_mgr.acquire_directory_lock(&dir_path).await;
let _dir_lock = dir_lock_arc.lock().await;
let start_time = Instant::now(); let start_time = Instant::now();
match timeout( match timeout(
@ -1140,7 +1142,8 @@ async fn test_concurrent_webdav_access() -> Result<()> {
}; };
// Acquire directory lock to prevent race conditions on same path // Acquire directory lock to prevent race conditions on same path
let _dir_lock = resource_mgr.acquire_directory_lock(path).await.lock().await; let dir_lock_arc = resource_mgr.acquire_directory_lock(path).await;
let _dir_lock = dir_lock_arc.lock().await;
match timeout( match timeout(
Duration::from_secs(timeout_secs), Duration::from_secs(timeout_secs),