hotfix: fix scheduler client tests (#2270)

airbytehq · Mar 3, 2021 · 6f9b8ea · 6f9b8ea
1 parent cb91ab7
commit 6f9b8ea
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...te-scheduler/src/test/java/io/airbyte/scheduler/client/DefaultSchedulerJobClientTest.java b/...te-scheduler/src/test/java/io/airbyte/scheduler/client/DefaultSchedulerJobClientTest.java
@@ -134,7 +134,7 @@ void testCreateResetConnectionJob() throws IOException {
     final StandardSync standardSync = mock(StandardSync.class);
     final String destinationDockerImage = "airbyte/spaceport";
     when(jobCreator.createResetConnectionJob(destination, standardSync, destinationDockerImage)).thenReturn(Optional.of(JOB_ID));
-    doReturn(job).when(client).waitUntilJobIsTerminalOrTimeout(JOB_ID);
+    when(jobPersistence.getJob(JOB_ID)).thenReturn(job);
 
     assertEquals(job, client.createOrGetActiveResetConnectionJob(destination, standardSync, destinationDockerImage));
   }
@@ -151,8 +151,7 @@ void testCreateResetConnectionJobAlreadyExist() throws IOException {
     final Job currentJob = mock(Job.class);
     when(currentJob.getId()).thenReturn(42L);
     when(jobPersistence.getLastReplicationJob(connectionUuid)).thenReturn(Optional.of(currentJob));
-
-    doReturn(currentJob).when(client).waitUntilJobIsTerminalOrTimeout(42L);
+    when(jobPersistence.getJob(42L)).thenReturn(currentJob);
 
     assertEquals(currentJob, client.createOrGetActiveResetConnectionJob(destination, standardSync, destinationDockerImage));
   }