From 7c667969c8ad12bcb01680a972cce22c5deeba2b Mon Sep 17 00:00:00 2001
From: Noah <noahskusaba@gmail.com>
Date: Sat, 22 Aug 2026 12:56:03 -0400
Subject: [PATCH] init

---
 ballista/scheduler/src/test_utils.rs | 79 +---------------------------
 1 file changed, 2 insertions(+), 77 deletions(-)

diff --git a/ballista/scheduler/src/test_utils.rs b/ballista/scheduler/src/test_utils.rs
index 25604673b..82b8f4fcb 100644
--- a/ballista/scheduler/src/test_utils.rs
+++ b/ballista/scheduler/src/test_utils.rs
@@ -601,76 +601,6 @@ impl SchedulerTest {
             .await
     }
 
-    /// Waits for job completion with a timeout in milliseconds.
-    pub async fn await_completion_timeout(
-        &self,
-        job_id: &JobId,
-        timeout_ms: u64,
-    ) -> Result<JobStatus> {
-        let mut time = 0;
-        let final_status: Result<JobStatus> = loop {
-            let status = self
-                .scheduler
-                .state
-                .task_manager
-                .get_job_status(job_id)
-                .await?;
-
-            if let Some(JobStatus {
-                status: Some(inner),
-                ..
-            }) = status.as_ref()
-            {
-                match inner {
-                    Status::Failed(_) | Status::Successful(_) => {
-                        break Ok(status.unwrap());
-                    }
-                    _ => {
-                        if time >= timeout_ms {
-                            break Ok(status.unwrap());
-                        } else {
-                            continue;
-                        }
-                    }
-                }
-            }
-
-            tokio::time::sleep(Duration::from_millis(100)).await;
-            time += 100;
-        };
-
-        final_status
-    }
-
-    /// Waits for job completion indefinitely.
-    pub async fn await_completion(&self, job_id: &JobId) -> Result<JobStatus> {
-        let final_status: Result<JobStatus> = loop {
-            let status = self
-                .scheduler
-                .state
-                .task_manager
-                .get_job_status(job_id)
-                .await?;
-
-            if let Some(JobStatus {
-                status: Some(inner),
-                ..
-            }) = status.as_ref()
-            {
-                match inner {
-                    Status::Failed(_) | Status::Successful(_) => {
-                        break Ok(status.unwrap());
-                    }
-                    _ => continue,
-                }
-            }
-
-            tokio::time::sleep(Duration::from_millis(100)).await
-        };
-
-        final_status
-    }
-
     /// Returns job status and job_id
     pub async fn run(
         &mut self,
@@ -719,16 +649,11 @@ impl SchedulerTest {
                 .await?;
 
             if let Some(JobStatus {
-                status: Some(inner),
+                status: Some(Status::Failed(_) | Status::Successful(_)),
                 ..
             }) = status.as_ref()
             {
-                match inner {
-                    Status::Failed(_) | Status::Successful(_) => {
-                        break Ok(status.unwrap());
-                    }
-                    _ => continue,
-                }
+                break Ok(status.unwrap());
             }
 
             tokio::time::sleep(Duration::from_millis(100)).await