jimaku-dl/tests/conftest.py

"""Global pytest fixtures for jimaku-dl tests."""

import os
import sys
import tempfile
from pathlib import Path
from unittest.mock import MagicMock, patch

import pytest

# Add the src directory to the Python path
project_root = Path(__file__).parent.parent
src_path = project_root / "src"
sys.path.insert(0, str(src_path))


@pytest.fixture
def temp_dir():
    """Create a temporary directory for test files."""
    with tempfile.TemporaryDirectory() as tmpdirname:
        yield tmpdirname


@pytest.fixture
def mock_anilist_response():
    """Mock response from AniList API."""
    return {
        "data": {
            "Media": {
                "id": 123456,
                "title": {
                    "romaji": "Test Anime",
                    "english": "Test Anime English",
                    "native": "テストアニメ",
                },
                "synonyms": ["Test Show"],
            }
        }
    }


@pytest.fixture
def mock_jimaku_entries_response():
    """Mock response from Jimaku entries endpoint."""
    return [
        {
            "id": 1,
            "english_name": "Test Anime",
            "japanese_name": "テストアニメ",
            "anilist_id": 123456,
        }
    ]


@pytest.fixture
def mock_jimaku_files_response():
    """Mock response from Jimaku files endpoint."""
    return [
        {
            "id": 101,
            "name": "Test Anime - 01.srt",
            "url": "https://jimaku.cc/api/files/101",
        },
        {
            "id": 102,
            "name": "Test Anime - 02.srt",
            "url": "https://jimaku.cc/api/files/102",
        },
    ]


@pytest.fixture
def mock_requests(
    monkeypatch,
    mock_anilist_response,
    mock_jimaku_entries_response,
    mock_jimaku_files_response,
):
    """Mock requests module for API calls."""
    mock_response = MagicMock()
    mock_response.raise_for_status = MagicMock()
    mock_response.json = MagicMock()

    def mock_requests_post(url, **kwargs):
        if "anilist.co" in url:
            mock_response.json.return_value = mock_anilist_response
        return mock_response

    def mock_requests_get(url, **kwargs):
        if "entries/search" in url:
            mock_response.json.return_value = mock_jimaku_entries_response
        elif "entries/" in url and "/files" in url:
            mock_response.json.return_value = mock_jimaku_files_response
        return mock_response

    # Patch both the direct imports used in downloader.py and the regular requests module
    monkeypatch.setattr("requests.post", mock_requests_post)
    monkeypatch.setattr("requests.get", mock_requests_get)
    monkeypatch.setattr("jimaku_dl.downloader.requests_post", mock_requests_post)
    monkeypatch.setattr("jimaku_dl.downloader.requests_get", mock_requests_get)

    return {
        "post": mock_requests_post,
        "get": mock_requests_get,
        "response": mock_response,
    }


@pytest.fixture
def mock_subprocess(monkeypatch):
    """Mock subprocess module for fzf and mpv calls."""
    mock_run = MagicMock()
    mock_result = MagicMock()
    mock_result.stdout = "1. Test Selection"
    mock_run.return_value = mock_result

    monkeypatch.setattr("subprocess.run", mock_run)
    return mock_run


@pytest.fixture
def sample_video_file(temp_dir):
    """Create a sample video file."""
    file_path = os.path.join(temp_dir, "Test Anime S01E01 [1080p].mkv")
    with open(file_path, "wb") as f:
        f.write(b"dummy video content")
    return file_path


@pytest.fixture
def sample_anime_directory(temp_dir):
    """Create a sample directory structure for anime."""
    # Main directory
    anime_dir = os.path.join(temp_dir, "Test Anime")
    os.makedirs(anime_dir)

    # Season subdirectory
    season_dir = os.path.join(anime_dir, "Season-1")
    os.makedirs(season_dir)

    # Episode files
    for i in range(1, 3):
        file_path = os.path.join(season_dir, f"Test Anime S01E0{i} [1080p].mkv")
        with open(file_path, "wb") as f:
            f.write(b"dummy video content")

    return anime_dir