liuwenxin
108bf87225
补齐源快照、资料预演、正式导入及工资核算的依赖链路。 新增版本化输入、逐段 FIFO、生成锁和结构化当月调整， 并提供默认关闭的周期采集、有限重试及 Windows 托管。 原始客户资料、私有配置及首页改动不纳入本次提交。 BREAKING CHANGE: 升级前需应用新增迁移，并通过 PAYROLL_MAKER_USER_ID 绑定实际工资制作人，否则业务只读。 旧单批次核算默认关闭；历史兼容需显式启用 PAYROLL_ALLOW_LEGACY_CALCULATION。生产启用和真实金额待验收。
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
"""停止标记控制真实CLI领取边界，不中断正在完成的单次处理。"""
|
|
|
|
from argparse import Namespace
|
|
from contextlib import nullcontext
|
|
from types import SimpleNamespace
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from app import cli
|
|
from app.core.config import Settings
|
|
|
|
|
|
@pytest.mark.parametrize("already_stopped", [False, True])
|
|
def test_停止后不再领取下一任务(monkeypatch, tmp_path, already_stopped):
|
|
marker = tmp_path / "stop.requested"
|
|
if already_stopped:
|
|
marker.touch()
|
|
settings = Settings(_env_file=None, app_secret_key="x" * 32)
|
|
monkeypatch.setattr(cli, "get_settings", lambda: settings)
|
|
monkeypatch.setattr(cli, "get_session_factory", lambda: lambda: nullcontext(object()))
|
|
|
|
def finish(_session, **_kwargs):
|
|
marker.touch()
|
|
return "completed"
|
|
|
|
consume = Mock(side_effect=finish)
|
|
pulse = Mock()
|
|
monkeypatch.setattr(cli, "run_once", consume)
|
|
monkeypatch.setattr(cli, "get_batch", lambda *_: SimpleNamespace(status="SUCCEEDED"))
|
|
monkeypatch.setattr("app.services.tplus.automation.heartbeat", pulse)
|
|
result = cli._tplus_worker(Namespace(connection_file=None, poll_seconds=2, watch=True, stop_file=str(marker)))
|
|
assert result == 0
|
|
assert consume.call_count == (0 if already_stopped else 1)
|
|
assert pulse.call_args.args[3] == "STOPPED"
|