diff options
-rw-r--r-- | multivimbroker/multivimbroker/tests/test_fileutil.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/multivimbroker/multivimbroker/tests/test_fileutil.py b/multivimbroker/multivimbroker/tests/test_fileutil.py index 521c3b4..9abd23c 100644 --- a/multivimbroker/multivimbroker/tests/test_fileutil.py +++ b/multivimbroker/multivimbroker/tests/test_fileutil.py @@ -40,3 +40,12 @@ class TestFileutil(unittest.TestCase): new_path = "/tmp/tests" fileutil.delete_dirs(new_path) mock_rmtree.assert_called_once_with(new_path) + + @mock.patch.object(os.path, "exists") + @mock.patch("shutil.rmtree") + def test_delete_dirs_failed(self, mock_rmtree, mock_exists): + mock_exists.return_value = True + mock_rmtree.side_effect = [Exception("Fake exception")] + new_path = "/tmp/tests" + fileutil.delete_dirs(new_path) + mock_rmtree.assert_called_once_with(new_path) |