from salt.returners.local_cache import _remove_job_dir
from tests.support.mock import patch


def test_remove_job_dir():
    # Test that _remove_job_dir job will catch error
    for e in (NotADirectoryError, OSError):
        with patch("shutil.rmtree", side_effect=e("Node Corruption!")):
            with patch("salt.returners.local_cache.log") as log_func:
                _remove_job_dir("cache")
                log_func.assert_called_once()

    # Test that _remove_job_dir job will not catch other errors
    with patch("shutil.rmtree", side_effect=FileExistsError()):
        try:
            _remove_job_dir("cache")
        except FileExistsError:
            pass

from salt.returners.local_cache import _remove_job_dir
from tests.support.mock import patch


def test_remove_job_dir():
    # Test that _remove_job_dir job will catch error
    for e in (NotADirectoryError, OSError):
        with patch("shutil.rmtree", side_effect=e("Node Corruption!")):
            with patch("salt.returners.local_cache.log.error") as log_func:
                _remove_job_dir("cache")
                log_func.assert_called_once()

    # Test that _remove_job_dir job will not catch other errors
    with patch("shutil.rmtree", side_effect=FileExistsError()):
        try:
            _remove_job_dir("cache")
        except FileExistsError:
            pass

Update test_local_cache.py