"""Tests for the SABnzbd and NZBGet usenet adapters.

Pins state-mapping behavior and the queue-vs-history merge logic so
get_all returns both active and completed jobs without losing
either bucket.
"""

from __future__ import annotations

import asyncio
import base64
from unittest.mock import MagicMock, patch

import pytest

from core.usenet_clients import adapter_for_type
from core.usenet_clients.base import UsenetClientAdapter, UsenetStatus
from core.usenet_clients.nzbget import NZBGetAdapter, _map_state as nzbget_map
from core.usenet_clients.sabnzbd import SABnzbdAdapter, _map_state as sab_map


def _run(coro):
    return asyncio.new_event_loop().run_until_complete(coro)


def _mock_response(status_code: int, json_body=None):
    resp = MagicMock()
    resp.ok = 200 <= status_code < 400
    resp.status_code = status_code
    if json_body is not None:
        resp.json.return_value = json_body
    return resp


# ---------------------------------------------------------------------------
# Factory + protocol conformance
# ---------------------------------------------------------------------------


def test_adapter_for_type_returns_concrete_classes() -> None:
    assert isinstance(adapter_for_type('sabnzbd'), SABnzbdAdapter)
    assert isinstance(adapter_for_type('nzbget'), NZBGetAdapter)


def test_adapter_for_type_returns_none_for_unknown() -> None:
    assert adapter_for_type('unknown') is None


def test_adapters_conform_to_protocol() -> None:
    for adapter in (SABnzbdAdapter(), NZBGetAdapter()):
        assert isinstance(adapter, UsenetClientAdapter)


# ---------------------------------------------------------------------------
# SABnzbd
# ---------------------------------------------------------------------------


def _sab_with_config(url='http://sab:8080', api_key='k'):
    adapter = SABnzbdAdapter.__new__(SABnzbdAdapter)
    adapter._url = url.rstrip('/')
    adapter._api_key = api_key
    adapter._category = 'soulsync'
    return adapter


def test_sab_is_configured_requires_url_and_key() -> None:
    assert _sab_with_config('http://x', '').is_configured() is False
    assert _sab_with_config('', 'k').is_configured() is False
    assert _sab_with_config('http://x', 'k').is_configured() is True


def test_sab_state_mapping_covers_queue_states() -> None:
    assert sab_map('Downloading') == 'downloading'
    assert sab_map('Verifying') == 'verifying'
    assert sab_map('Repairing') == 'repairing'
    assert sab_map('Extracting') == 'extracting'
    assert sab_map('Paused') == 'paused'
    assert sab_map('Failed') == 'failed'
    # Case-insensitive — SAB sometimes returns lowercase.
    assert sab_map('downloading') == 'downloading'
    assert sab_map('') == 'error'


def test_sab_state_mapping_covers_full_sab_status_enum() -> None:
    """Every Status value SAB's sabnzbd/constants.py:Status emits must
    map to a known adapter state, NOT to the default 'error' fallback.
    Pre-fix: SAB's ``Deleted`` and ``Propagating`` were unmapped,
    fell through to the 'error' default, and the poll loop treated
    'error' as neither complete nor failed — it just spun until the
    6-hour timeout."""
    canonical = [
        'Idle', 'Queued', 'Grabbing', 'Propagating',
        'Fetching', 'Downloading', 'Paused',
        'Checking', 'QuickCheck', 'Verifying', 'Repairing',
        'Extracting', 'Moving', 'Running',
        'Completed', 'Failed', 'Deleted',
    ]
    for state in canonical:
        assert sab_map(state) != 'error', f'{state!r} fell through to error default'


def test_sab_state_mapping_propagating_routes_to_queued() -> None:
    """Propagating is SAB's pre-download delay state — semantically
    'we're waiting for the NZB to be available', map to queued so
    the poll doesn't treat it as downloading progress."""
    assert sab_map('Propagating') == 'queued'


def test_sab_state_mapping_deleted_routes_to_failed() -> None:
    """User removed the job mid-flight — terminal failure from
    SoulSync's perspective. Without this, the poll would keep
    spinning waiting for a job that's never coming back."""
    assert sab_map('Deleted') == 'failed'


def test_sab_parse_timeleft_handles_hhmmss() -> None:
    # SABnzbd's timeleft is always HH:MM:SS (or H:MM:SS).
    assert SABnzbdAdapter._parse_timeleft('01:30:00') == 5400
    assert SABnzbdAdapter._parse_timeleft('00:05:30') == 330
    assert SABnzbdAdapter._parse_timeleft('00:30:00') == 1800
    # 2-part fallback covers MM:SS for robustness.
    assert SABnzbdAdapter._parse_timeleft('05:30') == 330
    assert SABnzbdAdapter._parse_timeleft('garbage') is None
    assert SABnzbdAdapter._parse_timeleft('') is None
    assert SABnzbdAdapter._parse_timeleft(None) is None


def test_sab_parse_queue_slot_converts_mb_to_bytes() -> None:
    adapter = _sab_with_config()
    status = adapter._parse_queue_slot({
        'nzo_id': 'SABnzbd_nzo_1',
        'filename': 'Album.nzb',
        'status': 'Downloading',
        'percentage': '42',
        'mb': '100',
        'mbleft': '58',
        'timeleft': '0:01:00',
        'cat': 'soulsync',
    })
    assert status.id == 'SABnzbd_nzo_1'
    assert status.state == 'downloading'
    assert status.progress == pytest.approx(0.42)
    assert status.size == 100 * 1024 * 1024
    assert status.downloaded == 42 * 1024 * 1024
    assert status.eta == 60


def test_sab_parse_history_slot_marks_failures() -> None:
    adapter = _sab_with_config()
    failed = adapter._parse_history_slot({
        'nzo_id': 'x', 'name': 'X', 'status': 'Failed',
        'bytes': 1024, 'fail_message': 'Damaged',
    })
    assert failed.state == 'failed'
    assert failed.progress == 0.0
    assert failed.error == 'Damaged'

    success = adapter._parse_history_slot({
        'nzo_id': 'y', 'name': 'Y', 'status': 'Completed',
        'bytes': 1024, 'storage': '/done',
    })
    assert success.state == 'completed'
    assert success.progress == 1.0
    assert success.save_path == '/done'


def test_sab_history_save_path_falls_back_to_path_field() -> None:
    """Older SAB releases populated ``path`` instead of ``storage``.
    The adapter must fall through the field-name chain so we pick it
    up either way."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'z', 'name': 'Z', 'status': 'Completed',
        'bytes': 0, 'path': '/legacy/sab/path',
    })
    assert slot.save_path == '/legacy/sab/path'


def test_sab_history_save_path_falls_back_to_download_path_field() -> None:
    """Some SAB forks expose ``download_path`` instead of the
    documented ``storage``. Same fallback chain catches it."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'z2', 'name': 'Z2', 'status': 'Completed',
        'bytes': 0, 'download_path': '/fork/dl',
    })
    assert slot.save_path == '/fork/dl'


def test_sab_history_save_path_prefers_storage_when_multiple_present() -> None:
    """Field priority: ``storage`` wins over the fallbacks. The
    documented final-path key must be preferred so SAB upgrades
    don't subtly change the resolved path."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'p', 'name': 'P', 'status': 'Completed',
        'bytes': 0,
        'storage': '/final/storage',
        'path': '/legacy/path',
        'download_path': '/fork/dl',
        'dirname': '/dirname',
    })
    assert slot.save_path == '/final/storage'


def test_sab_history_save_path_none_when_all_fields_empty() -> None:
    """Regression for #721: SAB's ``storage`` field lands a few
    seconds after the job flips to History. During that window
    EVERY known path field can be empty. The adapter must return
    ``save_path=None`` (not a stale string) so
    ``poll_album_download``'s retry loop can engage and wait for
    the next poll where ``storage`` lands."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'gap', 'name': 'Forty Licks', 'status': 'Completed',
        'bytes': 0,
        # No storage / path / download_path / dirname.
    })
    assert slot.state == 'completed'
    assert slot.save_path is None


def test_sab_history_save_path_ignores_whitespace_only_values() -> None:
    """A field present but with whitespace-only content shouldn't
    fool the fallback chain — keep walking until a real path lands."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'ws', 'name': 'W', 'status': 'Completed',
        'bytes': 0,
        'storage': '   ',
        'path': '\t',
        'download_path': '/actual/path',
    })
    assert slot.save_path == '/actual/path'


def test_sab_history_save_path_ignores_incomplete_path() -> None:
    """``incomplete_path`` is SAB's in-progress staging dir before
    post-process moves files to the final ``storage``. Using it as
    a save_path fallback would bypass ``poll_album_download``'s
    retry window AND point the bundle plugin at the wrong dir —
    the in-progress staging files might be gone by the time we
    walk it, or they might be partially-extracted. Safer to return
    ``None`` so the poll retries until ``storage`` lands. Pinned
    here so a future "let's add another fallback" change doesn't
    silently re-introduce the foot-gun."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'inc', 'name': 'Inc', 'status': 'Completed',
        'bytes': 0,
        'incomplete_path': '/sab/incomplete/job',
    })
    assert slot.save_path is None


def test_sab_history_post_processing_states_are_not_terminal() -> None:
    """#721 production root cause: SAB keeps a job in History while it
    post-processes (verify / repair / unpack / move), exposing the live
    stage in ``status``. Those must map to NON-terminal states so the
    poll keeps waiting — NOT to 'completed', which would make the poll
    treat a still-extracting album as "completed but no save_path" and
    bail mid-PP (the stuck-at-99% bug). Only a real 'Completed' is
    terminal success."""
    adapter = _sab_with_config()
    for sab_status, expected in [
        ('Verifying', 'verifying'),
        ('Repairing', 'repairing'),
        ('Extracting', 'extracting'),
        ('Moving', 'extracting'),
        ('Running', 'extracting'),
        ('Queued', 'queued'),
    ]:
        slot = adapter._parse_history_slot({
            'nzo_id': 'pp', 'name': 'PP', 'status': sab_status,
            'bytes': 1000,
            # ``storage`` not written yet — PP still in flight.
        })
        assert slot.state == expected, f'{sab_status} -> {slot.state}, want {expected}'
        # Non-terminal: not 'completed', not 'failed'.
        assert slot.state not in ('completed', 'failed')
        # Download is done, so progress is full, but no final path yet.
        assert slot.progress == 1.0
        assert slot.save_path is None


def test_sab_history_completed_still_resolves_save_path() -> None:
    """The true-completion path is unchanged: status 'Completed' with a
    populated ``storage`` resolves to terminal success + final path."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'done', 'name': 'D', 'status': 'Completed',
        'bytes': 1000, 'storage': '/data/downloads/music/Album',
    })
    assert slot.state == 'completed'
    assert slot.progress == 1.0
    assert slot.save_path == '/data/downloads/music/Album'


def test_sab_history_post_processing_ignores_storage_until_completed() -> None:
    """Even if SAB has written a (possibly incomplete-dir) path field
    while still post-processing, the adapter must NOT expose it as
    save_path until the slot flips to 'Completed' — otherwise the bundle
    could stage from a half-unpacked directory."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'pp2', 'name': 'PP2', 'status': 'Extracting',
        'bytes': 1000, 'storage': '/data/downloads/music/Album',
    })
    assert slot.state == 'extracting'
    assert slot.save_path is None


def test_sab_history_surfaces_incomplete_path_separately_from_save_path() -> None:
    """``incomplete_path`` must be exposed on its OWN field, never folded
    into ``save_path``. The poll loops fall back to it only as a last
    resort after the completed-no-path window is exhausted (#721) — using
    it as save_path would bypass that window and point staging at the
    pre-move dir on every normal completion."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'gap2', 'name': 'No Storage', 'status': 'Completed',
        'bytes': 0,
        # storage / path / download_path / dirname all absent — the
        # #721 gap window — but the in-progress dir is known.
        'incomplete_path': '/sab/incomplete/No Storage',
    })
    assert slot.save_path is None
    assert slot.incomplete_path == '/sab/incomplete/No Storage'


def test_sab_history_incomplete_path_ignores_whitespace_only() -> None:
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'gap3', 'name': 'WS', 'status': 'Completed',
        'bytes': 0, 'incomplete_path': '   ',
    })
    assert slot.incomplete_path is None


def test_sab_history_prefers_storage_and_still_carries_incomplete_path() -> None:
    """When ``storage`` IS present, save_path resolves to it normally,
    and incomplete_path is carried alongside (harmless — the poll only
    consults it when save_path never lands)."""
    adapter = _sab_with_config()
    slot = adapter._parse_history_slot({
        'nzo_id': 'both', 'name': 'B', 'status': 'Completed',
        'bytes': 0, 'storage': '/final', 'incomplete_path': '/inc',
    })
    assert slot.save_path == '/final'
    assert slot.incomplete_path == '/inc'


def test_sab_add_nzb_via_url_returns_first_nzo_id() -> None:
    adapter = _sab_with_config()
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               return_value=_mock_response(200, {'status': True, 'nzo_ids': ['SABnzbd_1']})) as mock_get:
        job_id = _run(adapter.add_nzb('https://example.com/x.nzb', category='cat'))
    assert job_id == 'SABnzbd_1'
    params = mock_get.call_args.kwargs['params']
    assert params['mode'] == 'addurl'
    assert params['apikey'] == 'k'
    assert params['name'] == 'https://example.com/x.nzb'
    assert params['cat'] == 'cat'


def test_sab_add_nzb_via_bytes_uses_addfile_multipart() -> None:
    adapter = _sab_with_config()
    with patch('core.usenet_clients.sabnzbd.http_requests.post',
               return_value=_mock_response(200, {'status': True, 'nzo_ids': ['SABnzbd_2']})) as mock_post:
        job_id = _run(adapter.add_nzb(b'<nzb/>', category='cat'))
    assert job_id == 'SABnzbd_2'
    assert mock_post.call_args.kwargs['params']['mode'] == 'addfile'
    files = mock_post.call_args.kwargs['files']
    assert 'name' in files
    assert files['name'][1] == b'<nzb/>'


def test_sab_get_all_merges_queue_and_history() -> None:
    """SAB's queue and history are separate endpoints; the adapter
    must hit both so completed jobs surface in the global list."""
    adapter = _sab_with_config()
    queue_resp = _mock_response(200, {'queue': {'slots': [
        {'nzo_id': 'q1', 'filename': 'A.nzb', 'status': 'Downloading',
         'percentage': '10', 'mb': '100', 'mbleft': '90', 'timeleft': '0:01:00'},
    ]}})
    history_resp = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'h1', 'name': 'B.nzb', 'status': 'Completed', 'bytes': 1024},
    ]}})
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               side_effect=[queue_resp, history_resp]):
        statuses = adapter._get_all_sync()
    assert [s.id for s in statuses] == ['q1', 'h1']
    assert [s.state for s in statuses] == ['downloading', 'completed']


def test_sab_get_status_uses_direct_nzo_ids_lookup_against_queue() -> None:
    """Targeted nzo_ids query against queue first — avoids paging
    the full 50-entry history bulk fetch on every poll."""
    adapter = _sab_with_config()
    queue_resp = _mock_response(200, {'queue': {'slots': [
        {'nzo_id': 'target', 'filename': 'Album.nzb', 'status': 'Downloading',
         'percentage': '50', 'mb': '100', 'mbleft': '50', 'timeleft': '0:01:00'},
    ]}})
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               return_value=queue_resp) as mock_get:
        status = adapter._get_status_sync('target')
    assert status is not None
    assert status.id == 'target'
    assert status.state == 'downloading'
    # First call must include the nzo_ids filter — that's the whole
    # point of the change.
    assert mock_get.call_args.kwargs['params']['mode'] == 'queue'
    assert mock_get.call_args.kwargs['params']['nzo_ids'] == 'target'


def test_sab_get_status_falls_through_to_history_when_queue_empty() -> None:
    """Job already moved out of queue → check history with the same
    nzo_ids filter. Direct lookup means SoulSync doesn't lose the
    job on a busy SAB where it's rolled past the bulk history limit."""
    adapter = _sab_with_config()
    empty_queue = _mock_response(200, {'queue': {'slots': []}})
    history_resp = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'target', 'name': 'Album.nzb', 'status': 'Completed',
         'bytes': 1024, 'storage': '/done/Album'},
    ]}})
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               side_effect=[empty_queue, history_resp]) as mock_get:
        status = adapter._get_status_sync('target')
    assert status is not None
    assert status.id == 'target'
    assert status.state == 'completed'
    assert status.save_path == '/done/Album'
    # Second call must hit the history endpoint, also filtered by id.
    second_params = mock_get.call_args_list[1].kwargs['params']
    assert second_params['mode'] == 'history'
    assert second_params['nzo_ids'] == 'target'


def test_sab_get_status_returns_none_when_neither_endpoint_finds_id() -> None:
    """Mid SAB queue→history transition window: the slot is gone
    from the queue but not yet in history. Direct lookup returns
    None — the poll layer treats this as a transient miss and
    retries, NOT as a terminal failure. Pre-fix this was the most
    likely trigger for the user's stuck-at-downloading bug (#706).

    Bulk fallback also returns nothing (both endpoints reported
    empty); ``_get_status_sync`` returns None rather than raising."""
    adapter = _sab_with_config()
    empty_queue = _mock_response(200, {'queue': {'slots': []}})
    empty_history = _mock_response(200, {'history': {'slots': []}})
    # Three calls: direct queue, direct history, bulk fallback queue
    # + bulk fallback history. Empty for all four.
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               side_effect=[empty_queue, empty_history, empty_queue, empty_history]):
        status = adapter._get_status_sync('target')
    assert status is None


def test_sab_get_status_empty_job_id_returns_none_without_hitting_api() -> None:
    """Defensive — an empty job_id from upstream shouldn't fire
    HTTP queries we know will be wrong."""
    adapter = _sab_with_config()
    with patch('core.usenet_clients.sabnzbd.http_requests.get') as mock_get:
        assert adapter._get_status_sync('') is None
        mock_get.assert_not_called()


def test_sab_poll_recovers_after_queue_to_history_handoff_gap() -> None:
    """Integration test: simulate the SAB queue→history transition
    window end-to-end through the adapter. Sequence: 3 polls where
    SAB has moved the slot out of queue but hasn't added it to
    history yet (both endpoints return empty), followed by the slot
    appearing in history as Completed with a save_path. Pre-fix,
    the first None read on the SAB side surfaced to the poll layer
    as 'disappeared' → terminal failure even though SAB was healthy
    and just mid-handoff. Post-fix the adapter still returns None
    during the gap, but the poll helper's TransientMissCounter
    absorbs the gap and recovers when the history entry appears."""
    from core.download_plugins.album_bundle import poll_album_download

    adapter = _sab_with_config()
    empty_queue = _mock_response(200, {'queue': {'slots': []}})
    empty_history = _mock_response(200, {'history': {'slots': []}})
    final_history = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'target', 'name': 'Album.nzb', 'status': 'Completed',
         'bytes': 1024, 'storage': '/done/Album'},
    ]}})

    # Each _get_status_sync call hits two endpoints (queue + history).
    # Three gap polls + one recovery poll = 4 * 2 = 8 HTTP calls.
    # On recovery the queue is still empty but the history finds the job.
    poll_results = [
        empty_queue, empty_history,   # gap poll 1
        empty_queue, empty_history,   # gap poll 2
        empty_queue, empty_history,   # gap poll 3
        empty_queue, final_history,   # recovery
    ]

    class _Clock:
        def __init__(self): self.now = 0.0
        def monotonic(self): return self.now
        def sleep(self, s): self.now += s

    clock = _Clock()
    emits: list = []
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               side_effect=poll_results):
        result = poll_album_download(
            get_status=lambda: adapter._get_status_sync('target'),
            title='Linkin Park - From Zero',
            emit=lambda state, **kw: emits.append((state, kw)),
            complete_states=frozenset(['completed']),
            failed_states=frozenset(['failed']),
            transient_miss_threshold=5,
            sleep=clock.sleep, monotonic=clock.monotonic,
            poll_interval=2.0, timeout=60.0,
        )

    assert result == '/done/Album'
    # Terminal failure must NOT have been emitted — the gap was transient.
    assert 'failed' not in [e[0] for e in emits]


def test_sab_poll_waits_through_history_post_processing_then_completes() -> None:
    """Integration regression for the #721 PRODUCTION root cause
    (David Bowie - Hunky Dory, 1.7 GB FLAC, stuck at 99%).

    SAB moves a finished download into History and runs par2 verify +
    unpack THERE, exposing the live stage in ``status`` while ``storage``
    stays empty until the final move. For a 1.7 GB FLAC album that
    post-processing window is far longer than the ~10s completed-no-path
    budget. Pre-fix the adapter mapped 'Extracting' → 'completed', so the
    poll saw "completed but no save_path" the instant download finished,
    burned its budget mid-PP, and bailed — freezing the UI on the last
    'downloading' emit (99%). SAB then finished fine, which is why the
    job shows Completed in History but SoulSync never staged it.

    Post-fix the in-PP History slots map to NON-terminal states, so the
    poll just keeps waiting (as 'downloading') until SAB flips the slot
    to 'Completed' with a real ``storage`` path — no matter how long PP
    takes — then returns it."""
    from core.download_plugins.album_bundle import poll_album_download

    adapter = _sab_with_config()
    downloading = _mock_response(200, {'queue': {'slots': [
        {'nzo_id': 'hd', 'filename': 'Hunky Dory', 'status': 'Downloading',
         'percentage': '99', 'mb': '1700', 'mbleft': '17', 'timeleft': '0:00:05'},
    ]}})
    empty_queue = _mock_response(200, {'queue': {'slots': []}})
    # Long post-processing window, reported via HISTORY (download already
    # left the queue). storage NOT yet written.
    pp_verify = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'hd', 'name': 'Hunky Dory', 'status': 'Verifying', 'bytes': 1_700_000_000},
    ]}})
    pp_extract = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'hd', 'name': 'Hunky Dory', 'status': 'Extracting', 'bytes': 1_700_000_000},
    ]}})
    done = _mock_response(200, {'history': {'slots': [
        {'nzo_id': 'hd', 'name': 'Hunky Dory', 'status': 'Completed',
         'bytes': 1_700_000_000,
         'storage': '/data/downloads/music/David.Bowie-Hunky.Dory'},
    ]}})

    # Each poll = queue call + (if empty) history call.
    poll_results = [
        downloading,                 # poll 1: queue has it @99%  (1 call)
        empty_queue, pp_verify,      # poll 2: PP verifying in history
        empty_queue, pp_verify,      # poll 3: still verifying
        empty_queue, pp_extract,     # poll 4: extracting
        empty_queue, pp_extract,     # poll 5: still extracting
        empty_queue, pp_extract,     # poll 6: still extracting (past old 5-poll budget)
        empty_queue, pp_extract,     # poll 7
        empty_queue, done,           # poll 8: completed + storage
    ]

    class _Clock:
        def __init__(self): self.now = 0.0
        def monotonic(self): return self.now
        def sleep(self, s): self.now += s

    clock = _Clock()
    emits: list = []
    with patch('core.usenet_clients.sabnzbd.http_requests.get',
               side_effect=poll_results):
        result = poll_album_download(
            get_status=lambda: adapter._get_status_sync('hd'),
            title='David Bowie - Hunky Dory',
            emit=lambda state, **kw: emits.append((state, kw)),
            complete_states=frozenset(['completed']),
            failed_states=frozenset(['failed']),
            transient_miss_threshold=5,
            # Short no-path budget proves PP is NOT consuming it anymore.
            completed_no_path_threshold=2,
            sleep=clock.sleep, monotonic=clock.monotonic,
            poll_interval=2.0, timeout=600.0,
        )

    assert result == '/data/downloads/music/David.Bowie-Hunky.Dory'
    # Never failed despite PP outlasting both the miss budget AND the
    # (deliberately tiny) completed-no-path budget — PP is non-terminal.
    assert 'failed' not in [e[0] for e in emits]


# ---------------------------------------------------------------------------
# NZBGet
# ---------------------------------------------------------------------------


def _nzbget_with_config(url='http://nzbget:6789', username='u', password='p'):
    adapter = NZBGetAdapter.__new__(NZBGetAdapter)
    from itertools import count
    adapter._id_counter = count(1)
    adapter._url = url.rstrip('/')
    adapter._username = username
    adapter._password = password
    adapter._category = 'soulsync'
    return adapter


def test_nzbget_is_configured_requires_all_three() -> None:
    assert _nzbget_with_config('', 'u', 'p').is_configured() is False
    assert _nzbget_with_config('http://x', '', 'p').is_configured() is False
    assert _nzbget_with_config('http://x', 'u', '').is_configured() is False
    assert _nzbget_with_config('http://x', 'u', 'p').is_configured() is True


def test_nzbget_state_mapping_covers_post_process_phases() -> None:
    assert nzbget_map('DOWNLOADING') == 'downloading'
    assert nzbget_map('PAUSED') == 'paused'
    assert nzbget_map('LOADING_PARS') == 'verifying'
    assert nzbget_map('REPAIRING') == 'repairing'
    assert nzbget_map('UNPACKING') == 'extracting'
    assert nzbget_map('PP_FINISHED') == 'completed'
    assert nzbget_map('') == 'error'


def test_nzbget_mb_value_prefers_64bit_split() -> None:
    """NZBGet ships size as FileSizeHi << 32 | FileSizeLo for clients
    that need precision past 2 GB. Prefer that over the legacy MB
    field when both are present."""
    val = NZBGetAdapter._mb_value({'FileSizeLo': 1024 * 1024, 'FileSizeHi': 0, 'FileSizeMB': 999}, 'FileSize')
    assert val == 1.0


def test_nzbget_mb_value_falls_back_to_mb() -> None:
    val = NZBGetAdapter._mb_value({'FileSizeMB': 500}, 'FileSize')
    assert val == 500.0


def test_nzbget_add_nzb_url_passes_through_unchanged() -> None:
    adapter = _nzbget_with_config()
    captured = {}

    def fake_post(url, json=None, auth=None, headers=None, timeout=None):
        captured['payload'] = json
        return _mock_response(200, {'result': 42})

    with patch('core.usenet_clients.nzbget.http_requests.post', side_effect=fake_post):
        job_id = _run(adapter.add_nzb('https://x/x.nzb', category='cat'))

    assert job_id == '42'
    params = captured['payload']['params']
    assert params[0] == ''  # NZBFilename empty when content is a URL
    assert params[1] == 'https://x/x.nzb'
    assert params[2] == 'cat'


def test_nzbget_add_nzb_bytes_base64_encodes() -> None:
    adapter = _nzbget_with_config()
    captured = {}

    def fake_post(url, json=None, auth=None, headers=None, timeout=None):
        captured['payload'] = json
        return _mock_response(200, {'result': 7})

    with patch('core.usenet_clients.nzbget.http_requests.post', side_effect=fake_post):
        _run(adapter.add_nzb(b'<nzb/>', category='cat'))

    params = captured['payload']['params']
    assert params[0] == 'soulsync.nzb'
    assert params[1] == base64.b64encode(b'<nzb/>').decode('ascii')


def test_nzbget_remove_uses_groupfinal_when_deleting_files() -> None:
    """``GroupFinalDelete`` deletes downloaded data on disk;
    ``GroupDelete`` just removes the queue entry. The adapter must
    pick the right one based on the ``delete_files`` flag."""
    adapter = _nzbget_with_config()
    with patch.object(adapter, '_rpc_sync', return_value=True) as mock_rpc:
        adapter._remove_sync('42', delete_files=True)
        adapter._remove_sync('42', delete_files=False)
    cmds = [c.args[1][0] for c in mock_rpc.call_args_list]
    assert cmds == ['GroupFinalDelete', 'GroupDelete']


def test_nzbget_parse_group_computes_progress() -> None:
    adapter = _nzbget_with_config()
    status = adapter._parse_group({
        'NZBID': 99,
        'NZBName': 'Album.nzb',
        'Status': 'DOWNLOADING',
        'FileSizeLo': 200 * 1024 * 1024, 'FileSizeHi': 0,
        'RemainingSizeLo': 100 * 1024 * 1024, 'RemainingSizeHi': 0,
        'DownloadRate': 500_000,
        'DestDir': '/incomplete',
        'Category': 'soulsync',
    })
    assert status.id == '99'
    assert status.state == 'downloading'
    assert status.size == 200 * 1024 * 1024
    assert status.downloaded == 100 * 1024 * 1024
    assert status.progress == pytest.approx(0.5)
    assert status.download_speed == 500_000


def test_nzbget_queue_group_never_offers_a_save_path() -> None:
    """A queued group's DestDir is the in-progress '….#NZBID' dir, which is gone
    after the move — never expose it as a final save_path. Finalisation must come
    from the HISTORY entry. (Swigs: imported from /…/incomplete/….#2141.)"""
    adapter = _nzbget_with_config()
    status = adapter._parse_group({
        'NZBID': 2141, 'NZBName': 'xRepentancex-The.Sickness.Of.Eden',
        'Status': 'DOWNLOADING',
        'DestDir': '/data/usenet/incomplete/xRepentancex-The.Sickness.Of.Eden.#2141',
        'Category': 'soulsync',
    })
    assert status.save_path is None


def test_nzbget_pp_finished_group_is_completed_but_has_no_path() -> None:
    """PP_FINISHED maps to 'completed' but is still a QUEUE group on the
    in-progress dir — it must report no save_path so the plugin waits for the
    history entry instead of finalising on the incomplete folder."""
    adapter = _nzbget_with_config()
    status = adapter._parse_group({
        'NZBID': 2141, 'NZBName': 'Album', 'Status': 'PP_FINISHED',
        'DestDir': '/data/usenet/incomplete/Album.#2141', 'Category': 'soulsync',
    })
    assert status.state == 'completed'
    assert status.save_path is None


def test_nzbget_history_prefers_finaldir_over_destdir() -> None:
    """After a post-processing move, FinalDir is the real location; DestDir can
    still be the intermediate dir. Swigs' exact case."""
    adapter = _nzbget_with_config()
    status = adapter._parse_history({
        'NZBID': 2141, 'Name': 'xRepentancex-The.Sickness.Of.Eden',
        'Status': 'SUCCESS/ALL',
        'DestDir': '/data/usenet/incomplete/xRepentancex-The.Sickness.Of.Eden.#2141',
        'FinalDir': '/data/soulseek/xRepentancex-The.Sickness.Of.Eden-CD-FLAC-2015-CATARACT',
        'Category': 'soulsync',
    })
    assert status.state == 'completed'
    assert status.save_path == '/data/soulseek/xRepentancex-The.Sickness.Of.Eden-CD-FLAC-2015-CATARACT'


def test_nzbget_history_falls_back_to_destdir_when_no_finaldir() -> None:
    """No PP move -> FinalDir empty -> use DestDir (the final dest in that case)."""
    adapter = _nzbget_with_config()
    status = adapter._parse_history({
        'NZBID': 7, 'Name': 'Album', 'Status': 'SUCCESS/HEALTH',
        'DestDir': '/data/usenet/completed/Album', 'FinalDir': '', 'Category': 'c',
    })
    assert status.save_path == '/data/usenet/completed/Album'


def test_nzbget_history_empty_dirs_yield_none() -> None:
    adapter = _nzbget_with_config()
    status = adapter._parse_history({
        'NZBID': 7, 'Name': 'Album', 'Status': 'SUCCESS/ALL',
        'DestDir': '   ', 'FinalDir': '', 'Category': 'c',
    })
    assert status.save_path is None


def test_nzbget_remove_rejects_non_numeric_id() -> None:
    """NZBGet IDs are ints; passing a string id like 'abc' must
    fail fast instead of corrupting the editqueue call."""
    adapter = _nzbget_with_config()
    with patch.object(adapter, '_rpc_sync') as mock_rpc:
        assert adapter._remove_sync('not-a-number', delete_files=False) is False
    mock_rpc.assert_not_called()