decluttarr/tests/jobs/test_remove_stalled.py

import pytest

from tests.jobs.utils import shared_fix_affected_items, shared_test_affected_items
from src.jobs.remove_stalled import RemoveStalled


# 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": "warning", "errorMessage": "The download is stalled with no connections"},  # Valid item
                {"downloadId": "2", "status": "completed", "errorMessage": "The download is stalled with no connections"},  # Wrong status
                {"downloadId": "3", "status": "warning", "errorMessage": "Some other error"},  # Incorrect errorMessage
            ],
            ["1"],  # Only the item with "warning" status and the correct errorMessage should be affected
        ),
        (
            [
                {"downloadId": "1", "status": "warning", "errorMessage": "Some other error"},  # Incorrect errorMessage
                {"downloadId": "2", "status": "completed", "errorMessage": "The download is stalled with no connections"},  # Wrong status
                {"downloadId": "3", "status": "warning", "errorMessage": "The download is stalled with no connections"},  # Correct item
            ],
            ["3"],  # Only the item with "warning" status and the correct errorMessage should be affected
        ),
        (
            [
                {"downloadId": "1", "status": "warning", "errorMessage": "The download is stalled with no connections"},  # Valid item
                {"downloadId": "2", "status": "warning", "errorMessage": "The download is stalled with no connections"},  # Another valid item
            ],
            ["1", "2"],  # Both items match the condition
        ),
        (
            [
                {"downloadId": "1", "status": "completed", "errorMessage": "The download is stalled with no connections"},  # Wrong status
                {"downloadId": "2", "status": "warning", "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(RemoveStalled, queue_data)

    # Act and Assert
    await shared_test_affected_items(removal_job, expected_download_ids)