Nikolay Tatarinov
67ba9826c7
- Updated the language configuration to use a single source of truth from `DEFAULT_LANGUAGE` for the bot, API, and Mini App, eliminating auto-detection from user settings. - Refactored the `get_lang` function to always return `DEFAULT_LANGUAGE`, ensuring consistent language usage throughout the application. - Modified the handling of language in various components, including API responses and UI elements, to reflect the new language management approach. - Enhanced documentation and comments to clarify the changes in language handling. - Added unit tests to verify the new language handling behavior and ensure coverage for the updated functionality.
137 lines
5.0 KiB
Python
137 lines
5.0 KiB
Python
"""Tests for duty_teller.api.telegram_auth.validate_init_data and validate_init_data_with_reason."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import duty_teller.config as config
|
|
from duty_teller.api.telegram_auth import (
|
|
validate_init_data,
|
|
validate_init_data_with_reason,
|
|
)
|
|
from tests.helpers import make_init_data
|
|
|
|
|
|
def test_valid_payload_returns_username():
|
|
bot_token = "123:ABC"
|
|
user = {"id": 123, "username": "testuser", "first_name": "Test"}
|
|
init_data = make_init_data(user, bot_token)
|
|
assert validate_init_data(init_data, bot_token) == "testuser"
|
|
|
|
|
|
def test_valid_payload_username_lowercase():
|
|
bot_token = "123:ABC"
|
|
user = {"id": 123, "username": "TestUser", "first_name": "Test"}
|
|
init_data = make_init_data(user, bot_token)
|
|
assert validate_init_data(init_data, bot_token) == "testuser"
|
|
|
|
|
|
def test_invalid_hash_returns_none():
|
|
bot_token = "123:ABC"
|
|
user = {"id": 123, "username": "testuser"}
|
|
init_data = make_init_data(user, bot_token)
|
|
# Tamper with hash
|
|
init_data = init_data.replace("hash=", "hash=x")
|
|
assert validate_init_data(init_data, bot_token) is None
|
|
|
|
|
|
def test_wrong_bot_token_returns_none():
|
|
bot_token = "123:ABC"
|
|
user = {"id": 123, "username": "testuser"}
|
|
init_data = make_init_data(user, bot_token)
|
|
assert validate_init_data(init_data, "other:token") is None
|
|
|
|
|
|
def test_missing_user_returns_none():
|
|
bot_token = "123:ABC"
|
|
init_data = make_init_data(None, bot_token) # no user key
|
|
assert validate_init_data(init_data, bot_token) is None
|
|
|
|
|
|
def test_user_without_username_returns_none_from_validate_init_data():
|
|
"""validate_init_data returns None when user has no username (backward compat)."""
|
|
bot_token = "123:ABC"
|
|
user = {"id": 123, "first_name": "Test"} # no username
|
|
init_data = make_init_data(user, bot_token)
|
|
assert validate_init_data(init_data, bot_token) is None
|
|
|
|
|
|
def test_user_without_username_but_with_id_succeeds_with_reason():
|
|
"""With validate_init_data_with_reason, valid user.id is enough; lang is DEFAULT_LANGUAGE."""
|
|
bot_token = "123:ABC"
|
|
user = {"id": 456, "first_name": "Test", "language_code": "ru"}
|
|
init_data = make_init_data(user, bot_token)
|
|
telegram_user_id, username, reason, lang = validate_init_data_with_reason(
|
|
init_data, bot_token
|
|
)
|
|
assert telegram_user_id == 456
|
|
assert username is None
|
|
assert reason == "ok"
|
|
assert lang == config.DEFAULT_LANGUAGE
|
|
|
|
|
|
def test_user_without_id_returns_no_user_id():
|
|
"""When user object exists but has no 'id', return no_user_id; lang is DEFAULT_LANGUAGE."""
|
|
bot_token = "123:ABC"
|
|
user = {"first_name": "Test"} # no id
|
|
init_data = make_init_data(user, bot_token)
|
|
telegram_user_id, username, reason, lang = validate_init_data_with_reason(
|
|
init_data, bot_token
|
|
)
|
|
assert telegram_user_id is None
|
|
assert username is None
|
|
assert reason == "no_user_id"
|
|
assert lang == config.DEFAULT_LANGUAGE
|
|
|
|
|
|
def test_validate_init_data_with_reason_returns_default_language_ignoring_user_lang():
|
|
"""Returned lang is always config.DEFAULT_LANGUAGE, not user.language_code."""
|
|
with patch("duty_teller.api.telegram_auth.config.DEFAULT_LANGUAGE", "ru"):
|
|
user = {"id": 1, "first_name": "U", "language_code": "en"}
|
|
init_data = make_init_data(user, "123:ABC")
|
|
_, _, reason, lang = validate_init_data_with_reason(init_data, "123:ABC")
|
|
assert reason == "ok"
|
|
assert lang == "ru"
|
|
|
|
|
|
def test_empty_init_data_returns_none():
|
|
assert validate_init_data("", "token") is None
|
|
assert validate_init_data(" ", "token") is None
|
|
|
|
|
|
def test_empty_bot_token_returns_none():
|
|
user = {"id": 1, "username": "u"}
|
|
init_data = make_init_data(user, "token")
|
|
assert validate_init_data(init_data, "") is None
|
|
|
|
|
|
def test_auth_date_expiry_rejects_old_init_data():
|
|
"""When max_age_seconds is set, initData older than that is rejected."""
|
|
import time as t
|
|
|
|
bot_token = "123:ABC"
|
|
user = {"id": 1, "username": "testuser"}
|
|
# auth_date 100 seconds ago
|
|
old_ts = int(t.time()) - 100
|
|
init_data = make_init_data(user, bot_token, auth_date=old_ts)
|
|
assert validate_init_data(init_data, bot_token, max_age_seconds=60) is None
|
|
assert validate_init_data(init_data, bot_token, max_age_seconds=200) == "testuser"
|
|
|
|
|
|
def test_auth_date_expiry_accepts_fresh_init_data():
|
|
"""Fresh auth_date within max_age_seconds is accepted."""
|
|
import time as t
|
|
|
|
bot_token = "123:ABC"
|
|
user = {"id": 1, "username": "testuser"}
|
|
fresh_ts = int(t.time()) - 10
|
|
init_data = make_init_data(user, bot_token, auth_date=fresh_ts)
|
|
assert validate_init_data(init_data, bot_token, max_age_seconds=60) == "testuser"
|
|
|
|
|
|
def test_auth_date_expiry_requires_auth_date_when_max_age_set():
|
|
"""When max_age_seconds is set but auth_date is missing, return None."""
|
|
bot_token = "123:ABC"
|
|
user = {"id": 1, "username": "testuser"}
|
|
init_data = make_init_data(user, bot_token) # no auth_date
|
|
assert validate_init_data(init_data, bot_token, max_age_seconds=86400) is None
|
|
assert validate_init_data(init_data, bot_token) == "testuser"
|