36 lines
1013 B
Python
36 lines
1013 B
Python
from pathlib import Path
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from croaker import pidfile
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"pid,terminate,kill_result,broken",
|
|
[
|
|
("pid", False, None, False), # running proc, no terminate
|
|
("pid", True, True, False), # running proc, terminate
|
|
("pid", True, ProcessLookupError, True), # stale pid
|
|
(None, None, None, False), # no running proc
|
|
],
|
|
)
|
|
def test_pidfile(monkeypatch, pid, terminate, kill_result, broken):
|
|
monkeypatch.setattr(
|
|
pidfile._pidfile,
|
|
"TimeoutPIDLockFile",
|
|
MagicMock(
|
|
**{
|
|
"return_value.read_pid.return_value": pid,
|
|
}
|
|
),
|
|
)
|
|
monkeypatch.setattr(
|
|
pidfile.os,
|
|
"kill",
|
|
MagicMock(**{"side_effect": kill_result if type(kill_result) is Exception else [kill_result]}),
|
|
)
|
|
|
|
ret = pidfile.pidfile(pidfile_path=Path("/dev/null"), terminate_if_running=terminate)
|
|
assert ret.break_lock.called == broken
|