gavindiaz/PolyWeather
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: implement Telegram-to-Web account binding flow and add supporting database and UI components

Browse Source
This commit is contained in:
2569718930@qq.com
2026-05-19 18:26:45 +08:00
parent a5c508cce8
commit 82d60cf05f
6 changed files with 423 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_bot_basic_handler.py
+133 -5
View File
@@ -10,14 +10,16 @@ class DummyBot:
self.sent_messages = []
self.approved_join_requests = []
self.declined_join_requests = []
self.callback_handlers = []
def reply_to(self, message, text, parse_mode=None, disable_web_page_preview=None):
def reply_to(self, message, text, parse_mode=None, disable_web_page_preview=None, **kwargs):
self.replies.append(
{
"text": text,
"parse_mode": parse_mode,
"chat_id": message.chat.id,
"disable_web_page_preview": disable_web_page_preview,
"reply_markup": kwargs.get("reply_markup"),
}
)
@@ -50,6 +52,13 @@ class DummyBot:
return _decorator
def callback_query_handler(self, *args, **kwargs): # pragma: no cover - decorator stub
def _decorator(func):
self.callback_handlers.append((kwargs.get("func"), func))
return func
return _decorator
def approve_chat_join_request(self, chat_id, user_id):
self.approved_join_requests.append({"chat_id": chat_id, "user_id": user_id})
@@ -94,14 +103,12 @@ def test_basic_handler_diag_returns_html():
def test_start_bind_token_binds_telegram_to_web_account():
bot = DummyBot()
db = SimpleNamespace(
consume_web_bind_token=lambda token: {
peek_web_bind_token=lambda token: {
"supabase_user_id": "user-1",
"supabase_email": "u@example.com",
}
if token == "abc123"
else None,
upsert_user=lambda *_args, **_kwargs: None,
bind_supabase_identity=lambda **_kwargs: {"ok": True, "reason": "bound"},
)
io_layer = SimpleNamespace(
build_welcome_text=lambda: "WELCOME",
@@ -124,7 +131,54 @@ def test_start_bind_token_binds_telegram_to_web_account():
handler.handle_start_help(_message("/start bind_abc123"))
assert len(bot.replies) == 1
assert "账号绑定完成" in bot.replies[0]["text"]
assert "确认绑定" in bot.replies[0]["text"]
assert "u***@example.com" in bot.replies[0]["text"]
def test_confirm_bind_callback_consumes_token_and_binds_account():
bot = DummyBot()
consumed = []
bound = []
def _consume(token):
consumed.append(token)
return {"supabase_user_id": "user-1", "supabase_email": "u@example.com"}
db = SimpleNamespace(
consume_web_bind_token=_consume,
upsert_user=lambda *_args, **_kwargs: None,
bind_supabase_identity=lambda **kwargs: bound.append(kwargs)
or {"ok": True, "reason": "bound"},
)
io_layer = SimpleNamespace(
build_welcome_text=lambda: "WELCOME",
build_points_rank_text=lambda _user: "TOP",
display_name=lambda user: user.username,
db=db,
)
handler = BasicCommandHandler(
bot=bot,
io_layer=io_layer,
runtime_status_provider=lambda: RuntimeStatus(
started_at="2026-03-12 00:00:00 UTC",
loops=[],
command_access_mode="group_member",
protected_commands=["/city", "/deb"],
required_group_chat_id="-1001234567890",
),
)
call = SimpleNamespace(
data="confirm_bind:abc123",
from_user=SimpleNamespace(id=12345, username="ada", first_name="Ada"),
message=_message("callback"),
)
result = handler.handle_bind_confirm_callback(call)
assert result == "bound"
assert consumed == ["abc123"]
assert bound[0]["telegram_id"] == 12345
assert bound[0]["supabase_user_id"] == "user-1"
def test_basic_handler_markets_returns_summary():
@@ -253,6 +307,80 @@ def test_join_request_keeps_unbound_user_pending_by_default(monkeypatch):
assert bot.declined_join_requests == []
def test_join_request_keeps_trial_user_pending(monkeypatch):
monkeypatch.setenv("POLYWEATHER_TELEGRAM_GROUP_ID", "-100123")
bot = DummyBot()
db = SimpleNamespace(list_supabase_user_ids_for_telegram=lambda telegram_id: ["user-1"])
io_layer = SimpleNamespace(
build_welcome_text=lambda: "WELCOME",
build_points_rank_text=lambda _user: "TOP",
db=db,
)
entitlement = SimpleNamespace(
get_latest_active_subscription=lambda user_id, respect_requirement=False: {
"plan_code": "signup_trial_3d",
"source": "signup_trial",
}
)
handler = BasicCommandHandler(
bot=bot,
io_layer=io_layer,
runtime_status_provider=lambda: RuntimeStatus(
started_at="2026-03-12 00:00:00 UTC",
loops=[],
command_access_mode="group_member",
protected_commands=["/city", "/deb"],
required_group_chat_id="-100123",
),
entitlement_service=entitlement,
)
result = handler.handle_chat_join_request(_join_request())
assert result == "pending:no_active_subscription"
assert bot.approved_join_requests == []
def test_join_request_approves_trial_user_with_queued_paid_subscription(monkeypatch):
monkeypatch.setenv("POLYWEATHER_TELEGRAM_GROUP_ID", "-100123")
bot = DummyBot()
db = SimpleNamespace(list_supabase_user_ids_for_telegram=lambda telegram_id: ["user-1"])
io_layer = SimpleNamespace(
build_welcome_text=lambda: "WELCOME",
build_points_rank_text=lambda _user: "TOP",
db=db,
)
entitlement = SimpleNamespace(
get_subscription_window=lambda user_id, respect_requirement=False: {
"rows": [
{"plan_code": "signup_trial_3d", "source": "signup_trial"},
{"plan_code": "pro_monthly", "source": "payment_contract"},
]
},
get_latest_active_subscription=lambda user_id, respect_requirement=False: {
"plan_code": "signup_trial_3d",
"source": "signup_trial",
},
)
handler = BasicCommandHandler(
bot=bot,
io_layer=io_layer,
runtime_status_provider=lambda: RuntimeStatus(
started_at="2026-03-12 00:00:00 UTC",
loops=[],
command_access_mode="group_member",
protected_commands=["/city", "/deb"],
required_group_chat_id="-100123",
),
entitlement_service=entitlement,
)
result = handler.handle_chat_join_request(_join_request())
assert result == "approved"
assert bot.approved_join_requests == [{"chat_id": -100123, "user_id": 12345}]
def test_join_request_can_decline_ineligible_user_when_configured(monkeypatch):
monkeypatch.setenv("POLYWEATHER_TELEGRAM_GROUP_ID", "-100123")
monkeypatch.setenv("POLYWEATHER_TELEGRAM_JOIN_INELIGIBLE_ACTION", "decline")