diff --git a/cloudbaseinit/osutils/windows.py b/cloudbaseinit/osutils/windows.py index 5630d212..ae5b1884 100644 --- a/cloudbaseinit/osutils/windows.py +++ b/cloudbaseinit/osutils/windows.py @@ -858,13 +858,12 @@ class WindowsUtils(base.BaseOSUtils): 'value "%(mtu)s" failed' % {'name': name, 'mtu': mtu}) def rename_network_adapter(self, old_name, new_name): - base_dir = self._get_system_dir() - netsh_path = os.path.join(base_dir, 'netsh.exe') - - args = [netsh_path, "interface", "set", "interface", - 'name=%s' % old_name, 'newname=%s' % new_name] - (out, err, ret_val) = self.execute_process(args, shell=False) - if ret_val: + net_adapter = self._get_network_adapter(old_name) + net_adapter.NetConnectionID = new_name + try: + net_adapter.put() + self._get_network_adapter(new_name) + except Exception: raise exception.CloudbaseInitException( 'Renaming interface "%(old_name)s" to "%(new_name)s" ' 'failed' % {'old_name': old_name, 'new_name': new_name}) diff --git a/cloudbaseinit/tests/osutils/test_windows.py b/cloudbaseinit/tests/osutils/test_windows.py index a9eec533..48ecff5a 100644 --- a/cloudbaseinit/tests/osutils/test_windows.py +++ b/cloudbaseinit/tests/osutils/test_windows.py @@ -839,28 +839,34 @@ class TestWindowsUtils(testutils.CloudbaseInitTestBase): self._test_set_static_network_config(ipv6=True) @mock.patch('cloudbaseinit.osutils.windows.WindowsUtils' - '.execute_process') - @mock.patch('cloudbaseinit.osutils.windows.WindowsUtils' - '._get_system_dir') - def _test_rename_network_adapter(self, should_fail, mock_get_system_dir, - mock_execute_process): - base_dir = "fake path" + '._get_network_adapter') + def _test_rename_network_adapter(self, put_exception, + mock_get_network_adapter): old_name = "fake_old" new_name = "fake_new" - mock_get_system_dir.return_value = base_dir - ret_val = 1 if should_fail else 0 - mock_execute_process.return_value = (None, None, ret_val) - if should_fail: + adapter = mock.Mock() + adapter.Name = mock.sentinel.old_name + mock_get_network_adapter.return_value = adapter + get_network_adapter_called = 1 + if put_exception: + adapter.put.side_effect = Exception("fake exception") with self.assertRaises(exception.CloudbaseInitException): self._winutils.rename_network_adapter(old_name, new_name) else: + get_network_adapter_called = 2 self._winutils.rename_network_adapter(old_name, new_name) - mock_get_system_dir.assert_called_once_with() - args = [os.path.join(base_dir, "netsh.exe"), "interface", "set", - "interface", 'name=%s' % old_name, 'newname=%s' % new_name] - mock_execute_process.assert_called_once_with(args, shell=False) + adapter.put.assert_called() + self.assertEqual(new_name, adapter.NetConnectionID) + + self.assertEqual(mock_get_network_adapter.call_count, + get_network_adapter_called) + self.assertEqual(mock_get_network_adapter.call_args_list[0].args, + (old_name,)) + if not exception: + self.assertEqual(mock_get_network_adapter.call_args_list[1].args, + (new_name,)) def _test_get_config_key_name(self, section): response = self._winutils._get_config_key_name(section) @@ -879,7 +885,7 @@ class TestWindowsUtils(testutils.CloudbaseInitTestBase): def test_rename_network_adapter(self): self._test_rename_network_adapter(False) - def test_rename_network_adapter_fail(self): + def test_rename_network_adapter_fails(self): self._test_rename_network_adapter(True) @mock.patch('cloudbaseinit.osutils.windows.WindowsUtils'