mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-07 01:47:47 +08:00
fawney19andLewisPen
04ab4bd9f2
- 将 usage.billing_status 默认值从 settled 改为 pending，完善 pending -> settled/void 的状态转换逻辑，确保终态不可逆 - 新增 WalletDailyUsageLedger 模型和聚合服务，按账单日汇总 每个钱包的消费金额、请求数和 token 用量 - 前端钱包中心页面集成每日消费流水展示，支持与充值记录混合 排序和分页 - 新增两个数据库迁移：修复历史数据状态一致性、创建每日汇总表 - 补充计费状态机单元测试 Closes #218 Co-authored-by: LewisPen <[email protected]>
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
from types import SimpleNamespace
|
|
from typing import cast
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from src.core.api_format.conversion.internal_video import InternalVideoPollResult, VideoStatus
|
|
from src.models.database import VideoTask
|
|
from src.services.task.video.poller_adapter import VideoTaskPollerAdapter
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_poll_task_status_routes_gemini_video_to_gemini(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
adapter = VideoTaskPollerAdapter()
|
|
|
|
task = cast(
|
|
VideoTask,
|
|
SimpleNamespace(
|
|
endpoint_id="e1",
|
|
key_id="k1",
|
|
provider_api_format="gemini:video",
|
|
external_task_id="operations/123",
|
|
),
|
|
)
|
|
endpoint = SimpleNamespace(id="e1", base_url="https://example.com", api_format="gemini:video")
|
|
key = SimpleNamespace(id="k1", api_key="enc")
|
|
|
|
monkeypatch.setattr(adapter, "_get_endpoint", lambda _db, _id: endpoint)
|
|
monkeypatch.setattr(adapter, "_get_key", lambda _db, _id: key)
|
|
monkeypatch.setattr(
|
|
"src.services.task.video.poller_adapter.crypto_service.decrypt", lambda _v: "decrypted"
|
|
)
|
|
|
|
auth_info = SimpleNamespace(auth_header="authorization", auth_value="Bearer x")
|
|
monkeypatch.setattr(
|
|
"src.services.task.video.poller_adapter.get_provider_auth",
|
|
AsyncMock(return_value=auth_info),
|
|
)
|
|
|
|
poll_gemini = AsyncMock(return_value=InternalVideoPollResult(status=VideoStatus.PROCESSING))
|
|
poll_openai = AsyncMock(return_value=InternalVideoPollResult(status=VideoStatus.PROCESSING))
|
|
monkeypatch.setattr(adapter, "_poll_gemini", poll_gemini)
|
|
monkeypatch.setattr(adapter, "_poll_openai", poll_openai)
|
|
|
|
result = await adapter._poll_task_status(MagicMock(), task)
|
|
assert result.status == VideoStatus.PROCESSING
|
|
assert poll_gemini.await_count == 1
|
|
assert poll_openai.await_count == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_after_poll_skips_terminal_cancelled_task(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
finalize = AsyncMock()
|
|
adapter = VideoTaskPollerAdapter(finalize_video_task_fn=finalize)
|
|
|
|
cancelled_task = SimpleNamespace(
|
|
id="t1",
|
|
status=VideoStatus.CANCELLED.value,
|
|
)
|
|
|
|
session = MagicMock()
|
|
session.__enter__.return_value = session
|
|
session.__exit__.return_value = None
|
|
session.get.return_value = cancelled_task
|
|
|
|
monkeypatch.setattr("src.services.task.video.poller_adapter.create_session", lambda: session)
|
|
|
|
await adapter.update_task_after_poll(
|
|
task_id="t1",
|
|
result=InternalVideoPollResult(status=VideoStatus.COMPLETED),
|
|
ctx=None,
|
|
redis_client=None,
|
|
)
|
|
|
|
finalize.assert_not_awaited()
|
|
session.commit.assert_not_called()
|