Version 1.1

2026-03-01 14:10:22 -05:00 · 2026-02-18 08:24:54 -05:00
parent 4318c8f642
commit fdba869a8d
33 changed files with 2142 additions and 1942 deletions
--- a/tests/test_scrapers.py
+++ b/tests/test_scrapers.py
@@ -0,0 +1,111 @@
+"""Tests for scrapers.py — uses mocks to avoid real HTTP calls."""
+
+from __future__ import annotations
+
+from unittest.mock import MagicMock, patch
+
+from scrapers import (
+    _parse_jackpot_from_lotto_net,
+    clear_cache,
+    get_all_jackpots,
+    scrape_mega_millions,
+    scrape_powerball,
+)
+
+# ---------------------------------------------------------------------------
+# HTML parser unit tests
+# ---------------------------------------------------------------------------
+
+POWERBALL_HTML = """
+<html><body>
+<h2>Powerball</h2>
+<div>
+Next Jackpot
+$350 Million
+</div>
+</body></html>
+"""
+
+BILLION_HTML = """
+<html><body>
+<div>
+Next Jackpot
+$1.5 Billion
+</div>
+</body></html>
+"""
+
+
+def test_parse_jackpot_millions():
+    result = _parse_jackpot_from_lotto_net(POWERBALL_HTML)
+    assert result == 350_000_000
+
+
+def test_parse_jackpot_billions():
+    result = _parse_jackpot_from_lotto_net(BILLION_HTML)
+    assert result == 1_500_000_000
+
+
+def test_parse_jackpot_no_match():
+    result = _parse_jackpot_from_lotto_net("<html><body>No jackpot here</body></html>")
+    assert result is None
+
+
+# ---------------------------------------------------------------------------
+# Scraper integration tests (mocked HTTP)
+# ---------------------------------------------------------------------------
+
+@patch("scrapers.requests.get")
+def test_scrape_powerball_success(mock_get):
+    mock_resp = MagicMock()
+    mock_resp.text = POWERBALL_HTML
+    mock_resp.raise_for_status = MagicMock()
+    mock_get.return_value = mock_resp
+
+    result = scrape_powerball()
+    assert result == 350_000_000
+
+
+@patch("scrapers.requests.get")
+def test_scrape_powerball_failure(mock_get):
+    mock_get.side_effect = Exception("Network error")
+    result = scrape_powerball()
+    assert result is None
+
+
+@patch("scrapers.requests.get")
+def test_scrape_mega_millions_success(mock_get):
+    html = POWERBALL_HTML.replace("Powerball", "Mega Millions")
+    mock_resp = MagicMock()
+    mock_resp.text = html
+    mock_resp.raise_for_status = MagicMock()
+    mock_get.return_value = mock_resp
+
+    result = scrape_mega_millions()
+    assert result == 350_000_000
+
+
+# ---------------------------------------------------------------------------
+# Cache tests
+# ---------------------------------------------------------------------------
+
+@patch("scrapers.scrape_powerball", return_value=100_000_000)
+@patch("scrapers.scrape_mega_millions", return_value=200_000_000)
+@patch("scrapers.scrape_canadian_lotteries", return_value={"lottoMax": 50_000_000, "lotto649": 10_000_000})
+def test_get_all_jackpots_caches(mock_ca, mock_mm, mock_pb):
+    clear_cache()
+    r1 = get_all_jackpots()
+    r2 = get_all_jackpots()
+    # Second call should use cache — scrapers called only once
+    assert mock_pb.call_count == 1
+    assert r1 == r2
+
+
+@patch("scrapers.scrape_powerball", return_value=100_000_000)
+@patch("scrapers.scrape_mega_millions", return_value=200_000_000)
+@patch("scrapers.scrape_canadian_lotteries", return_value={"lottoMax": 50_000_000, "lotto649": 10_000_000})
+def test_get_all_jackpots_force_refresh(mock_ca, mock_mm, mock_pb):
+    clear_cache()
+    get_all_jackpots()
+    get_all_jackpots(force_refresh=True)
+    assert mock_pb.call_count == 2