Harvey
110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import httpx
|
|
|
|
from core.exceptions import ExternalServiceError, PlaybackNotAllowedError
|
|
from infrastructure.constants import JELLYFIN_TICKS_PER_SECOND
|
|
from services.jellyfin_playback_service import (
|
|
JellyfinPlaybackService,
|
|
)
|
|
|
|
|
|
def _make_repo() -> AsyncMock:
|
|
repo = AsyncMock()
|
|
repo.get_playback_info = AsyncMock(
|
|
return_value={"PlaySessionId": "sess-123"}
|
|
)
|
|
repo.report_playback_start = AsyncMock()
|
|
repo.report_playback_progress = AsyncMock()
|
|
repo.report_playback_stopped = AsyncMock()
|
|
return repo
|
|
|
|
|
|
@pytest.fixture
|
|
def service():
|
|
repo = _make_repo()
|
|
svc = JellyfinPlaybackService(jellyfin_repo=repo)
|
|
return svc, repo
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_playback_returns_session_id(service):
|
|
svc, repo = service
|
|
result = await svc.start_playback("item-1")
|
|
assert result == "sess-123"
|
|
repo.report_playback_start.assert_called_once_with("item-1", "sess-123")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_playback_raises_on_error_code(service):
|
|
svc, repo = service
|
|
repo.get_playback_info.return_value = {"ErrorCode": "NotAllowed"}
|
|
|
|
with pytest.raises(PlaybackNotAllowedError, match="NotAllowed"):
|
|
await svc.start_playback("item-1")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_playback_handles_null_session_id(service):
|
|
svc, repo = service
|
|
repo.get_playback_info.return_value = {"PlaySessionId": None}
|
|
|
|
result = await svc.start_playback("item-1")
|
|
assert result == ""
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_progress_sends_to_jellyfin(service):
|
|
svc, repo = service
|
|
await svc.report_progress("item-1", "sess-123", 5.0, False)
|
|
expected_ticks = int(5.0 * JELLYFIN_TICKS_PER_SECOND)
|
|
repo.report_playback_progress.assert_called_once_with(
|
|
"item-1", "sess-123", expected_ticks, False
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_progress_skips_empty_session(service):
|
|
svc, repo = service
|
|
await svc.report_progress("item-1", "", 5.0, False)
|
|
repo.report_playback_progress.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_progress_handles_http_failure(service):
|
|
svc, repo = service
|
|
repo.report_playback_progress.side_effect = httpx.ConnectError("network error")
|
|
await svc.report_progress("item-1", "sess-123", 5.0, False)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_report_progress_handles_external_service_failure(service):
|
|
svc, repo = service
|
|
repo.report_playback_progress.side_effect = ExternalServiceError("server error")
|
|
await svc.report_progress("item-1", "sess-123", 5.0, False)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stop_playback_sends_to_jellyfin(service):
|
|
svc, repo = service
|
|
await svc.stop_playback("item-1", "sess-123", 10.0)
|
|
expected_ticks = int(10.0 * JELLYFIN_TICKS_PER_SECOND)
|
|
repo.report_playback_stopped.assert_called_once_with(
|
|
"item-1", "sess-123", expected_ticks
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stop_playback_skips_empty_session(service):
|
|
svc, repo = service
|
|
await svc.stop_playback("item-1", "", 10.0)
|
|
repo.report_playback_stopped.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stop_playback_handles_failure(service):
|
|
svc, repo = service
|
|
repo.report_playback_stopped.side_effect = httpx.ConnectError("timeout")
|
|
await svc.stop_playback("item-1", "sess-123", 10.0)
|