83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
import unittest
|
|
import json
|
|
|
|
from pydantic import ValidationError
|
|
|
|
from config.settings import Settings
|
|
|
|
|
|
class SettingsTests(unittest.TestCase):
|
|
def test_blank_postgres_password_is_rejected(self):
|
|
with self.assertRaises(ValidationError):
|
|
Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="",
|
|
)
|
|
|
|
def test_webapp_secrets_are_generated_when_missing(self):
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
)
|
|
|
|
self.assertTrue(settings.WEBAPP_SESSION_SECRET)
|
|
self.assertTrue(settings.WEBHOOK_SECRET_TOKEN)
|
|
self.assertEqual(settings.WEBAPP_SESSION_TTL_SECONDS, 86400)
|
|
|
|
def test_tariffs_config_missing_uses_legacy_fallback(self):
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH="missing-tariffs.json",
|
|
TRAFFIC_PACKAGES="10:199",
|
|
)
|
|
|
|
self.assertIsNone(settings.tariffs_config)
|
|
self.assertTrue(settings.traffic_sale_mode)
|
|
|
|
def test_existing_tariffs_config_disables_legacy_traffic_mode(self):
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
path = Path(tmpdir) / "tariffs.json"
|
|
path.write_text(
|
|
json.dumps(
|
|
{
|
|
"default_tariff": "standard",
|
|
"tariffs": [
|
|
{
|
|
"key": "standard",
|
|
"names": {"ru": "Стандарт"},
|
|
"descriptions": {},
|
|
"squad_uuids": ["uuid"],
|
|
"billing_model": "period",
|
|
"monthly_gb": 100,
|
|
"prices_rub": {"1": 150},
|
|
"prices_stars": {"1": 0},
|
|
"enabled_periods": [1],
|
|
"enabled": True,
|
|
}
|
|
],
|
|
}
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH=str(path),
|
|
TRAFFIC_PACKAGES="10:199",
|
|
)
|
|
|
|
self.assertIsNotNone(settings.tariffs_config)
|
|
self.assertFalse(settings.traffic_sale_mode)
|