mirror of
https://github.com/agessaman/meshcore-bot.git
synced 2026-07-03 04:41:41 +00:00
agessaman
5aa84c4ffb
- Added new utility functions `espn_dates_for_local_day` and `filter_events_local_day` to handle local timezone date calculations and event filtering. - Updated `ESPNClient` to support optional date parameters in `fetch_scoreboard_with_calendar` and `fetch_match_states` methods, allowing for more precise data retrieval. - Modified `WorldCupCommand` and `WorldCupLiveService` to utilize the new date range functionality, ensuring accurate event reporting based on local time. - Implemented unit tests to verify the correct behavior of date handling and event filtering in various scenarios.
49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
"""Tests for ESPN local-day date span helpers used by World Cup features."""
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
import pytz
|
|
|
|
from modules.utils import espn_dates_for_local_day, filter_events_local_day
|
|
|
|
|
|
class TestEspnDatesForLocalDay:
|
|
def test_june_20_pt_spans_two_utc_days(self):
|
|
tz = pytz.timezone("America/Los_Angeles")
|
|
# 9:30pm PT on June 20 — evening match in progress
|
|
now = tz.localize(datetime(2026, 6, 20, 21, 30))
|
|
start, end, _, _ = espn_dates_for_local_day(tz, now)
|
|
assert start == "20260620"
|
|
assert end == "20260621"
|
|
|
|
def test_utc_midday_single_utc_day(self):
|
|
tz = timezone.utc
|
|
now = datetime(2026, 6, 20, 12, 0, tzinfo=timezone.utc)
|
|
start, end, _, _ = espn_dates_for_local_day(tz, now)
|
|
assert start == end == "20260620"
|
|
|
|
|
|
class TestFilterEventsLocalDay:
|
|
def test_keeps_afternoon_and_evening_drops_next_local_day(self):
|
|
tz = pytz.timezone("America/Los_Angeles")
|
|
now = tz.localize(datetime(2026, 6, 20, 21, 30))
|
|
_, _, local_start_ts, local_end_ts = espn_dates_for_local_day(tz, now)
|
|
|
|
afternoon_ts = datetime(2026, 6, 20, 21, 0, tzinfo=timezone.utc).timestamp() # 2pm PT
|
|
evening_ts = datetime(2026, 6, 21, 4, 0, tzinfo=timezone.utc).timestamp() # 9pm PT
|
|
next_day_ts = datetime(2026, 6, 21, 21, 0, tzinfo=timezone.utc).timestamp() # 2pm PT June 21
|
|
|
|
games = [
|
|
{"id": "afternoon", "event_timestamp": afternoon_ts},
|
|
{"id": "evening", "event_timestamp": evening_ts},
|
|
{"id": "tomorrow", "event_timestamp": next_day_ts},
|
|
]
|
|
filtered = filter_events_local_day(games, local_start_ts, local_end_ts)
|
|
assert [g["id"] for g in filtered] == ["afternoon", "evening"]
|
|
|
|
def test_keeps_events_missing_timestamp(self):
|
|
filtered = filter_events_local_day(
|
|
[{"id": "unknown"}], local_start_ts=0, local_end_ts=9999999999,
|
|
)
|
|
assert filtered == [{"id": "unknown"}]
|