fix(host): daemon 循环把显式 --state 透传给 recover/run-due 子进程
复核结论:应继承,不是原行为有意。daemon 本体的 pidfile/ logfile 按解析后 args.state 落盘,但旧循环只传 --config, 子进程回落到配置 state(或默认 state);CLI --state 与 配置 state 不同时,recover/run-due 跑的不是同一 state 文件。 修法:main() 解析前先记 explicit_state(argparse --state 默认 None,未给即 None);循环里仅当 explicit_state 非 None 时追加 --state,未显式给时不加、子进程按原规则解析, 与 daemon 本体一致、行为不变。daemonize() 本身只接 pidfile/logfile,不涉及 state,问题在 main() 循环里。 回归:DaemonStatePassthroughTest(2 项),旧代码上显式 项 fail、无显式项 pass,新代码上 2 项全 pass。
This commit is contained in:
1 parent
1009e9d783
commit
acd3b5314f
2 files changed
+105
-3
No files matched your search
@@ -153,5 +153,96 @@ class ConfigFileCloseTest(unittest.TestCase):
|
||||
if issubclass(w.category, ResourceWarning)], [])
|
||||
|
||||
|
||||
class DaemonStatePassthroughTest(unittest.TestCase):
|
||||
"""第 3 项:daemon 循环的子进程是否继承显式 --state。
|
||||
|
||||
复核结论:应继承。daemon 本体的 pidfile/logfile 都按
|
||||
解析后的 args.state 落盘,但旧循环只给子进程传 --config,
|
||||
子进程会回落到配置文件里的 state(或默认 state);当
|
||||
CLI --state 与配置 state 不同时,recover/run-due 跑的
|
||||
就不是 daemon 本体那个 state 文件。daemonize() 本身只
|
||||
接 pidfile/logfile,不涉及 state,问题在 main() 的循环里。
|
||||
|
||||
测试直接跑 main() 的 daemon --foreground 循环:子进程
|
||||
调用被 mock 捕获(不真拉 host.py 子进程),循环由定时
|
||||
发给本进程的 SIGTERM 停下(daemon 的 handler 只置 flag)。
|
||||
"""
|
||||
|
||||
def _run_daemon_once(self, argv):
|
||||
import os
|
||||
import signal
|
||||
import time
|
||||
from unittest import mock
|
||||
|
||||
calls = []
|
||||
|
||||
def fake_run(cmd, **kw):
|
||||
calls.append(list(cmd))
|
||||
return mock.Mock(returncode=0)
|
||||
|
||||
def sender():
|
||||
time.sleep(1.5)
|
||||
try:
|
||||
os.kill(os.getpid(), signal.SIGTERM)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
old_term = signal.getsignal(signal.SIGTERM)
|
||||
old_int = signal.getsignal(signal.SIGINT)
|
||||
sender_thread = threading.Thread(target=sender, daemon=True)
|
||||
out = io.StringIO()
|
||||
try:
|
||||
sender_thread.start()
|
||||
with mock.patch.object(sys, "argv", argv):
|
||||
with mock.patch("subprocess.run",
|
||||
side_effect=fake_run):
|
||||
with contextlib.redirect_stdout(out):
|
||||
host.main()
|
||||
finally:
|
||||
sender_thread.join(timeout=5)
|
||||
signal.signal(signal.SIGTERM, old_term)
|
||||
signal.signal(signal.SIGINT, old_int)
|
||||
return calls, out.getvalue()
|
||||
|
||||
def test_explicit_state_passed_to_children(self):
|
||||
import os
|
||||
import tempfile
|
||||
with tempfile.TemporaryDirectory() as tmp:
|
||||
cfg_path = os.path.join(tmp, "cfg.json")
|
||||
cli_state = os.path.join(tmp, "cli-state.json")
|
||||
with open(cfg_path, "w", encoding="utf-8") as f:
|
||||
json.dump({"state": "config-state.json"}, f)
|
||||
argv = ["host.py", "--config", cfg_path, "--state", cli_state,
|
||||
"daemon", "--foreground", "--interval", "60"]
|
||||
calls, out = self._run_daemon_once(argv)
|
||||
self.assertEqual(len(calls), 2)
|
||||
self.assertIn(f"state={cli_state}", out)
|
||||
for call in calls:
|
||||
self.assertEqual(call[1], str(REPO_ROOT / "host.py"))
|
||||
self.assertIn("--config", call)
|
||||
self.assertIn(cfg_path, call)
|
||||
self.assertIn("--state", call)
|
||||
self.assertIn(cli_state, call)
|
||||
self.assertNotIn("config-state.json", " ".join(call))
|
||||
self.assertEqual(calls[0][-3:],
|
||||
["recover", "--older-than", "3600"])
|
||||
self.assertEqual(calls[1][-1], "run-due")
|
||||
|
||||
def test_no_explicit_state_children_unchanged(self):
|
||||
import os
|
||||
import tempfile
|
||||
with tempfile.TemporaryDirectory() as tmp:
|
||||
cfg_path = os.path.join(tmp, "cfg.json")
|
||||
with open(cfg_path, "w", encoding="utf-8") as f:
|
||||
json.dump({"state": "config-state.json"}, f)
|
||||
argv = ["host.py", "--config", cfg_path,
|
||||
"daemon", "--foreground", "--interval", "60"]
|
||||
calls, out = self._run_daemon_once(argv)
|
||||
self.assertEqual(len(calls), 2)
|
||||
self.assertIn("config-state.json", out)
|
||||
for call in calls:
|
||||
self.assertNotIn("--state", call)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user