mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-01-26 06:40:31 +00:00
427011fe41
2) У промогрупп появится приоритет 3) У пользователя может быть несколько промогрупп, но влиять будет только с наивысшим приоритетом 4) К промокодам можно будет добавить промогруппу. Все активировавшие промокод получат её 5) При выводе пользователей с промогруппой будет также выводиться ссылка на каждого. Можно будет отследить сливы промокодов "для своих". Я в целом это добавлю во все места, где пользователь выводится в админке 6) Исправить баг исчезновения триалки при пополнении 7) Исправить падающие тесты и добавить новых 8) Трафик: 0 ГБ в тестовой подписке исправить на Трафик: Безлимит
160 lines
5.3 KiB
Python
160 lines
5.3 KiB
Python
from types import SimpleNamespace
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
ROOT_DIR = Path(__file__).resolve().parents[2]
|
|
if str(ROOT_DIR) not in sys.path:
|
|
sys.path.insert(0, str(ROOT_DIR))
|
|
|
|
from app.config import settings
|
|
from app.services.system_settings_service import bot_configuration_service
|
|
|
|
|
|
async def test_env_override_prevents_set_value(monkeypatch):
|
|
bot_configuration_service.initialize_definitions()
|
|
|
|
env_value = "env_support"
|
|
monkeypatch.setattr(settings, "SUPPORT_USERNAME", env_value)
|
|
original_values = dict(bot_configuration_service._original_values)
|
|
original_values["SUPPORT_USERNAME"] = env_value
|
|
monkeypatch.setattr(bot_configuration_service, "_original_values", original_values)
|
|
|
|
env_keys = set(bot_configuration_service._env_override_keys)
|
|
env_keys.add("SUPPORT_USERNAME")
|
|
monkeypatch.setattr(bot_configuration_service, "_env_override_keys", env_keys)
|
|
monkeypatch.setattr(bot_configuration_service, "_overrides_raw", {})
|
|
|
|
async def fake_upsert(db, key, value, description=None): # noqa: ANN001
|
|
return None
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.system_settings_service.upsert_system_setting",
|
|
fake_upsert,
|
|
)
|
|
|
|
await bot_configuration_service.set_value(
|
|
object(),
|
|
"SUPPORT_USERNAME",
|
|
"db_support",
|
|
)
|
|
|
|
assert settings.SUPPORT_USERNAME == env_value
|
|
assert not bot_configuration_service.has_override("SUPPORT_USERNAME")
|
|
|
|
|
|
async def test_env_override_prevents_reset_value(monkeypatch):
|
|
bot_configuration_service.initialize_definitions()
|
|
|
|
env_value = "env_support"
|
|
monkeypatch.setattr(settings, "SUPPORT_USERNAME", env_value)
|
|
original_values = dict(bot_configuration_service._original_values)
|
|
original_values["SUPPORT_USERNAME"] = env_value
|
|
monkeypatch.setattr(bot_configuration_service, "_original_values", original_values)
|
|
|
|
env_keys = set(bot_configuration_service._env_override_keys)
|
|
env_keys.add("SUPPORT_USERNAME")
|
|
monkeypatch.setattr(bot_configuration_service, "_env_override_keys", env_keys)
|
|
monkeypatch.setattr(bot_configuration_service, "_overrides_raw", {"SUPPORT_USERNAME": "db"})
|
|
|
|
async def fake_delete(db, key): # noqa: ANN001
|
|
return None
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.system_settings_service.delete_system_setting",
|
|
fake_delete,
|
|
)
|
|
|
|
await bot_configuration_service.reset_value(
|
|
object(),
|
|
"SUPPORT_USERNAME",
|
|
)
|
|
|
|
assert settings.SUPPORT_USERNAME == env_value
|
|
assert not bot_configuration_service.has_override("SUPPORT_USERNAME")
|
|
|
|
|
|
async def test_initialize_skips_db_value_for_env_override(monkeypatch):
|
|
bot_configuration_service.initialize_definitions()
|
|
|
|
env_value = "env_support"
|
|
monkeypatch.setattr(settings, "SUPPORT_USERNAME", env_value)
|
|
original_values = dict(bot_configuration_service._original_values)
|
|
original_values["SUPPORT_USERNAME"] = env_value
|
|
monkeypatch.setattr(bot_configuration_service, "_original_values", original_values)
|
|
|
|
env_keys = set(bot_configuration_service._env_override_keys)
|
|
env_keys.add("SUPPORT_USERNAME")
|
|
monkeypatch.setattr(bot_configuration_service, "_env_override_keys", env_keys)
|
|
monkeypatch.setattr(bot_configuration_service, "_overrides_raw", {})
|
|
|
|
class DummyResult:
|
|
def scalars(self):
|
|
return self
|
|
|
|
def all(self):
|
|
return [SimpleNamespace(key="SUPPORT_USERNAME", value="db_support")]
|
|
|
|
class DummySession:
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, exc_type, exc, tb): # noqa: ANN001
|
|
return False
|
|
|
|
async def execute(self, query): # noqa: ANN001
|
|
return DummyResult()
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.system_settings_service.AsyncSessionLocal",
|
|
lambda: DummySession(),
|
|
)
|
|
|
|
async def fake_sync():
|
|
return True
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.system_settings_service.ensure_default_web_api_token",
|
|
fake_sync,
|
|
raising=False,
|
|
)
|
|
|
|
await bot_configuration_service.initialize()
|
|
|
|
assert settings.SUPPORT_USERNAME == env_value
|
|
assert "SUPPORT_USERNAME" not in bot_configuration_service._overrides_raw
|
|
assert not bot_configuration_service.has_override("SUPPORT_USERNAME")
|
|
|
|
|
|
async def test_set_value_applies_without_env_override(monkeypatch):
|
|
bot_configuration_service.initialize_definitions()
|
|
|
|
monkeypatch.setattr(bot_configuration_service, "_env_override_keys", set())
|
|
monkeypatch.setattr(bot_configuration_service, "_overrides_raw", {})
|
|
|
|
initial_value = True
|
|
target_value = False
|
|
|
|
monkeypatch.setattr(settings, "SUPPORT_MENU_ENABLED", initial_value)
|
|
original_values = dict(bot_configuration_service._original_values)
|
|
original_values["SUPPORT_MENU_ENABLED"] = initial_value
|
|
monkeypatch.setattr(bot_configuration_service, "_original_values", original_values)
|
|
|
|
async def fake_upsert(db, key, value, description=None): # noqa: ANN001
|
|
return None
|
|
|
|
monkeypatch.setattr(
|
|
"app.services.system_settings_service.upsert_system_setting",
|
|
fake_upsert,
|
|
)
|
|
|
|
await bot_configuration_service.set_value(
|
|
object(),
|
|
"SUPPORT_MENU_ENABLED",
|
|
target_value,
|
|
)
|
|
|
|
assert settings.SUPPORT_MENU_ENABLED is target_value
|
|
assert bot_configuration_service.has_override("SUPPORT_MENU_ENABLED")
|