diff --git a/cloudbaseinit/tests/plugins/windows/test_azureguestagent.py b/cloudbaseinit/tests/plugins/windows/test_azureguestagent.py index 4ea9c9c5..51499ceb 100644 --- a/cloudbaseinit/tests/plugins/windows/test_azureguestagent.py +++ b/cloudbaseinit/tests/plugins/windows/test_azureguestagent.py @@ -80,6 +80,7 @@ class AzureGuestAgentPluginTest(unittest.TestCase): mock_exists, mock_rmtree): mock_rmtree.side_effect = (None, Exception) mock_exists.return_value = True + mock_os_getenv.return_value = "fake_path" with self.snatcher: self._azureagentplugin._remove_azure_dirs() mock_os_getenv.assert_called_with("SystemDrive")