geMoldInsight/tests/test_redis_no_fallback.py

"""批次 2（D7 / D9）回归测试。

- D7：Redis 任务管理器不再有进程内存回退——Redis 不可用时写 no-op、读返回 None，
  状态查询路径落到 PG（PG 为单一事实源）
- D9：存储服务的数据本体写方法只 flush 不 commit，事务由编排层收口
"""
import pytest
from unittest.mock import AsyncMock, MagicMock

from shared.services.redis_task_manager import RedisTaskManager


@pytest.mark.asyncio
async def test_no_memory_fallback_when_disconnected():
    """Redis 未连接：写 no-op、读 None——不得再出现进程内可见的副本。"""
    mgr = RedisTaskManager()  # 不 connect
    assert not mgr.is_connected

    await mgr.set_task("t1", {"status": "processing"})
    assert await mgr.get_task("t1") is None

    await mgr.update_task("t1", {"status": "completed"})  # no-op，不得抛异常

    assert await mgr.get_all_tasks() == {}
    assert await mgr.get_task_count() == 0
    await mgr.delete_task("t1")  # no-op，不得抛异常

    assert await mgr.get_task("t1") is None


def test_fallback_storage_removed():
    """防回归：内存回退存储必须已删除，防止静默回归。"""
    assert not hasattr(RedisTaskManager, "_fallback_tasks")
    assert not hasattr(RedisTaskManager, "_fallback_set")
    assert not hasattr(RedisTaskManager, "cleanup_old_tasks")


@pytest.mark.asyncio
async def test_storage_writes_flush_but_never_commit():
    """D9：数据本体写方法仅 flush；commit 由编排层/请求侧负责。"""
    pytest.importorskip("minio")
    from sqlalchemy.ext.asyncio import AsyncSession
    from moldinsight.services.task_storage_service import TaskStorageService

    svc = TaskStorageService()
    session = AsyncMock(spec=AsyncSession)
    # update_task_parameters：select 返回 None（任务不存在）→ 直接 return
    result_mock = MagicMock()
    result_mock.scalar_one_or_none.return_value = None
    session.execute.return_value = result_mock

    await svc.update_task_parameters(session, "task-x", {"a": 1})
    await svc.create_processing_task(session, "task-y", stp_file_id=1, batch_id="b-1")

    assert session.flush.await_count >= 1
    # 注意：不能用 .awaited（AsyncMock 上访问会自动创建 truthy 子 mock），用 await_count
    assert session.commit.await_count == 0, "存储写方法不得自行 commit（D9 事务收口）"