mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-23 12:53:41 +00:00
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
168 lines
4.6 KiB
Python
168 lines
4.6 KiB
Python
import pytest
|
|
|
|
from app.utils import subscription_utils
|
|
from app.utils.subscription_utils import (
|
|
resolve_hwid_device_limit,
|
|
resolve_hwid_device_limit_for_payload,
|
|
resolve_simple_subscription_device_limit,
|
|
)
|
|
|
|
|
|
class DummySubscription:
|
|
def __init__(self, device_limit=None):
|
|
self.device_limit = device_limit
|
|
|
|
|
|
class StubSettings:
|
|
def __init__(
|
|
self,
|
|
enabled: bool,
|
|
disabled_amount,
|
|
*,
|
|
simple_limit: int = 3,
|
|
disabled_selection_amount=None,
|
|
):
|
|
self._enabled = enabled
|
|
self._disabled_amount = disabled_amount
|
|
self._disabled_selection_amount = disabled_selection_amount
|
|
self.SIMPLE_SUBSCRIPTION_DEVICE_LIMIT = simple_limit
|
|
|
|
def is_devices_selection_enabled(self) -> bool:
|
|
return self._enabled
|
|
|
|
def get_disabled_mode_device_limit(self):
|
|
return self._disabled_amount
|
|
|
|
def get_devices_selection_disabled_amount(self):
|
|
return self._disabled_selection_amount
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'forced_amount, expected',
|
|
[
|
|
(None, None),
|
|
(0, 0),
|
|
(5, 5),
|
|
],
|
|
)
|
|
def test_resolve_hwid_device_limit_disabled_mode(monkeypatch, forced_amount, expected):
|
|
subscription = DummySubscription(device_limit=42)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(
|
|
enabled=False,
|
|
disabled_amount=forced_amount,
|
|
disabled_selection_amount=forced_amount,
|
|
),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) == expected
|
|
|
|
|
|
def test_resolve_hwid_device_limit_enabled_mode(monkeypatch):
|
|
subscription = DummySubscription(device_limit=4)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=True, disabled_amount=None),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) == 4
|
|
|
|
|
|
def test_resolve_hwid_device_limit_enabled_ignores_non_positive(monkeypatch):
|
|
subscription = DummySubscription(device_limit=0)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=True, disabled_amount=None),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) is None
|
|
|
|
|
|
def test_resolve_hwid_device_limit_for_payload_returns_subscription_limit(monkeypatch):
|
|
subscription = DummySubscription(device_limit=42)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=False, disabled_amount=None, disabled_selection_amount=None),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) is None
|
|
assert resolve_hwid_device_limit_for_payload(subscription) == 42
|
|
|
|
|
|
def test_resolve_hwid_device_limit_for_payload_ignores_non_positive(monkeypatch):
|
|
subscription = DummySubscription(device_limit=0)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=False, disabled_amount=None, disabled_selection_amount=None),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) is None
|
|
assert resolve_hwid_device_limit_for_payload(subscription) is None
|
|
|
|
|
|
def test_resolve_hwid_device_limit_for_payload_prefers_forced_limit(monkeypatch):
|
|
subscription = DummySubscription(device_limit=42)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=False, disabled_amount=7, disabled_selection_amount=7),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit_for_payload(subscription) == 7
|
|
|
|
|
|
def test_resolve_hwid_device_limit_for_payload_handles_zero(monkeypatch):
|
|
subscription = DummySubscription(device_limit=42)
|
|
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(enabled=False, disabled_amount=0, disabled_selection_amount=0),
|
|
)
|
|
|
|
assert resolve_hwid_device_limit(subscription) == 0
|
|
assert resolve_hwid_device_limit_for_payload(subscription) == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'enabled, simple_limit, disabled_amount, disabled_selection_amount, expected',
|
|
[
|
|
(True, 4, None, None, 4),
|
|
(False, 4, None, None, 4),
|
|
(False, 4, 0, 0, 0),
|
|
(False, 4, 7, 7, 7),
|
|
],
|
|
)
|
|
def test_resolve_simple_subscription_device_limit(
|
|
monkeypatch,
|
|
enabled,
|
|
simple_limit,
|
|
disabled_amount,
|
|
disabled_selection_amount,
|
|
expected,
|
|
):
|
|
monkeypatch.setattr(
|
|
subscription_utils,
|
|
'settings',
|
|
StubSettings(
|
|
enabled=enabled,
|
|
disabled_amount=disabled_amount,
|
|
simple_limit=simple_limit,
|
|
disabled_selection_amount=disabled_selection_amount,
|
|
),
|
|
)
|
|
|
|
assert resolve_simple_subscription_device_limit() == expected
|