@@ -314,7 +314,7 @@ public void testPrepareExternalProvisioning() {
314314 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
315315
316316 PrepareExternalProvisioningAnswer answer = provisioner .prepareExternalProvisioning (
317- "host-guid " , "test-extension" , "test-extension.sh" , cmd );
317+ "host-name " , "test-extension" , "test-extension.sh" , cmd );
318318
319319 assertTrue (answer .getResult ());
320320 assertEquals ("test-net-uuid" , answer .getVirtualMachineTO ().getNics ()[0 ].getNetworkUuid ());
@@ -326,11 +326,14 @@ public void testPrepareExternalProvisioning() {
326326 public void testPrepareExternalProvisioning_ExtensionNotConfigured () {
327327 PrepareExternalProvisioningCommand cmd = mock (PrepareExternalProvisioningCommand .class );
328328
329+ String extensionName = "test-extension" ;
330+ String hostName = "host-name" ;
329331 PrepareExternalProvisioningAnswer answer = provisioner .prepareExternalProvisioning (
330- "host-guid" , "test-extension" , "nonexistent.sh" , cmd );
332+ hostName , extensionName , "nonexistent.sh" , cmd );
331333
332334 assertFalse (answer .getResult ());
333- assertEquals ("Extension not configured" , answer .getDetails ());
335+ assertNotNull (answer );
336+ assertEquals (String .format ("Extension: %s not configured for host: %s" , extensionName , hostName ), answer .getDetails ());
334337 }
335338
336339 @ Test
@@ -345,7 +348,7 @@ public void testStartInstance() {
345348 doReturn (new Pair <>(true , "{\" status\" : \" success\" , \" message\" : \" Instance started\" }" )).when (provisioner )
346349 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
347350
348- StartAnswer answer = provisioner .startInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
351+ StartAnswer answer = provisioner .startInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
349352
350353 assertTrue (answer .getResult ());
351354 Mockito .verify (logger ).debug ("Starting VM test-uuid on the external system" );
@@ -366,7 +369,7 @@ public void testStartInstanceDeploy() {
366369 doReturn (new Pair <>(true , "{\" status\" : \" success\" , \" message\" : \" Instance started\" }" )).when (provisioner )
367370 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
368371
369- StartAnswer answer = provisioner .startInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
372+ StartAnswer answer = provisioner .startInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
370373
371374 assertTrue (answer .getResult ());
372375 Mockito .verify (logger ).debug ("Deploying VM test-uuid on the external system" );
@@ -384,7 +387,7 @@ public void testStartInstanceError() {
384387 doReturn (new Pair <>(false , "{\" error\" : \" Instance failed to start\" }" )).when (provisioner )
385388 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
386389
387- StartAnswer answer = provisioner .startInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
390+ StartAnswer answer = provisioner .startInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
388391
389392 assertFalse (answer .getResult ());
390393 assertEquals ("{\" error\" : \" Instance failed to start\" }" , answer .getDetails ());
@@ -403,7 +406,7 @@ public void testStopInstance() {
403406 doReturn (new Pair <>(true , "success" )).when (provisioner )
404407 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
405408
406- StopAnswer answer = provisioner .stopInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
409+ StopAnswer answer = provisioner .stopInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
407410
408411 assertTrue (answer .getResult ());
409412 }
@@ -420,7 +423,7 @@ public void testRebootInstance() {
420423 doReturn (new Pair <>(true , "success" )).when (provisioner )
421424 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
422425
423- RebootAnswer answer = provisioner .rebootInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
426+ RebootAnswer answer = provisioner .rebootInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
424427
425428 assertTrue (answer .getResult ());
426429 }
@@ -437,7 +440,7 @@ public void testExpungeInstance() {
437440 doReturn (new Pair <>(true , "success" )).when (provisioner )
438441 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
439442
440- StopAnswer answer = provisioner .expungeInstance ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
443+ StopAnswer answer = provisioner .expungeInstance ("host-name " , "test-extension" , "test-extension.sh" , cmd );
441444
442445 assertTrue (answer .getResult ());
443446 }
@@ -490,7 +493,7 @@ public void testRunCustomAction() {
490493 doReturn (new Pair <>(true , "success" )).when (provisioner )
491494 .executeExternalCommand (anyString (), anyString (), anyMap (), anyInt (), anyString (), anyString ());
492495
493- RunCustomActionAnswer answer = provisioner .runCustomAction ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
496+ RunCustomActionAnswer answer = provisioner .runCustomAction ("host-name " , "test-extension" , "test-extension.sh" , cmd );
494497
495498 assertTrue (answer .getResult ());
496499 Mockito .verify (logger ).debug ("Executing custom action '{}' in the external system" , "test-action" );
@@ -770,7 +773,7 @@ public void getInstanceConsoleReturnsAnswerWhenConsoleDetailsAreValid() {
770773 doReturn (new Pair <>(true , validOutput )).when (provisioner )
771774 .getInstanceConsoleOnExternalSystem (anyString (), anyString (), anyString (), anyMap (), anyInt ());
772775
773- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
776+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-name " , "test-extension" , "test-extension.sh" , cmd );
774777
775778 assertNotNull (result );
776779 assertEquals ("127.0.0.1" , result .getHost ());
@@ -784,11 +787,13 @@ public void getInstanceConsoleReturnsErrorWhenExtensionNotConfigured() {
784787 GetExternalConsoleCommand cmd = mock (GetExternalConsoleCommand .class );
785788 when (provisioner .getExtensionCheckedPath (anyString (), anyString ())).thenReturn (null );
786789
787- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid" ,
788- "test-extension" , "test-extension.sh" , cmd );
790+ String extensionName = "test-extension" ;
791+ String hostName = "host-name" ;
792+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole (hostName ,
793+ extensionName , "test-extension.sh" , cmd );
789794
790795 assertNotNull (result );
791- assertEquals ("Extension not configured" , result .getDetails ());
796+ assertEquals (String . format ( "Extension: %s not configured for host: %s" , extensionName , hostName ) , result .getDetails ());
792797 }
793798
794799 @ Test
@@ -801,7 +806,7 @@ public void getInstanceConsoleReturnsErrorWhenExternalSystemFails() {
801806 doReturn (new Pair <>(false , "External system error" )).when (provisioner )
802807 .getInstanceConsoleOnExternalSystem (anyString (), anyString (), anyString (), anyMap (), anyInt ());
803808
804- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
809+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-name " , "test-extension" , "test-extension.sh" , cmd );
805810
806811 assertNotNull (result );
807812 assertEquals ("External system error" , result .getDetails ());
@@ -818,7 +823,7 @@ public void getInstanceConsoleReturnsErrorWhenConsoleObjectIsMissing() {
818823 doReturn (new Pair <>(true , invalidOutput )).when (provisioner )
819824 .getInstanceConsoleOnExternalSystem (anyString (), anyString (), anyString (), anyMap (), anyInt ());
820825
821- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
826+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-name " , "test-extension" , "test-extension.sh" , cmd );
822827
823828 assertNotNull (result );
824829 assertEquals ("Missing console object in output" , result .getDetails ());
@@ -835,7 +840,7 @@ public void getInstanceConsoleReturnsErrorWhenRequiredFieldsAreMissing() {
835840 doReturn (new Pair <>(true , incompleteOutput )).when (provisioner )
836841 .getInstanceConsoleOnExternalSystem (anyString (), anyString (), anyString (), anyMap (), anyInt ());
837842
838- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
843+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-name " , "test-extension" , "test-extension.sh" , cmd );
839844
840845 assertNotNull (result );
841846 assertEquals ("Missing required fields in output" , result .getDetails ());
@@ -852,7 +857,7 @@ public void getInstanceConsoleReturnsErrorWhenOutputParsingFails() {
852857 doReturn (new Pair <>(true , malformedOutput )).when (provisioner )
853858 .getInstanceConsoleOnExternalSystem (anyString (), anyString (), anyString (), anyMap (), anyInt ());
854859
855- GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-guid " , "test-extension" , "test-extension.sh" , cmd );
860+ GetExternalConsoleAnswer result = provisioner .getInstanceConsole ("host-name " , "test-extension" , "test-extension.sh" , cmd );
856861
857862 assertNotNull (result );
858863 assertEquals ("Failed to parse output" , result .getDetails ());
@@ -947,4 +952,22 @@ public void getSanitizedJsonStringForLogHandlesMalformedJsonGracefully() {
947952 String result = provisioner .getSanitizedJsonStringForLog (json );
948953 assertEquals ("{password:\" secret\" " , result );
949954 }
955+
956+ @ Test
957+ public void getExtensionConfigureErrorReturnsMessageWhenHostNameIsNotBlank () {
958+ String result = provisioner .getExtensionConfigureError ("test-extension" , "test-host" );
959+ assertEquals ("Extension: test-extension not configured for host: test-host" , result );
960+ }
961+
962+ @ Test
963+ public void getExtensionConfigureErrorReturnsMessageWhenHostNameIsBlank () {
964+ String result = provisioner .getExtensionConfigureError ("test-extension" , "" );
965+ assertEquals ("Extension: test-extension not configured" , result );
966+ }
967+
968+ @ Test
969+ public void getExtensionConfigureErrorReturnsMessageWhenHostNameIsNull () {
970+ String result = provisioner .getExtensionConfigureError ("test-extension" , null );
971+ assertEquals ("Extension: test-extension not configured" , result );
972+ }
950973}
0 commit comments