fix: workload launcher retry test (#14992)

airbytehq · Jan 13, 2025 · 350af62 · 350af62
1 parent a0c7ca1
commit 350af62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airbyte-workload-launcher/src/test/kotlin/pods/KubePodLauncherTest.kt b/airbyte-workload-launcher/src/test/kotlin/pods/KubePodLauncherTest.kt
@@ -303,7 +303,7 @@ class KubePodLauncherTest {
     every { pods.inNamespace(any()) } returns namespaceable
     every { namespaceable.withLabels(any()) } returns labels
     every { labels.waitUntilCondition(any(), any(), any()) } throws
-      KubernetesClientTimeoutException(hasMetadata, 2L, TimeUnit.SECONDS)
+      RuntimeException()
     every { kubernetesClient.pods() } returns pods
 
     val kubePodLauncher =
@@ -316,7 +316,7 @@ class KubePodLauncherTest {
         null,
       )
 
-    assertThrows<KubernetesClientException> {
+    assertThrows<RuntimeException> {
       kubePodLauncher.waitForPodReadyOrTerminal(mapOf("label" to "value"), Duration.ofSeconds(30))
     }
     assertEquals(0, counter.get())