@@ -263,7 +263,8 @@ public void testRecalculateHostCapacitiesLogsAndContinuesOnTaskFailure() {
263263 @ Test
264264 public void testRecalculateHostCapacitiesReusesExecutorAcrossCalls () throws Exception {
265265 Mockito .when (hostDao .listIdsByType (Host .Type .Routing )).thenReturn (List .of (1L ));
266- Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (Mockito .mock (HostVO .class ));
266+ HostVO hostMock = Mockito .mock (HostVO .class );
267+ Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (hostMock );
267268
268269 alertManagerImplMock .recalculateHostCapacities ();
269270 ExecutorService firstExecutor = getCapacityExecutorService ();
@@ -281,7 +282,8 @@ public void testRecalculateHostCapacitiesReusesExecutorAcrossCalls() throws Exce
281282 @ Test
282283 public void testRecalculateHostCapacitiesRecreatesExecutorAfterShutdown () throws Exception {
283284 Mockito .when (hostDao .listIdsByType (Host .Type .Routing )).thenReturn (List .of (1L ));
284- Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (Mockito .mock (HostVO .class ));
285+ HostVO hostMock = Mockito .mock (HostVO .class );
286+ Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (hostMock );
285287
286288 alertManagerImplMock .recalculateHostCapacities ();
287289 ExecutorService firstExecutor = getCapacityExecutorService ();
@@ -299,7 +301,8 @@ public void testStopShutsDownCapacityExecutorServiceWhenPresent() throws Excepti
299301 Timer timerMock = Mockito .mock (Timer .class );
300302 setTimer (timerMock );
301303 Mockito .when (hostDao .listIdsByType (Host .Type .Routing )).thenReturn (List .of (1L ));
302- Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (Mockito .mock (HostVO .class ));
304+ HostVO hostMock = Mockito .mock (HostVO .class );
305+ Mockito .when (hostDao .findById (Mockito .anyLong ())).thenReturn (hostMock );
303306 alertManagerImplMock .recalculateHostCapacities ();
304307
305308 boolean result = alertManagerImplMock .stop ();
0 commit comments