feat: implement Supabase entitlement service for signup trials and subscription management

This commit is contained in:
2569718930@qq.com
2026-04-19 20:19:38 +08:00
parent 6914e6277b
commit f5fe8b9b95
3 changed files with 59 additions and 12 deletions
+40
View File
@@ -100,3 +100,43 @@ def test_latest_active_subscription_ignores_future_start(monkeypatch):
assert result is not None
assert result["plan_code"] == "signup_trial_3d"
def test_subscription_window_keeps_queued_renewal_after_current_cache(monkeypatch):
monkeypatch.setenv("SUPABASE_URL", "https://example.supabase.co")
monkeypatch.setenv("SUPABASE_ANON_KEY", "anon-key")
monkeypatch.setenv("SUPABASE_SERVICE_ROLE_KEY", "service-role")
service = SupabaseEntitlementService()
now = datetime.now(timezone.utc)
current = {
"id": 1,
"user_id": "user-1",
"status": "active",
"plan_code": "pro_monthly",
"starts_at": (now - timedelta(days=29)).isoformat(),
"expires_at": (now + timedelta(days=1)).isoformat(),
}
queued = {
"id": 2,
"user_id": "user-1",
"status": "active",
"plan_code": "pro_monthly",
"starts_at": (now + timedelta(days=1)).isoformat(),
"expires_at": (now + timedelta(days=31)).isoformat(),
}
def _fake_get(url, headers=None, params=None, timeout=None):
return _Response(200, [queued, current])
monkeypatch.setattr(entitlement_module.requests, "get", _fake_get)
assert service._query_latest_active_subscription("user-1") == current
window = service.get_subscription_window("user-1", respect_requirement=False)
assert window["current"] == current
assert window["total_expires_at"] == queued["expires_at"]
assert window["queued_days"] == 30
assert window["queued_count"] == 1