decluttarr/tests/jobs/test_remove_metadata_missing.py

import pytest

from src.jobs.remove_metadata_missing import RemoveMetadataMissing
from tests.jobs.utils import shared_fix_affected_items, shared_test_affected_items


# Test to check if items with the specific error message are included in affected items with parameterized data
@pytest.mark.asyncio
@pytest.mark.parametrize(
    ("queue_data", "expected_download_ids"),
    [
        (
            [
                {
                    "downloadId": "1",
                    "status": "queued",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Valid item
                {
                    "downloadId": "2",
                    "status": "completed",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Wrong status
                {
                    "downloadId": "3",
                    "status": "queued",
                    "errorMessage": "Some other error",
                },  # Incorrect errorMessage
            ],
            [
                "1"
            ],  # Only the item with "queued" status and the correct errorMessage should be affected
        ),
        (
            [
                {
                    "downloadId": "1",
                    "status": "queued",
                    "errorMessage": "Some other error",
                },  # Incorrect errorMessage
                {
                    "downloadId": "2",
                    "status": "completed",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Wrong status
                {
                    "downloadId": "3",
                    "status": "queued",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Correct item
            ],
            [
                "3"
            ],  # Only the item with "queued" status and the correct errorMessage should be affected
        ),
        (
            [
                {
                    "downloadId": "1",
                    "status": "queued",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Valid item
                {
                    "downloadId": "2",
                    "status": "queued",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Another valid item
            ],
            ["1", "2"],  # Both items match the condition
        ),
        (
            [
                {
                    "downloadId": "1",
                    "status": "completed",
                    "errorMessage": "qBittorrent is downloading metadata",
                },  # Wrong status
                {
                    "downloadId": "2",
                    "status": "queued",
                    "errorMessage": "Some other error",
                },  # Incorrect errorMessage
            ],
            [],  # No items match the condition
        ),
    ],
)
async def test_find_affected_items(queue_data, expected_download_ids):
    # Arrange
    removal_job = shared_fix_affected_items(RemoveMetadataMissing, queue_data)

    # Act and Assert
    await shared_test_affected_items(removal_job, expected_download_ids)