saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Add clean_proc_dir tests

Browse source
This commit is contained in:
Max Arnold 2024-02-08 02:57:38 -06:00 • committed by Daniel Wozniak
parent 0401be21ed
commit 135223313a
1 changed files with 53 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
tests/pytests/unit/utils/test_master.py
View file

@ -8,8 +8,15 @@ from tests.support.mock import mock_open, patch
@pytest.fixture
def proc_file(tmp_path):
path = str(tmp_path / "20240208071139934305")
def proc_dir(tmp_path):
path = tmp_path / "proc"
path.mkdir()
return str(path)
@pytest.fixture
def proc_file(proc_dir):
path = os.path.join(proc_dir, "20240208071139934305")
data = {
"fun": "runner.state.orch",
"jid": "20240208071139934305",
@ -125,3 +132,47 @@ def test_read_proc_file_running_not_salt_error(proc_file):
), patch("os.remove", side_effect=OSError):
assert salt.utils.master._read_proc_file(proc_file[0], {}) is None
assert os.path.exists(proc_file[0]) is True
def test_clean_proc_dir_no_data(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", return_value=None):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is False
def test_clean_proc_dir_no_data_error(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", return_value=None), patch(
"os.remove", side_effect=OSError
):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is True
def test_clean_proc_dir_running(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", return_value=proc_file[1]), patch(
"salt.utils.master._check_cmdline", return_value=True
):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is True
def test_clean_proc_dir_running_read_error(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", side_effect=OSError):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is True
def test_clean_proc_dir_not_running(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", return_value=proc_file[1]), patch(
"salt.utils.master._check_cmdline", return_value=False
):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is False
def test_clean_proc_dir_not_running_error(tmp_path, proc_file):
with patch("salt.utils.master._read_proc_file", return_value=proc_file[1]), patch(
"salt.utils.master._check_cmdline", return_value=False
), patch("os.remove", side_effect=OSError):
salt.utils.master.clean_proc_dir({"cachedir": str(tmp_path)})
assert os.path.exists(proc_file[0]) is True