nvidia-stock-bot/tests/test_notifier.py

import importlib
import sys

VALID_WEBHOOK = "https://discord.com/api/webhooks/123456789012345678/abcdef"


def _import_notifier(monkeypatch, test_mode="False", discord_roles=None, keep_discord=True):
    if keep_discord:
        monkeypatch.setenv("DISCORD_WEBHOOK_URL", VALID_WEBHOOK)
    monkeypatch.setenv("PRODUCT_NAMES", "RTX 5090 Founders Edition")
    monkeypatch.setenv("TEST_MODE", test_mode)
    if not keep_discord:
        monkeypatch.delenv("DISCORD_WEBHOOK_URL", raising=False)
    if discord_roles is not None:
        monkeypatch.setenv("DISCORD_ROLES", discord_roles)
    else:
        monkeypatch.delenv("DISCORD_ROLES", raising=False)
    for mod in ("env_config", "notifier"):
        sys.modules.pop(mod, None)
    return importlib.import_module("notifier")


class FakeResponse:
    def __init__(self, status_code=204, text=""):
        self.status_code = status_code
        self.text = text


def test_test_mode_skips_network_call(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="True")
    calls = []
    monkeypatch.setattr(notifier.requests, "post", lambda *a, **k: calls.append((a, k)))

    notifier.send_discord_notification("RTX 5090 Founders Edition", "https://example.com", "1999")

    assert calls == []


def test_in_stock_notification_posts_expected_payload(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    captured = {}

    def fake_post(url, json=None, **kwargs):
        captured["url"] = url
        captured["json"] = json
        return FakeResponse()

    monkeypatch.setattr(notifier.requests, "post", fake_post)

    notifier.send_discord_notification(
        "RTX 5090 Founders Edition", "https://example.com/buy", "1999"
    )

    assert captured["url"] == notifier.DISCORD_WEBHOOK_URL
    assert captured["json"]["content"] == "@everyone"
    embed = captured["json"]["embeds"][0]
    assert "RTX 5090 Founders Edition" in embed["title"]


def test_discord_notification_survives_http_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    monkeypatch.setattr(
        notifier.requests, "post", lambda *a, **k: FakeResponse(status_code=500, text="boom")
    )

    notifier.send_discord_notification("RTX 5090 Founders Edition", "https://example.com", "1999")


def test_discord_notification_survives_connection_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")

    def raise_error(*a, **k):
        raise notifier.requests.exceptions.ConnectionError("boom")

    monkeypatch.setattr(notifier.requests, "post", raise_error)

    # Should not raise even though the request itself blew up (network down, DNS, etc.)
    notifier.send_discord_notification("RTX 5090 Founders Edition", "https://example.com", "1999")


def test_out_of_stock_test_mode_skips_network_call(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="True")
    calls = []
    monkeypatch.setattr(notifier.requests, "post", lambda *a, **k: calls.append((a, k)))

    notifier.send_out_of_stock_notification(
        "RTX 5090 Founders Edition", "https://example.com", "1999"
    )

    assert calls == []


def test_out_of_stock_notification_posts_on_success(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    captured = {}

    def fake_post(url, json=None, **kwargs):
        captured["json"] = json
        return FakeResponse(status_code=204)

    monkeypatch.setattr(notifier.requests, "post", fake_post)

    notifier.send_out_of_stock_notification(
        "RTX 5090 Founders Edition", "https://example.com/buy", "1999"
    )

    assert captured["json"]["embeds"][0]["url"] == "https://example.com/buy"


def test_out_of_stock_notification_survives_http_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    monkeypatch.setattr(
        notifier.requests, "post", lambda *a, **k: FakeResponse(status_code=500, text="boom")
    )

    # Should not raise even though the webhook call "fails"
    notifier.send_out_of_stock_notification(
        "RTX 5090 Founders Edition", "https://example.com", "1999"
    )


def test_out_of_stock_notification_survives_connection_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")

    def raise_error(*a, **k):
        raise notifier.requests.exceptions.ConnectionError("boom")

    monkeypatch.setattr(notifier.requests, "post", raise_error)

    notifier.send_out_of_stock_notification(
        "RTX 5090 Founders Edition", "https://example.com", "1999"
    )


def test_sku_change_test_mode_skips_network_call(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="True")
    calls = []
    monkeypatch.setattr(notifier.requests, "post", lambda *a, **k: calls.append((a, k)))

    notifier.send_sku_change_notification(
        "RTX 5090 Founders Edition", "old-sku", "new-sku", "https://example.com"
    )

    assert calls == []


def test_sku_change_notification_survives_http_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    monkeypatch.setattr(
        notifier.requests, "post", lambda *a, **k: FakeResponse(status_code=500, text="boom")
    )

    notifier.send_sku_change_notification(
        "RTX 5090 Founders Edition", "old-sku", "new-sku", "https://example.com"
    )


def test_sku_change_notification_survives_connection_error(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")

    def raise_error(*a, **k):
        raise notifier.requests.exceptions.ConnectionError("boom")

    monkeypatch.setattr(notifier.requests, "post", raise_error)

    notifier.send_sku_change_notification(
        "RTX 5090 Founders Edition", "old-sku", "new-sku", "https://example.com"
    )


def test_sku_change_notification_mentions_role_and_skus(monkeypatch):
    notifier = _import_notifier(
        monkeypatch, test_mode="False", discord_roles="<@&123456789012345678>"
    )
    captured = {}

    def fake_post(url, json=None, **kwargs):
        captured["json"] = json
        return FakeResponse()

    monkeypatch.setattr(notifier.requests, "post", fake_post)

    notifier.send_sku_change_notification(
        "RTX 5090 Founders Edition", "old-sku-123", "new-sku-456", "https://example.com"
    )

    assert "<@&123456789012345678>" in captured["json"]["content"]
    description = captured["json"]["embeds"][0]["description"]
    assert "old-sku-123" in description
    assert "new-sku-456" in description


def _import_telegram_notifier(monkeypatch, with_discord=False):
    if not with_discord:
        monkeypatch.delenv("DISCORD_WEBHOOK_URL", raising=False)
    monkeypatch.setenv("TELEGRAM_BOT_TOKEN", "123:abc")
    monkeypatch.setenv("TELEGRAM_CHAT_ID", "-100123")
    return _import_notifier(monkeypatch, test_mode="False", keep_discord=with_discord)


def test_telegram_in_stock_posts_to_bot_api(monkeypatch):
    notifier = _import_telegram_notifier(monkeypatch)
    captured = {}

    def fake_post(url, json=None, **kwargs):
        captured["url"] = url
        captured["json"] = json
        return FakeResponse(status_code=200)

    monkeypatch.setattr(notifier.requests, "post", fake_post)

    notifier.send_telegram_notification("RTX 5090 Founders Edition", "https://example.com/buy", "1999")

    assert captured["url"] == "https://api.telegram.org/bot123:abc/sendMessage"
    assert captured["json"]["chat_id"] == "-100123"
    assert "RTX 5090 Founders Edition" in captured["json"]["text"]
    assert "https://example.com/buy" in captured["json"]["text"]


def test_telegram_disabled_sends_nothing(monkeypatch):
    notifier = _import_notifier(monkeypatch, test_mode="False")
    calls = []
    monkeypatch.setattr(notifier.requests, "post", lambda *a, **k: calls.append(a))

    notifier.send_telegram_notification("RTX 5090 Founders Edition", "https://example.com", "1999")
    notifier.send_telegram_out_of_stock_notification("RTX 5090", "https://example.com", "1999")
    notifier.send_telegram_sku_change_notification("RTX 5090", "a", "b", "https://example.com")

    assert calls == []


def test_discord_disabled_sends_nothing(monkeypatch):
    notifier = _import_telegram_notifier(monkeypatch)
    calls = []
    monkeypatch.setattr(notifier.requests, "post", lambda *a, **k: calls.append(a))

    notifier.send_discord_notification("RTX 5090 Founders Edition", "https://example.com", "1999")

    assert calls == []


def test_telegram_error_does_not_leak_token_or_raise(monkeypatch, caplog):
    notifier = _import_telegram_notifier(monkeypatch)

    def raise_error(*a, **k):
        raise notifier.requests.exceptions.ConnectionError("https://api.telegram.org/bot123:abc/")

    monkeypatch.setattr(notifier.requests, "post", raise_error)

    notifier.send_telegram_out_of_stock_notification("RTX 5090", "https://example.com", "1999")

    assert "123:abc" not in caplog.text


def test_telegram_sku_change_and_http_error(monkeypatch):
    notifier = _import_telegram_notifier(monkeypatch)
    monkeypatch.setattr(
        notifier.requests, "post", lambda *a, **k: FakeResponse(status_code=400, text="bad")
    )

    notifier.send_telegram_sku_change_notification("RTX 5090", "old", "new", "https://example.com")