@@ -137,8 +137,6 @@ public class HighAvailabilityManagerImplTest {
137137
138138 @ Mock
139139 private HaWorkVO mockWork ;
140- @ Mock
141- private VirtualMachine mockVm ;
142140
143141 HighAvailabilityManagerImpl highAvailabilityManager ;
144142 HighAvailabilityManagerImpl highAvailabilityManagerSpy ;
@@ -438,15 +436,15 @@ public void testCheckAndCancelWorkIfNeeded_Success() {
438436 Mockito .when (mockWork .getHostId ()).thenReturn (1L );
439437 Mockito .doReturn (Status .Up ).when (highAvailabilityManagerSpy ).investigate (1L );
440438 Mockito .doNothing ().when (mockWork ).setStep (Step .Cancelled );
441- boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork , mockVm );
439+ boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork );
442440 assertTrue (result );
443441 Mockito .verify (mockWork ).setStep (Step .Cancelled );
444442 }
445443
446444 @ Test
447445 public void testCheckAndCancelWorkIfNeeded_StepNotInvestigating () {
448446 Mockito .when (mockWork .getStep ()).thenReturn (Step .Cancelled );
449- boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork , mockVm );
447+ boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork );
450448 assertFalse (result );
451449 Mockito .verify (mockWork , Mockito .never ()).setStep (Mockito .any ());
452450 }
@@ -455,7 +453,7 @@ public void testCheckAndCancelWorkIfNeeded_StepNotInvestigating() {
455453 public void testCheckAndCancelWorkIfNeeded_InvalidReasonType () {
456454 Mockito .when (mockWork .getStep ()).thenReturn (Step .Investigating );
457455 Mockito .when (mockWork .getReasonType ()).thenReturn (HighAvailabilityManager .ReasonType .Unknown );
458- boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork , mockVm );
456+ boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork );
459457 assertFalse (result );
460458 Mockito .verify (mockWork , Mockito .never ()).setStep (Mockito .any ());
461459 }
@@ -466,7 +464,7 @@ public void testCheckAndCancelWorkIfNeeded_HostStatusNotUp() {
466464 Mockito .when (mockWork .getReasonType ()).thenReturn (HighAvailabilityManager .ReasonType .HostDown );
467465 Mockito .when (mockWork .getHostId ()).thenReturn (1L );
468466 Mockito .doReturn (Status .Down ).when (highAvailabilityManagerSpy ).investigate (1L );
469- boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork , mockVm );
467+ boolean result = highAvailabilityManagerSpy .checkAndCancelWorkIfNeeded (mockWork );
470468 assertFalse (result );
471469 Mockito .verify (mockWork , Mockito .never ()).setStep (Mockito .any ());
472470 }
0 commit comments