130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
import json
|
|
import unittest
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
from bot.app.web import subscription_webapp
|
|
from bot.handlers.user import referral
|
|
from bot.keyboards.inline.user_keyboards import (
|
|
get_bot_interface_inline_keyboard,
|
|
get_information_links_keyboard,
|
|
get_language_selection_keyboard,
|
|
get_main_menu_inline_keyboard,
|
|
get_referral_link_keyboard,
|
|
get_subscription_options_keyboard,
|
|
)
|
|
|
|
|
|
class JsonI18nStub:
|
|
def __init__(self):
|
|
self.translations = json.loads(Path("locales/en.json").read_text(encoding="utf-8"))
|
|
|
|
def gettext(self, lang, key, **kwargs):
|
|
text = self.translations[key]
|
|
return text.format(**kwargs) if kwargs else text
|
|
|
|
|
|
class UserBotMenuTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.i18n = JsonI18nStub()
|
|
self.settings = SimpleNamespace(
|
|
SUBSCRIPTION_MINI_APP_URL="https://app.example.com/",
|
|
SUPPORT_LINK="https://t.me/support",
|
|
PRIVACY_POLICY_URL="https://example.com/privacy",
|
|
USER_AGREEMENT_URL="https://example.com/agreement",
|
|
TERMS_OF_SERVICE_URL="",
|
|
TRIAL_ENABLED=True,
|
|
SERVER_STATUS_URL="",
|
|
)
|
|
|
|
def _callback_data(self, markup):
|
|
return [
|
|
button.callback_data
|
|
for row in markup.inline_keyboard
|
|
for button in row
|
|
if button.callback_data
|
|
]
|
|
|
|
def test_main_menu_exposes_bot_menu_and_information(self):
|
|
markup = get_main_menu_inline_keyboard("en", self.i18n, self.settings)
|
|
|
|
callbacks = self._callback_data(markup)
|
|
|
|
self.assertIn("main_action:bot_interface", callbacks)
|
|
self.assertIn("main_action:info", callbacks)
|
|
|
|
def test_bot_interface_buttons_return_to_bot_interface(self):
|
|
markup = get_bot_interface_inline_keyboard("en", self.i18n, self.settings)
|
|
|
|
callbacks = self._callback_data(markup)
|
|
|
|
self.assertIn("main_action:bot_subscribe", callbacks)
|
|
self.assertIn("main_action:bot_my_subscription", callbacks)
|
|
self.assertIn("main_action:bot_referral", callbacks)
|
|
self.assertIn("main_action:bot_info", callbacks)
|
|
self.assertIn("main_action:back_to_main", callbacks)
|
|
|
|
def test_nested_bot_menu_keyboards_can_target_bot_interface_back(self):
|
|
subscription_markup = get_subscription_options_keyboard(
|
|
{1: 100},
|
|
"RUB",
|
|
"en",
|
|
self.i18n,
|
|
back_callback="main_action:bot_interface",
|
|
)
|
|
referral_markup = get_referral_link_keyboard(
|
|
"en",
|
|
self.i18n,
|
|
back_callback="main_action:bot_interface",
|
|
)
|
|
info_markup = get_information_links_keyboard(
|
|
"en",
|
|
self.i18n,
|
|
"https://example.com/privacy",
|
|
"https://example.com/agreement",
|
|
back_callback="main_action:bot_interface",
|
|
)
|
|
language_markup = get_language_selection_keyboard(
|
|
self.i18n,
|
|
"en",
|
|
back_callback="main_action:bot_interface",
|
|
)
|
|
|
|
self.assertIn("main_action:bot_interface", self._callback_data(subscription_markup))
|
|
self.assertIn("main_action:bot_interface", self._callback_data(referral_markup))
|
|
self.assertIn("main_action:bot_interface", self._callback_data(info_markup))
|
|
self.assertIn("set_lang_ru:bot", self._callback_data(language_markup))
|
|
|
|
def test_webapp_referral_link_uses_ref_query_and_is_normalized(self):
|
|
link = referral._build_webapp_referral_link(
|
|
"https://app.example.com/invite?utm=channel",
|
|
"AbC123xYz",
|
|
)
|
|
|
|
self.assertEqual(
|
|
link,
|
|
"https://app.example.com/invite?utm=channel&ref=uAbC123xYz",
|
|
)
|
|
self.assertEqual(subscription_webapp._normalize_referral_param("uAbC123xYz"), "ABC123XYZ")
|
|
self.assertEqual(
|
|
subscription_webapp._normalize_referral_param("ref_uAbC123xYz"), "ABC123XYZ"
|
|
)
|
|
|
|
def test_referral_text_places_web_link_after_telegram_link(self):
|
|
text = self.i18n.gettext(
|
|
"en",
|
|
"referral_program_info_new",
|
|
referral_link="https://t.me/bot?start=ref_uABC123XYZ",
|
|
webapp_link_section=self.i18n.gettext(
|
|
"en",
|
|
"referral_webapp_link_line",
|
|
webapp_referral_link="https://app.example.com/?ref=uABC123XYZ",
|
|
),
|
|
bonus_details="bonus",
|
|
invited_count=1,
|
|
purchased_count=0,
|
|
)
|
|
|
|
self.assertLess(text.index("Telegram link:"), text.index("Web link:"))
|
|
self.assertLess(text.index("Web link:"), text.index("Invitation bonuses:"))
|