Fix test.

apache · Mar 19, 2024 · d7eb681 · d7eb681
1 parent ef2e278
commit d7eb681
Showing 1 changed file with 2 additions and 15 deletions.
diff --git a/...lti-stage-query/src/test/java/org/apache/druid/msq/indexing/IndexerWorkerContextTest.java b/...lti-stage-query/src/test/java/org/apache/druid/msq/indexing/IndexerWorkerContextTest.java
@@ -54,20 +54,6 @@ public void setup()
     );
   }
 
-  @Test
-  public void testControllerCheckerRunnableExitsWhenEmptyStatus()
-  {
-    final ServiceLocator controllerLocatorMock = Mockito.mock(ServiceLocator.class);
-    Mockito.when(controllerLocatorMock.locate())
-           .thenReturn(Futures.immediateFuture(ServiceLocations.forLocations(Collections.emptySet())));
-
-    final Worker workerMock = Mockito.mock(Worker.class);
-
-    indexerWorkerContext.controllerCheckerRunnable(controllerLocatorMock, workerMock);
-    Mockito.verify(controllerLocatorMock, Mockito.times(1)).locate();
-    Mockito.verify(workerMock, Mockito.times(1)).controllerFailed();
-  }
-
   @Test
   public void testControllerCheckerRunnableExitsOnlyWhenClosedStatus()
   {
@@ -76,12 +62,13 @@ public void testControllerCheckerRunnableExitsOnlyWhenClosedStatus()
            .thenReturn(Futures.immediateFuture(ServiceLocations.forLocation(new ServiceLocation("h", 1, -1, "/"))))
            // Done to check the behavior of the runnable, the situation of exiting after success might not occur actually
            .thenReturn(Futures.immediateFuture(ServiceLocations.forLocation(new ServiceLocation("h", 1, -1, "/"))))
+           .thenReturn(Futures.immediateFuture(ServiceLocations.forLocations(Collections.emptySet())))
            .thenReturn(Futures.immediateFuture(ServiceLocations.closed()));
 
     final Worker workerMock = Mockito.mock(Worker.class);
 
     indexerWorkerContext.controllerCheckerRunnable(controllerLocatorMock, workerMock);
-    Mockito.verify(controllerLocatorMock, Mockito.times(3)).locate();
+    Mockito.verify(controllerLocatorMock, Mockito.times(4)).locate();
     Mockito.verify(workerMock, Mockito.times(1)).controllerFailed();
   }
 }