Djeex
530aeba667
Bring nvidia-stock-bot up to the same CI/CD maturity as adguard-cidre: pytest suite (19 tests), multi-stage Dockerfile with a dedicated test stage, pinned requests dependency, Gitea Actions workflow (build, smoke test, unit tests, deprecation check, Trivy critical/high scan, version bump + registry publish + release), and a Renovate config for automated dependency updates.
100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
import importlib
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
MODULE_NAME = "env_config"
|
|
|
|
VALID_WEBHOOK = "https://discord.com/api/webhooks/123456789012345678/abcdef"
|
|
|
|
ENV_KEYS = [
|
|
"DISCORD_WEBHOOK_URL",
|
|
"DISCORD_SERVER_NAME",
|
|
"DISCORD_ROLES",
|
|
"COUNTRY",
|
|
"REFRESH_TIME",
|
|
"TEST_MODE",
|
|
"PRODUCT_NAMES",
|
|
]
|
|
|
|
|
|
def _reload_env_config(monkeypatch, env):
|
|
for key in ENV_KEYS:
|
|
monkeypatch.delenv(key, raising=False)
|
|
for key, value in env.items():
|
|
monkeypatch.setenv(key, value)
|
|
sys.modules.pop(MODULE_NAME, None)
|
|
return importlib.import_module(MODULE_NAME)
|
|
|
|
|
|
def test_missing_webhook_exits(monkeypatch):
|
|
with pytest.raises(SystemExit):
|
|
_reload_env_config(monkeypatch, {"PRODUCT_NAMES": "RTX 5090"})
|
|
|
|
|
|
def test_missing_product_names_exits(monkeypatch):
|
|
with pytest.raises(SystemExit):
|
|
_reload_env_config(monkeypatch, {"DISCORD_WEBHOOK_URL": VALID_WEBHOOK})
|
|
|
|
|
|
def test_default_role_map_is_everyone(monkeypatch):
|
|
cfg = _reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090, RTX 5080",
|
|
})
|
|
assert cfg.DISCORD_ROLE_MAP == {"RTX 5090": "@everyone", "RTX 5080": "@everyone"}
|
|
|
|
|
|
def test_role_count_mismatch_exits(monkeypatch):
|
|
with pytest.raises(SystemExit):
|
|
_reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090, RTX 5080",
|
|
"DISCORD_ROLES": "<@&123456789012345678>",
|
|
})
|
|
|
|
|
|
def test_invalid_role_format_exits(monkeypatch):
|
|
with pytest.raises(SystemExit):
|
|
_reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090",
|
|
"DISCORD_ROLES": "not-a-role",
|
|
})
|
|
|
|
|
|
def test_valid_role_format_accepted(monkeypatch):
|
|
cfg = _reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090",
|
|
"DISCORD_ROLES": "<@&123456789012345678>",
|
|
})
|
|
assert cfg.DISCORD_ROLE_MAP["RTX 5090"] == "<@&123456789012345678>"
|
|
|
|
|
|
def test_unknown_country_falls_back_to_us(monkeypatch):
|
|
cfg = _reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090",
|
|
"COUNTRY": "ZZ",
|
|
})
|
|
assert cfg.currency == "$"
|
|
|
|
|
|
def test_known_country_currency(monkeypatch):
|
|
cfg = _reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090",
|
|
"COUNTRY": "GB",
|
|
})
|
|
assert cfg.currency == "£"
|
|
|
|
|
|
def test_refresh_time_invalid_exits(monkeypatch):
|
|
with pytest.raises(SystemExit):
|
|
_reload_env_config(monkeypatch, {
|
|
"DISCORD_WEBHOOK_URL": VALID_WEBHOOK,
|
|
"PRODUCT_NAMES": "RTX 5090",
|
|
"REFRESH_TIME": "not-a-number",
|
|
})
|