add missing part for assigning private ip address

microsoft · May 6, 2023 · 85201c5 · 85201c5
1 parent 4a4015f
commit 85201c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lisa/sut_orchestrator/azure/features.py b/lisa/sut_orchestrator/azure/features.py
@@ -125,11 +125,12 @@ def _start(self, wait: bool = True) -> None:
         self._node = cast(RemoteNode, self._node)
         platform: AzurePlatform = self._platform  # type: ignore
 
-        public_ip, _ = get_primary_ip_addresses(
+        public_ip, private_ip = get_primary_ip_addresses(
             platform, self._resource_group_name, get_vm(platform, self._node)
         )
         node_info = self._node.connection_info
         node_info[constants.ENVIRONMENTS_NODES_REMOTE_PUBLIC_ADDRESS] = public_ip
+        node_info[constants.ENVIRONMENTS_NODES_REMOTE_ADDRESS] = private_ip
         self._node.set_connection_info(**node_info)
         self._node._is_initialized = False
         self._node.initialize()