2222import static org .junit .Assert .assertTrue ;
2323import static org .mockito .ArgumentMatchers .any ;
2424import static org .mockito .ArgumentMatchers .anyBoolean ;
25+ import static org .mockito .ArgumentMatchers .anyInt ;
2526import static org .mockito .ArgumentMatchers .anyString ;
2627import static org .mockito .Mockito .spy ;
2728import static org .mockito .Mockito .when ;
@@ -236,7 +237,7 @@ public void testCloneHypervisorVM_NoExternalVM() throws Exception {
236237 }); MockedConstruction <DatacenterMO > ignored3 = Mockito .mockConstruction (DatacenterMO .class , withSettings ().spiedInstance (dataCenterMO ), (mockDatacenterMO , contextDatacenterMO ) -> {
237238 Mockito .doReturn (null ).when (mockDatacenterMO ).findVm (vmName );
238239 })) {
239- vMwareGuru .cloneHypervisorVMOutOfBand (hostIp , vmName , params );
240+ vMwareGuru .getHypervisorVMOutOfBandAndCloneIfRequired (hostIp , vmName , params );
240241 }
241242 }
242243
@@ -280,7 +281,7 @@ public void testCloneHypervisorVM_WindowsVMRunning() throws Exception {
280281 }); MockedConstruction <DatacenterMO > ignored3 = Mockito .mockConstruction (DatacenterMO .class , withSettings ().spiedInstance (dataCenterMO ), (mockDatacenterMO , contextDatacenterMO ) -> {
281282 Mockito .doReturn (vmMo ).when (mockDatacenterMO ).findVm (vmName );
282283 })) {
283- vMwareGuru .cloneHypervisorVMOutOfBand (hostIp , vmName , params );
284+ vMwareGuru .getHypervisorVMOutOfBandAndCloneIfRequired (hostIp , vmName , params );
284285 }
285286 }
286287
@@ -307,8 +308,12 @@ public void testCloneHypervisorVM_GetDatastoresFailed() throws Exception {
307308 DatacenterMO dataCenterMO = spy (new DatacenterMO (vmwareContext , datacenterName ));
308309 VirtualMachineMO vmMo = Mockito .mock (VirtualMachineMO .class );
309310 HostMO hostMo = Mockito .mock (HostMO .class );
310- Mockito .doReturn (VirtualMachinePowerState .POWERED_OFF ).when (vmMo ).getPowerState ();
311+ Mockito .doReturn (VirtualMachinePowerState .POWERED_ON ).when (vmMo ).getPowerState ();
311312 Mockito .doReturn (hostMo ).when (vmMo ).getRunningHost ();
313+ UnmanagedInstanceTO instance = Mockito .mock (UnmanagedInstanceTO .class );
314+ Mockito .doReturn ("CentOS" ).when (instance ).getOperatingSystem ();
315+ Mockito .doReturn ("test-cluster" ).when (instance ).getClusterName ();
316+ when (VmwareHelper .getUnmanagedInstance (hostMo , vmMo )).thenReturn (instance );
312317 List <DatastoreMO > datastores = new ArrayList <>();
313318 Mockito .doReturn (datastores ).when (vmMo ).getAllDatastores ();
314319
@@ -323,7 +328,7 @@ public void testCloneHypervisorVM_GetDatastoresFailed() throws Exception {
323328 }); MockedConstruction <DatacenterMO > ignored3 = Mockito .mockConstruction (DatacenterMO .class , withSettings ().spiedInstance (dataCenterMO ), (mockDatacenterMO , contextDatacenterMO ) -> {
324329 Mockito .doReturn (vmMo ).when (mockDatacenterMO ).findVm (vmName );
325330 })) {
326- vMwareGuru .cloneHypervisorVMOutOfBand (hostIp , vmName , params );
331+ vMwareGuru .getHypervisorVMOutOfBandAndCloneIfRequired (hostIp , vmName , params );
327332 }
328333 }
329334
@@ -350,9 +355,13 @@ public void testCloneHypervisorVM_CloneVMFailed() throws Exception {
350355 DatacenterMO dataCenterMO = spy (new DatacenterMO (vmwareContext , datacenterName ));
351356 VirtualMachineMO vmMo = Mockito .mock (VirtualMachineMO .class );
352357 HostMO hostMo = Mockito .mock (HostMO .class );
353- Mockito .doReturn (VirtualMachinePowerState .POWERED_OFF ).when (vmMo ).getPowerState ();
358+ Mockito .doReturn (VirtualMachinePowerState .POWERED_ON ).when (vmMo ).getPowerState ();
354359 Mockito .doReturn (hostMo ).when (vmMo ).getRunningHost ();
355360 Mockito .doReturn (mor ).when (hostMo ).getHyperHostOwnerResourcePool ();
361+ UnmanagedInstanceTO instance = Mockito .mock (UnmanagedInstanceTO .class );
362+ Mockito .doReturn ("CentOS" ).when (instance ).getOperatingSystem ();
363+ Mockito .doReturn ("test-cluster" ).when (instance ).getClusterName ();
364+ when (VmwareHelper .getUnmanagedInstance (hostMo , vmMo )).thenReturn (instance );
356365 DatastoreMO datastoreMO = Mockito .mock (DatastoreMO .class );
357366 Mockito .doReturn (mor ).when (datastoreMO ).getMor ();
358367 List <DatastoreMO > datastores = new ArrayList <>();
@@ -373,7 +382,7 @@ public void testCloneHypervisorVM_CloneVMFailed() throws Exception {
373382 Mockito .doReturn (mor ).when (mockDatacenterMO ).getVmFolder ();
374383 Mockito .doReturn (mor ).when (mockDatacenterMO ).getMor ();
375384 })) {
376- vMwareGuru .cloneHypervisorVMOutOfBand (hostIp , vmName , params );
385+ vMwareGuru .getHypervisorVMOutOfBandAndCloneIfRequired (hostIp , vmName , params );
377386 }
378387 }
379388
@@ -400,7 +409,7 @@ public void testCloneHypervisorVM() throws Exception {
400409 DatacenterMO dataCenterMO = spy (new DatacenterMO (vmwareContext , datacenterName ));
401410 VirtualMachineMO vmMo = Mockito .mock (VirtualMachineMO .class );
402411 HostMO hostMo = Mockito .mock (HostMO .class );
403- Mockito .doReturn (VirtualMachinePowerState .POWERED_OFF ).when (vmMo ).getPowerState ();
412+ Mockito .doReturn (VirtualMachinePowerState .POWERED_ON ).when (vmMo ).getPowerState ();
404413 Mockito .doReturn (hostMo ).when (vmMo ).getRunningHost ();
405414 Mockito .doReturn (mor ).when (hostMo ).getHyperHostOwnerResourcePool ();
406415 Mockito .doReturn (mor ).when (hostMo ).getMor ();
@@ -411,6 +420,8 @@ public void testCloneHypervisorVM() throws Exception {
411420 Mockito .doReturn (datastores ).when (vmMo ).getAllDatastores ();
412421 Mockito .lenient ().doReturn (true ).when (vmMo ).createFullClone (anyString (), any (ManagedObjectReference .class ), any (ManagedObjectReference .class ), any (ManagedObjectReference .class ), any (Storage .ProvisioningType .class ));
413422 UnmanagedInstanceTO instance = Mockito .mock (UnmanagedInstanceTO .class );
423+ Mockito .doReturn ("CentOS" ).when (instance ).getOperatingSystem ();
424+ Mockito .doReturn ("test-cluster" ).when (instance ).getClusterName ();
414425 when (VmwareHelper .getUnmanagedInstance (hostMo , vmMo )).thenReturn (instance );
415426 UnmanagedInstanceTO .Disk disk = Mockito .mock (UnmanagedInstanceTO .Disk .class );
416427 Mockito .doReturn ("1" ).when (disk ).getDiskId ();
@@ -431,7 +442,7 @@ public void testCloneHypervisorVM() throws Exception {
431442 Mockito .doReturn (mor ).when (mockDatacenterMO ).getVmFolder ();
432443 Mockito .doReturn (mor ).when (mockDatacenterMO ).getMor ();
433444 })) {
434- UnmanagedInstanceTO clonedVm = vMwareGuru .cloneHypervisorVMOutOfBand (hostIp , vmName , params );
445+ Pair < UnmanagedInstanceTO , Boolean > clonedVm = vMwareGuru .getHypervisorVMOutOfBandAndCloneIfRequired (hostIp , vmName , params );
435446 assertNotNull (clonedVm );
436447 }
437448 }
@@ -470,7 +481,7 @@ public void testCreateVMTemplateFileOutOfBand_NoClonedVM() throws Exception {
470481 }); MockedConstruction <DatacenterMO > ignored3 = Mockito .mockConstruction (DatacenterMO .class , withSettings ().spiedInstance (dataCenterMO ), (mockDatacenterMO , contextDatacenterMO ) -> {
471482 Mockito .doReturn (null ).when (mockDatacenterMO ).findVm (vmName );
472483 })) {
473- vMwareGuru .createVMTemplateOutOfBand (hostIp , vmName , params , dataStore );
484+ vMwareGuru .createVMTemplateOutOfBand (hostIp , vmName , params , dataStore , - 1 );
474485 }
475486 }
476487
@@ -496,7 +507,7 @@ public void testCreateVMTemplateFileOutOfBand() throws Exception {
496507 Mockito .doReturn (mor ).when (vimClient ).getDecendentMoRef (any (ManagedObjectReference .class ), anyString (), anyString ());
497508 DatacenterMO dataCenterMO = spy (new DatacenterMO (vmwareContext , datacenterName ));
498509 VirtualMachineMO vmMo = Mockito .mock (VirtualMachineMO .class );
499- Mockito .doNothing ().when (vmMo ).exportVm (anyString (), anyString (), anyBoolean (), anyBoolean ());
510+ Mockito .doNothing ().when (vmMo ).exportVm (anyString (), anyString (), anyBoolean (), anyBoolean (), anyInt () );
500511 NfsTO dataStore = Mockito .mock (NfsTO .class );
501512 Mockito .doReturn ("nfs://10.1.1.4/testdir" ).when (dataStore ).getUrl ();
502513
@@ -511,7 +522,7 @@ public void testCreateVMTemplateFileOutOfBand() throws Exception {
511522 }); MockedConstruction <DatacenterMO > ignored3 = Mockito .mockConstruction (DatacenterMO .class , withSettings ().spiedInstance (dataCenterMO ), (mockDatacenterMO , contextDatacenterMO ) -> {
512523 Mockito .doReturn (vmMo ).when (mockDatacenterMO ).findVm (vmName );
513524 })) {
514- String vmTemplate = vMwareGuru .createVMTemplateOutOfBand (hostIp , vmName , params , dataStore );
525+ String vmTemplate = vMwareGuru .createVMTemplateOutOfBand (hostIp , vmName , params , dataStore , - 1 );
515526 assertNotNull (vmTemplate );
516527 assertTrue (UuidUtils .isUuid (vmTemplate ));
517528 }
0 commit comments