mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-13 08:15:21 +02:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from api.routes import organization_usage
|
|
|
|
|
|
def test_is_mps_billing_v2_depends_only_on_account_mode():
|
|
assert organization_usage._is_mps_billing_v2({"billing_mode": "v2"}) is True
|
|
assert organization_usage._is_mps_billing_v2({"billing_mode": "v1"}) is False
|
|
assert organization_usage._is_mps_billing_v2({"billing_mode": "shadow"}) is False
|
|
assert organization_usage._is_mps_billing_v2(None) is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_mps_billing_account_status_uses_user_provider_id(monkeypatch):
|
|
get_status = AsyncMock(return_value={"billing_mode": "v2"})
|
|
monkeypatch.setattr(
|
|
organization_usage.mps_service_key_client,
|
|
"get_billing_account_status",
|
|
get_status,
|
|
)
|
|
|
|
user = SimpleNamespace(provider_id="provider-123")
|
|
|
|
assert await organization_usage._get_mps_billing_account_status(user, 42) == {
|
|
"billing_mode": "v2"
|
|
}
|
|
get_status.assert_awaited_once_with(
|
|
organization_id=42,
|
|
created_by="provider-123",
|
|
)
|