Skip to content

Commit 1e911a9

Browse files
committed
change
Signed-off-by: Abhishek Kumar <abhishek.mrt22@gmail.com>
1 parent e06cbf9 commit 1e911a9

3 files changed

Lines changed: 20 additions & 1 deletion

File tree

plugins/hypervisors/external/src/main/java/org/apache/cloudstack/hypervisor/external/provisioner/ExternalPathPayloadProvisioner.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,10 @@ public String getName() {
138138
protected Map<String, Object> loadAccessDetails(Map<String, Map<String, String>> externalDetails,
139139
VirtualMachineTO virtualMachineTO) {
140140
Map<String, Object> modifiedDetails = new HashMap<>();
141+
if (MapUtils.isNotEmpty(externalDetails) && externalDetails.containsKey(ApiConstants.CALLER)) {
142+
modifiedDetails.put(ApiConstants.CALLER, externalDetails.get(ApiConstants.CALLER));
143+
externalDetails.remove(ApiConstants.CALLER);
144+
}
141145
if (MapUtils.isNotEmpty(externalDetails)) {
142146
modifiedDetails.put(ApiConstants.EXTERNAL_DETAILS, externalDetails);
143147
}

plugins/hypervisors/external/src/test/java/org/apache/cloudstack/hypervisor/external/provisioner/ExternalPathPayloadProvisionerTest.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@
9292
import com.cloud.vm.UserVmVO;
9393
import com.cloud.vm.VirtualMachine;
9494
import com.cloud.vm.VirtualMachineProfile;
95+
import com.cloud.vm.VmDetailConstants;
9596
import com.cloud.vm.dao.UserVmDao;
9697

9798
@RunWith(MockitoJUnitRunner.class)
@@ -205,6 +206,20 @@ public void testLoadAccessDetailsWithNullExternalDetails() {
205206
assertEquals("test-vm", result.get(ApiConstants.VIRTUAL_MACHINE_NAME));
206207
}
207208

209+
@Test
210+
public void testLoadAccessDetails_WithCaller() {
211+
Map<String, Map<String, String>> externalDetails = new HashMap<>();
212+
externalDetails.put(ApiConstants.EXTENSION, Map.of("key1", "value1"));
213+
externalDetails.put(ApiConstants.CALLER, Map.of("key2", "value2"));
214+
Map<String, Object> result = provisioner.loadAccessDetails(externalDetails, null);
215+
216+
assertNotNull(result);
217+
assertNotNull(result.get(ApiConstants.EXTERNAL_DETAILS));
218+
assertNotNull(((Map<String, String>) result.get(ApiConstants.EXTERNAL_DETAILS)).get(ApiConstants.EXTENSION));
219+
assertNotNull(result.get(ApiConstants.CALLER));
220+
assertNull(result.get(VmDetailConstants.CLOUDSTACK_VM_DETAILS));
221+
}
222+
208223
@Test
209224
public void testGetExtensionCheckedPathValidFile() {
210225
String result = provisioner.getExtensionCheckedPath("test-extension", "test-extension.sh");

server/src/test/java/org/apache/cloudstack/consoleproxy/ConsoleAccessManagerImplTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -413,7 +413,7 @@ public void getConsoleConnectionDetailsForExternalVmSetsDetailsWhenAnswerIsValid
413413

414414
Mockito.when(answer.getResult()).thenReturn(true);
415415
Mockito.when(answer.getUrl()).thenReturn(url);
416-
Mockito.when(answer.getProtocol()).thenReturn("direct");
416+
Mockito.when(answer.getProtocol()).thenReturn(ConsoleConnectionDetails.Mode.Direct.name());
417417
Mockito.when(managementServer.getExternalVmConsole(vm, host)).thenReturn(answer);
418418

419419
ConsoleConnectionDetails result = consoleAccessManager.getConsoleConnectionDetailsForExternalVm(details, vm, host);

0 commit comments

Comments
 (0)