95 lines
4.0 KiB
Python
95 lines
4.0 KiB
Python
|
"""自动采集保存值由独立维护读取，关闭不破坏正在运行的任务。"""
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from app.core.config import Settings
|
||
|
|
from app.core.exceptions import ConflictError
|
||
|
|
from app.models import OperationLog, TPlusAutomationConfig, TPlusSyncBatch, User
|
||
|
|
from app.schemas.tplus import TPlusAutomationUpdate
|
||
|
|
from app.services.tplus import automation, automation_config, task_service
|
||
|
|
from tests.unit.test_tplus_workbench import work_session as work_session
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def source_settings(monkeypatch):
|
||
|
|
settings = Settings(_env_file=None, app_secret_key="x" * 32, t_plus_enabled=True,
|
||
|
|
t_plus_database="test_source", t_plus_user="reader", t_plus_odbc_driver="unit",
|
||
|
|
t_plus_automation_enabled=True)
|
||
|
|
monkeypatch.setattr(task_service, "get_settings", lambda: settings)
|
||
|
|
return settings
|
||
|
|
|
||
|
|
|
||
|
|
def update(s, enabled, version=0):
|
||
|
|
actor = task_service.TaskActor(s.get(User, 1))
|
||
|
|
return automation_config.save(s, TPlusAutomationUpdate(
|
||
|
|
enabled=enabled, expected_version=version), actor)
|
||
|
|
|
||
|
|
|
||
|
|
def test_初次沿用环境且页面关闭后独立维护不再排队(work_session, source_settings):
|
||
|
|
s = work_session
|
||
|
|
initial = automation_config.configuration(s, source_settings)
|
||
|
|
assert initial == {"enabled": True, "version": 0, "editable": True,
|
||
|
|
"configuration_origin": "ENVIRONMENT"}
|
||
|
|
result = update(s, False)
|
||
|
|
assert result["enabled"] is False and result["version"] == 1
|
||
|
|
with Session(s.get_bind()) as other:
|
||
|
|
assert automation.tick(other, settings=source_settings) == {"status": "DISABLED", "queued": False}
|
||
|
|
assert s.scalar(select(TPlusSyncBatch)) is None
|
||
|
|
assert s.scalar(select(OperationLog).where(OperationLog.action == "修改自动采集")) is not None
|
||
|
|
|
||
|
|
|
||
|
|
def test_保存开启覆盖环境关闭且同周不重复(work_session, source_settings):
|
||
|
|
settings = source_settings.model_copy(update={"t_plus_automation_enabled": False})
|
||
|
|
update(work_session, True)
|
||
|
|
with Session(work_session.get_bind()) as other:
|
||
|
|
first = automation.tick(other, settings=settings, now=datetime(2026, 11, 2, 3))
|
||
|
|
assert first["queued"]
|
||
|
|
assert not automation.tick(other, settings=settings, now=datetime(2026, 11, 2, 4))["queued"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_关闭不终止活动批次且仍可手动排队(work_session, source_settings):
|
||
|
|
s = work_session
|
||
|
|
s.add(TPlusSyncBatch(batch_no="running", source_database="test_source", sync_mode="FULL",
|
||
|
|
status="RUNNING", entity_types_json=["personnel"]))
|
||
|
|
s.commit()
|
||
|
|
update(s, False)
|
||
|
|
assert s.scalar(select(TPlusSyncBatch.status)) == "RUNNING"
|
||
|
|
assert automation.tick(s, settings=source_settings)["status"] == "DISABLED"
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
from app.schemas.tplus import TPlusCreateIn
|
||
|
|
|
||
|
|
queued = task_service.create_task(s, TPlusCreateIn(request_key=uuid4(), entity_types=["personnel"]))
|
||
|
|
assert queued.status == "QUEUED"
|
||
|
|
|
||
|
|
|
||
|
|
def test_旧版本不能覆盖新的开关(work_session, source_settings):
|
||
|
|
s = work_session
|
||
|
|
update(s, False)
|
||
|
|
with pytest.raises(ConflictError):
|
||
|
|
update(s, True, 0)
|
||
|
|
assert not bool(s.scalar(select(TPlusAutomationConfig.enabled)))
|
||
|
|
result = update(s, True, 1)
|
||
|
|
assert result["enabled"] and result["version"] == 2
|
||
|
|
|
||
|
|
|
||
|
|
def test_不同源的开关不互相覆盖(work_session, source_settings):
|
||
|
|
update(work_session, False)
|
||
|
|
other = source_settings.model_copy(update={"t_plus_database": "other_source"})
|
||
|
|
assert automation_config.configuration(work_session, other)["enabled"] is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_没有采集权限不能保存(work_session, source_settings, monkeypatch):
|
||
|
|
from app.core.exceptions import ForbiddenError
|
||
|
|
|
||
|
|
def denied(*args):
|
||
|
|
raise ForbiddenError()
|
||
|
|
|
||
|
|
monkeypatch.setattr(automation_config, "require_permission", denied)
|
||
|
|
with pytest.raises(ForbiddenError):
|
||
|
|
update(work_session, False)
|
||
|
|
assert work_session.scalar(select(TPlusAutomationConfig)) is None
|