Files
office_translator/tests/test_plan_gating.py
sepehr 67365918ae
All checks were successful
Deploy to Production / Build and Deploy (push) Successful in 3m36s
feat(ui,api): wave 3 — editorial pricing, real cancel, server history, DeepL purge
Pricing: full editorial redesign — serif card headers with accent pills
replace the colored font-black blocks, tone sweep across toggle/metrics/
features/CTAs, PLAN_COLORS removed; one design system app-wide.

Translate: decorative titles one step down (CTA hierarchy restored);
glossary and image-translation blocks hidden entirely for free users
(progressive disclosure — three controls for free).

Reviews: XLIFF hint line explains the exchange format; backend errors
routed through a friendly mapper (session/not-found/rate-limit/server).

Landing: fabricated hero UI cards (fake 'Context Engine' overlay)
removed — the photo no longer promises screens that don't exist.

Nav: single DashboardNavLinks component shared by sidebar and mobile
drawer (was duplicated markup).

API: GET /api/v1/translations (user job history, paginated; completed
jobs retained 24h) and POST /api/v1/translations/{id}/cancel —
cooperative cancellation with worker checkpoints before dispatch and
before finalisation, reserved quota released immediately. Translate
monitor now offers a real 'Cancel translation' next to 'Back to start';
recent-jobs list reads server history first, localStorage fallback.

DeepL purge (backend): provider module, registry registration, config
attrs/defaults, dispatch branch, admin settings schema + test branch,
legacy availability block, validation rules, plan provider lists,
error-code mappings, MCP enums, translator prompt mention, related
tests updated/removed. Fallback resolver skips unknown providers, so
stale chains containing 'deepl' degrade gracefully.

Verified: backend 110 tests passed; frontend build exit 0, vitest 9/9,
0 missing i18n keys, eslint 63 errors (vs 64 at HEAD).
2026-08-30 22:42:29 +02:00

87 lines
2.9 KiB
Python

"""Plan-based engine gating and the expanded /languages endpoint."""
import pytest
from models.subscription import PlanType
from routes.translate_routes import (
_allowed_providers_for_plan,
_image_translation_allowed_for_plan,
_plan_from_user,
)
class _FakeUser:
def __init__(self, plan):
self.plan = plan
class TestAllowedProviders:
def test_free_gets_google_only(self):
assert _allowed_providers_for_plan(PlanType.FREE) == {"google"}
def test_starter_engines(self):
assert _allowed_providers_for_plan(PlanType.STARTER) == {"google"}
def test_pro_adds_cloud_and_openrouter(self):
allowed = _allowed_providers_for_plan(PlanType.PRO)
assert {"google_cloud", "openrouter"} <= allowed
assert "openai" not in allowed
def test_business_adds_premium_and_xai(self):
allowed = _allowed_providers_for_plan(PlanType.BUSINESS)
assert {"openrouter_premium", "openai", "zai"} <= allowed
def test_enterprise_has_all(self):
enterprise = _allowed_providers_for_plan(PlanType.ENTERPRISE)
assert enterprise >= _allowed_providers_for_plan(PlanType.BUSINESS)
class TestImageTranslationGate:
@pytest.mark.parametrize("plan", [PlanType.FREE, PlanType.STARTER])
def test_refused_below_pro(self, plan):
assert _image_translation_allowed_for_plan(plan) is False
@pytest.mark.parametrize("plan", [PlanType.PRO, PlanType.BUSINESS, PlanType.ENTERPRISE])
def test_allowed_from_pro(self, plan):
assert _image_translation_allowed_for_plan(plan) is True
class TestPlanFromUser:
def test_anonymous_is_free(self):
assert _plan_from_user(None) is PlanType.FREE
def test_enum_plan_passthrough(self):
assert _plan_from_user(_FakeUser(PlanType.PRO)) is PlanType.PRO
def test_string_plan_accepted(self):
assert _plan_from_user(_FakeUser("pro")) is PlanType.PRO
def test_garbage_falls_back_to_free(self):
assert _plan_from_user(_FakeUser("nonsense")) is PlanType.FREE
class TestLanguagesEndpoint:
@pytest.mark.asyncio
async def test_exposes_at_least_60_languages(self):
from routes.legacy_routes import get_supported_languages
response = await get_supported_languages()
langs = response["supported_languages"]
assert response["count"] >= 60
assert len(langs) >= 60
@pytest.mark.asyncio
async def test_no_auto_and_canonical_chinese(self):
from routes.legacy_routes import get_supported_languages
langs = (await get_supported_languages())["supported_languages"]
assert "auto" not in langs
assert "zh-CN" in langs and "zh-TW" in langs
@pytest.mark.asyncio
async def test_every_language_has_a_name(self):
from routes.legacy_routes import get_supported_languages
langs = (await get_supported_languages())["supported_languages"]
assert all(name and name != code.upper() for code, name in langs.items())