220 lines
9.8 KiB
Python
220 lines
9.8 KiB
Python
import asyncio
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from aiohttp import web
|
|
from aiohttp.test_utils import make_mocked_request
|
|
|
|
from bot.app.web import admin_api, subscription_webapp
|
|
from bot.app.web.admin_api_impl import auth as admin_auth_routes
|
|
from bot.app.web.webapp_auth import create_webapp_session_token
|
|
|
|
|
|
class _Request(dict):
|
|
def __init__(self, *, path="/", app=None, headers=None, cookies=None):
|
|
super().__init__()
|
|
self.path = path
|
|
self.app = app or {}
|
|
self.headers = headers or {}
|
|
self.cookies = cookies or {}
|
|
|
|
|
|
class _AsyncSessionFactory:
|
|
def __call__(self):
|
|
return self
|
|
|
|
async def __aenter__(self):
|
|
return object()
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
return False
|
|
|
|
|
|
def _route_map(app: web.Application) -> dict[tuple[str, str], str]:
|
|
return {
|
|
(route.method, route.resource.canonical): route.handler.__name__
|
|
for route in app.router.routes()
|
|
if route.method != "HEAD"
|
|
}
|
|
|
|
|
|
class WebAppRouteContractTests(unittest.TestCase):
|
|
def test_subscription_webapp_registers_public_and_api_routes(self):
|
|
app = web.Application()
|
|
|
|
subscription_webapp.setup_subscription_webapp_routes(app)
|
|
|
|
routes = _route_map(app)
|
|
expected = {
|
|
("GET", "/"): "index_route",
|
|
("GET", "/home"): "index_route",
|
|
("GET", "/invite"): "index_route",
|
|
("GET", "/devices"): "index_route",
|
|
("GET", "/settings"): "index_route",
|
|
("GET", "/admin"): "index_route",
|
|
("GET", "/admin/{section}"): "index_route",
|
|
("GET", "/admin/users/{user_id}"): "index_route",
|
|
("GET", "/auth/telegram/start"): "telegram_oauth_start_route",
|
|
("GET", "/auth/telegram/callback"): "telegram_oauth_callback_route",
|
|
("GET", "/health"): "health_route",
|
|
("GET", "/webapp-logo"): "webapp_logo_route",
|
|
("GET", "/webapp-uploaded-logo/{filename}"): "webapp_uploaded_logo_route",
|
|
("GET", "/webapp-emoji/{codepoints}/512.{ext}"): "webapp_animated_emoji_route",
|
|
("GET", "/subscription_webapp.css"): "css_asset_route",
|
|
("GET", "/webapp-theme-css/{path}"): "theme_css_asset_route",
|
|
("GET", "/subscription_webapp.min.{asset_hash}.js"): "js_asset_route",
|
|
("GET", "/subscription_webapp.js"): "js_asset_route",
|
|
("POST", "/api/auth/telegram/nonce"): "telegram_oauth_nonce_route",
|
|
("POST", "/api/auth/token"): "auth_token_route",
|
|
("POST", "/api/auth/email/request"): "email_auth_request_route",
|
|
("POST", "/api/auth/email/verify"): "email_auth_verify_route",
|
|
("POST", "/api/auth/email/magic"): "email_auth_magic_route",
|
|
("POST", "/api/auth/logout"): "logout_route",
|
|
("GET", "/api/me"): "me_route",
|
|
("GET", "/api/account/avatar"): "account_avatar_route",
|
|
("POST", "/api/account/language"): "account_language_route",
|
|
("POST", "/api/account/email/request"): "account_email_request_route",
|
|
("POST", "/api/account/email/verify"): "account_email_verify_route",
|
|
("POST", "/api/account/telegram/link"): "account_telegram_link_route",
|
|
("POST", "/api/promo/apply"): "apply_promo_route",
|
|
("POST", "/api/trial/activate"): "activate_trial_route",
|
|
("GET", "/api/devices"): "devices_route",
|
|
("POST", "/api/devices/disconnect"): "disconnect_device_route",
|
|
("GET", "/api/devices/topup-options"): "device_topup_options_route",
|
|
("GET", "/api/tariffs/topup-options"): "tariff_topup_options_route",
|
|
("GET", "/api/tariffs/change-options"): "tariff_change_options_route",
|
|
("POST", "/api/tariffs/change"): "tariff_change_route",
|
|
("POST", "/api/tariffs/change-payment"): "tariff_change_payment_route",
|
|
("POST", "/api/payments"): "create_payment_route",
|
|
("GET", "/api/payments/{payment_id}"): "payment_status_route",
|
|
}
|
|
|
|
for key, handler_name in expected.items():
|
|
self.assertEqual(routes.get(key), handler_name, key)
|
|
|
|
def test_admin_api_registers_expected_routes(self):
|
|
app = web.Application()
|
|
|
|
admin_api.setup_admin_routes(app)
|
|
|
|
routes = _route_map(app)
|
|
expected = {
|
|
("GET", "/api/admin/me"): "admin_me_route",
|
|
("GET", "/api/admin/stats"): "admin_stats_route",
|
|
("GET", "/api/admin/users"): "admin_users_list_route",
|
|
("GET", "/api/admin/users/{user_id}"): "admin_user_detail_route",
|
|
("GET", "/api/admin/users/{user_id}/avatar"): "admin_user_avatar_route",
|
|
("POST", "/api/admin/users/{user_id}/ban"): "admin_user_ban_route",
|
|
("POST", "/api/admin/users/{user_id}/message"): "admin_user_message_route",
|
|
(
|
|
"POST",
|
|
"/api/admin/users/{user_id}/message/preview",
|
|
): "admin_user_message_preview_route",
|
|
("POST", "/api/admin/users/{user_id}/reset-trial"): "admin_user_reset_trial_route",
|
|
("POST", "/api/admin/users/{user_id}/extend"): "admin_user_extend_route",
|
|
(
|
|
"POST",
|
|
"/api/admin/users/{user_id}/premium-override",
|
|
): "admin_user_premium_override_route",
|
|
(
|
|
"POST",
|
|
"/api/admin/users/{user_id}/regular-traffic-override",
|
|
): "admin_user_regular_traffic_override_route",
|
|
("POST", "/api/admin/users/{user_id}/traffic-grant"): "admin_user_traffic_grant_route",
|
|
("DELETE", "/api/admin/users/{user_id}"): "admin_user_delete_route",
|
|
("GET", "/api/admin/payments"): "admin_payments_list_route",
|
|
("GET", "/api/admin/payments/export.csv"): "admin_payments_export_route",
|
|
("GET", "/api/admin/promos"): "admin_promos_list_route",
|
|
("POST", "/api/admin/promos"): "admin_promo_create_route",
|
|
("PATCH", "/api/admin/promos/{promo_id}"): "admin_promo_update_route",
|
|
("DELETE", "/api/admin/promos/{promo_id}"): "admin_promo_delete_route",
|
|
("GET", "/api/admin/logs"): "admin_logs_route",
|
|
("POST", "/api/admin/broadcast"): "admin_broadcast_route",
|
|
("POST", "/api/admin/sync"): "admin_sync_route",
|
|
("GET", "/api/admin/ads"): "admin_ads_list_route",
|
|
("POST", "/api/admin/ads"): "admin_ad_create_route",
|
|
("POST", "/api/admin/ads/{campaign_id}/toggle"): "admin_ad_toggle_route",
|
|
("DELETE", "/api/admin/ads/{campaign_id}"): "admin_ad_delete_route",
|
|
("GET", "/api/admin/settings"): "admin_settings_get_route",
|
|
("PATCH", "/api/admin/settings"): "admin_settings_patch_route",
|
|
("GET", "/api/admin/tariffs"): "admin_tariffs_get_route",
|
|
("PUT", "/api/admin/tariffs"): "admin_tariffs_save_route",
|
|
("GET", "/api/admin/themes"): "admin_themes_get_route",
|
|
("PUT", "/api/admin/themes"): "admin_themes_save_route",
|
|
("POST", "/api/admin/appearance/logo"): "admin_appearance_logo_upload_route",
|
|
("GET", "/api/admin/panel/internal-squads"): "admin_panel_internal_squads_route",
|
|
}
|
|
|
|
for key, handler_name in expected.items():
|
|
self.assertEqual(routes.get(key), handler_name, key)
|
|
|
|
def test_admin_themes_page_route_is_not_registered(self):
|
|
app = web.Application()
|
|
subscription_webapp.setup_subscription_webapp_routes(app)
|
|
|
|
request = make_mocked_request("GET", "/admin/themes", app=app)
|
|
match_info = asyncio.run(app.router.resolve(request))
|
|
|
|
self.assertEqual(match_info.http_exception.status, 404)
|
|
|
|
def test_admin_appearance_page_route_is_registered(self):
|
|
app = web.Application()
|
|
subscription_webapp.setup_subscription_webapp_routes(app)
|
|
|
|
request = make_mocked_request("GET", "/admin/appearance", app=app)
|
|
match_info = asyncio.run(app.router.resolve(request))
|
|
|
|
self.assertEqual(match_info.handler.__name__, "index_route")
|
|
|
|
|
|
class AdminApiAuthContractTests(unittest.IsolatedAsyncioTestCase):
|
|
def _settings(self):
|
|
return SimpleNamespace(
|
|
ADMIN_IDS=[999],
|
|
WEBAPP_SESSION_SECRET="session-secret",
|
|
WEBAPP_SESSION_TTL_SECONDS=3600,
|
|
)
|
|
|
|
async def test_admin_auth_middleware_resolves_telegram_id_from_webapp_session(self):
|
|
settings = self._settings()
|
|
token = create_webapp_session_token(settings, 42)
|
|
request = _Request(
|
|
path="/api/admin/me",
|
|
app={
|
|
"settings": settings,
|
|
"async_session_factory": _AsyncSessionFactory(),
|
|
},
|
|
headers={},
|
|
cookies={"rw_webapp_session": token},
|
|
)
|
|
handler = AsyncMock(return_value=web.json_response({"ok": True}))
|
|
db_user = SimpleNamespace(user_id=42, telegram_id=999)
|
|
|
|
with patch.object(
|
|
admin_auth_routes.user_dal,
|
|
"get_user_by_id",
|
|
AsyncMock(return_value=db_user),
|
|
):
|
|
response = await admin_api.admin_auth_middleware(request, handler)
|
|
|
|
self.assertEqual(response.status, 200)
|
|
self.assertEqual(request["admin_telegram_id"], 999)
|
|
handler.assert_awaited_once_with(request)
|
|
|
|
async def test_admin_route_rejects_authenticated_non_admin(self):
|
|
settings = self._settings()
|
|
token = create_webapp_session_token(settings, 42)
|
|
request = _Request(
|
|
path="/api/admin/me",
|
|
app={"settings": settings},
|
|
headers={},
|
|
cookies={"rw_webapp_session": token},
|
|
)
|
|
request["admin_telegram_id"] = 123
|
|
|
|
with self.assertRaises(web.HTTPForbidden) as raised:
|
|
await admin_api.admin_me_route(request)
|
|
|
|
self.assertEqual(raised.exception.status, 403)
|