mirror of
https://github.com/mblanke/Lottery-Tracker.git
synced 2026-03-01 14:10:22 -05:00
112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
"""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
|