Function test_worker_pool_concurrent_execution

src/server/worker_pool.rs:229–269

()

    #[tokio::test]
    async fn test_worker_pool_concurrent_execution() {
        let pool = WorkerPool::new(2);

        // Submit 4 jobs to a pool with 2 workers
        let mut submitted = 0;
        let mut attempts = 0;
        while submitted < 4 && attempts < 100 {
            let job = TestJob {
                id: format!("job-{}", submitted),
                duration_ms: 50,
                should_fail: false,
            };
            if pool.try_submit(job).await.unwrap().is_some() {
                submitted += 1;
            } else {
                // Wait a bit for a worker to become available
                sleep(Duration::from_millis(10)).await;
            }
            attempts += 1;
        }
        assert_eq!(submitted, 4, "Should have submitted all 4 jobs");

        // Wait for all jobs to complete
        sleep(Duration::from_millis(150)).await;

        // Poll for completed jobs
        let mut all_completed = Vec::new();
        loop {
            let completed = pool.poll_completed().await;
            if completed.is_empty() {
                break;
            }
            all_completed.extend(completed);
        }

        assert_eq!(all_completed.len(), 4, "All 4 jobs should complete");
        for result in &all_completed {
            assert!(result.is_ok());
            assert!(result.as_ref().unwrap().success);
        }
    }

    #[tokio::test]
    async fn test_worker_pool_concurrency_limit() {