intercept/tests/test_wefax_scheduler.py

"""Tests for WeFax auto-scheduler behavior and regressions."""

from __future__ import annotations

from datetime import datetime, timedelta, timezone
from unittest.mock import MagicMock, patch

from utils.wefax_scheduler import ScheduledBroadcast, WeFaxScheduler


class TestWeFaxScheduler:
    """WeFaxScheduler regression tests."""

    @patch('threading.Timer')
    def test_refresh_reschedules_same_utc_slot_next_day(self, mock_timer):
        """Completed broadcasts must not block the next day's same UTC slot."""
        scheduler = WeFaxScheduler()
        scheduler._enabled = True
        scheduler._station = 'USCG Kodiak'
        scheduler._callsign = 'NOJ'
        scheduler._frequency_khz = 4298.0

        now = datetime.now(timezone.utc)
        utc_time = (now - timedelta(hours=2)).strftime('%H:%M')
        today = now.date().isoformat()

        prior = ScheduledBroadcast(
            station='USCG Kodiak',
            callsign='NOJ',
            frequency_khz=4298.0,
            utc_time=utc_time,
            duration_min=20,
            content='Chart',
            occurrence_date=today,
        )
        prior.status = 'complete'
        scheduler._broadcasts = [prior]

        mock_timer.return_value = MagicMock()

        with patch('utils.wefax_scheduler.get_station', return_value={
            'name': 'USCG Kodiak',
            'schedule': [{
                'utc': utc_time,
                'duration_min': 20,
                'content': 'Chart',
            }],
        }):
            scheduler._refresh_schedule()

        capture_calls = [
            c for c in mock_timer.call_args_list
            if len(c.args) >= 2 and getattr(c.args[1], '__name__', '') == '_execute_capture'
        ]
        assert capture_calls, "Expected a capture timer for the next-day occurrence"

        scheduled = [b for b in scheduler._broadcasts if b.status == 'scheduled']
        assert len(scheduled) == 1
        assert scheduled[0].occurrence_date != today

    def test_execute_capture_stops_immediately_if_window_elapsed(self):
        """If stop delay computes to <= 0, capture should close out immediately."""
        scheduler = WeFaxScheduler()
        scheduler._enabled = True
        scheduler._callsign = 'NOJ'
        scheduler._frequency_khz = 4298.0
        scheduler._device = 0
        scheduler._gain = 40.0
        scheduler._ioc = 576
        scheduler._lpm = 120
        scheduler._direct_sampling = True

        now = datetime.now(timezone.utc)
        sb = ScheduledBroadcast(
            station='USCG Kodiak',
            callsign='NOJ',
            frequency_khz=4298.0,
            utc_time=now.strftime('%H:%M'),
            duration_min=0,
            content='Late chart',
            occurrence_date=now.date().isoformat(),
        )
        sb.status = 'scheduled'

        mock_decoder = MagicMock()
        mock_decoder.is_running = False
        mock_decoder.start.return_value = True

        with patch('utils.wefax_scheduler.get_wefax_decoder', return_value=mock_decoder), \
             patch('utils.wefax_scheduler.WEFAX_CAPTURE_BUFFER_SECONDS', 0), \
             patch('app.claim_sdr_device', return_value=None), \
             patch.object(scheduler, '_stop_capture') as mock_stop_capture:
            scheduler._execute_capture_inner(sb)

        mock_stop_capture.assert_called_once()

    @patch('threading.Timer')
    def test_terminal_progress_releases_scheduler_device_early(self, mock_timer):
        """Scheduler captures must release SDR as soon as terminal progress arrives."""
        scheduler = WeFaxScheduler()
        scheduler._enabled = True
        scheduler._callsign = 'NOJ'
        scheduler._frequency_khz = 4298.0
        scheduler._device = 0
        scheduler._gain = 40.0
        scheduler._ioc = 576
        scheduler._lpm = 120
        scheduler._direct_sampling = True

        sb = ScheduledBroadcast(
            station='USCG Kodiak',
            callsign='NOJ',
            frequency_khz=4298.0,
            utc_time='12:00',
            duration_min=20,
            content='Chart',
            occurrence_date='2026-01-01',
        )
        sb.status = 'scheduled'

        mock_decoder = MagicMock()
        mock_decoder.is_running = False
        mock_decoder.start.return_value = True
        mock_timer.return_value = MagicMock()

        with patch('utils.wefax_scheduler.get_wefax_decoder', return_value=mock_decoder), \
             patch('app.claim_sdr_device', return_value=None), \
             patch('app.release_sdr_device') as mock_release:
            scheduler._execute_capture_inner(sb)
            progress_cb = mock_decoder.set_callback.call_args[0][0]
            progress_cb({
                'type': 'wefax_progress',
                'status': 'error',
                'message': 'rtl_fm failed',
            })

        mock_release.assert_called_once_with(0)
        assert sb.status == 'skipped'

    def test_stop_capture_non_capturing_only_releases(self):
        """_stop_capture should be idempotent when capture already ended."""
        scheduler = WeFaxScheduler()
        sb = ScheduledBroadcast(
            station='USCG Kodiak',
            callsign='NOJ',
            frequency_khz=4298.0,
            utc_time='12:00',
            duration_min=20,
            content='Chart',
            occurrence_date='2026-01-01',
        )
        sb.status = 'complete'
        release_fn = MagicMock()

        with patch('utils.wefax_scheduler.get_wefax_decoder') as mock_get_decoder:
            scheduler._stop_capture(sb, release_fn)

        release_fn.assert_called_once()
        mock_get_decoder.assert_not_called()