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):
    monkeypatch.setenv("DISCORD_WEBHOOK_URL", VALID_WEBHOOK)
    monkeypatch.setenv("PRODUCT_NAMES", "RTX 5090 Founders Edition")
    monkeypatch.setenv("TEST_MODE", test_mode)
    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