81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
|
"""真实分页进度在每页提交后可读；总量未知或失败时不伪造成功。"""
|
||
|
|
from datetime import datetime
|
||
|
|
from unittest.mock import Mock
|
||
|
|
|
||
|
|
from sqlalchemy import select
|
||
|
|
|
||
|
|
from app.models import TPlusSyncBatch
|
||
|
|
from app.services.tplus import progress
|
||
|
|
from app.services.tplus.sync_service import TPlusSyncService
|
||
|
|
from tests.unit.test_tplus_sync import target as target
|
||
|
|
|
||
|
|
|
||
|
|
def test_每页提交真实已读数及类别(target):
|
||
|
|
service = TPlusSyncService(Mock(), target)
|
||
|
|
service.reader = Mock()
|
||
|
|
service.reader.count_entities.return_value = {"products": 2}
|
||
|
|
|
||
|
|
def rows(*args, **kwargs):
|
||
|
|
yield {"id": 1}
|
||
|
|
batch = target.scalar(select(TPlusSyncBatch))
|
||
|
|
item = progress.public(batch)
|
||
|
|
assert item["processed_records"] == 1 and item["total_records"] == 2
|
||
|
|
assert item["percentage"] == 50.0 and item["active_entity"] == "products"
|
||
|
|
yield {"id": 2}
|
||
|
|
|
||
|
|
service.reader.iter_entity.side_effect = rows
|
||
|
|
batch = service.run(entity_types=["products"], batch_size=1)
|
||
|
|
item = progress.public(batch)
|
||
|
|
assert batch.status == "SUCCEEDED" and item["percentage"] == 100.0
|
||
|
|
assert item["entities"][0]["status"] == "COMPLETED"
|
||
|
|
|
||
|
|
|
||
|
|
def test_总量统计失败仍可读且执行中总量未知(target):
|
||
|
|
source = Mock()
|
||
|
|
service = TPlusSyncService(source, target)
|
||
|
|
service.reader = Mock()
|
||
|
|
service.reader.count_entities.side_effect = RuntimeError("不得公开的数据库详情")
|
||
|
|
|
||
|
|
def rows(*args, **kwargs):
|
||
|
|
batch = target.scalar(select(TPlusSyncBatch))
|
||
|
|
assert progress.public(batch)["percentage"] is None
|
||
|
|
yield {"id": 1}
|
||
|
|
|
||
|
|
service.reader.iter_entity.side_effect = rows
|
||
|
|
batch = service.run(entity_types=["products"])
|
||
|
|
assert batch.status == "SUCCEEDED"
|
||
|
|
source.rollback.assert_called_once()
|
||
|
|
|
||
|
|
|
||
|
|
def test_读取中断保留进度及错误终态(target):
|
||
|
|
service = TPlusSyncService(Mock(), target)
|
||
|
|
service.reader = Mock()
|
||
|
|
service.reader.count_entities.return_value = {"products": 4}
|
||
|
|
|
||
|
|
def rows(*args, **kwargs):
|
||
|
|
yield {"id": 1}
|
||
|
|
raise TimeoutError("敏感详情")
|
||
|
|
|
||
|
|
service.reader.iter_entity.side_effect = rows
|
||
|
|
batch = service.run(entity_types=["products"], batch_size=1)
|
||
|
|
item = progress.public(batch)
|
||
|
|
assert batch.status == "FAILED" and item["percentage"] == 25.0
|
||
|
|
assert item["phase"] == "FAILED" and item["entities"][0]["status"] == "FAILED"
|
||
|
|
|
||
|
|
|
||
|
|
def test_旧批次未知总量不显示编造百分比():
|
||
|
|
batch = TPlusSyncBatch(batch_no="old", sync_mode="FULL", status="RUNNING",
|
||
|
|
source_record_count=1000, entity_types_json=["personnel"], started_at=datetime.now())
|
||
|
|
item = progress.public(batch)
|
||
|
|
assert item["percentage"] is None and item["total_records"] is None
|
||
|
|
assert item["processed_records"] == 1000
|
||
|
|
|
||
|
|
|
||
|
|
def test_读到预统计总量仍等终态才显示百分之百():
|
||
|
|
batch = TPlusSyncBatch(batch_no="b", sync_mode="FULL", status="RUNNING",
|
||
|
|
source_record_count=10, entity_types_json=["products"], started_at=datetime.now(),
|
||
|
|
audit_result_json={"progress": {"total_records": 10}})
|
||
|
|
assert progress.public(batch)["percentage"] == 99.9
|
||
|
|
batch.status = "SUCCEEDED"
|
||
|
|
assert progress.public(batch)["percentage"] == 100.0
|