mirror of
https://github.com/ManiMatter/decluttarr.git
synced 2026-04-18 19:54:07 +02:00
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from src.jobs.removal_handler import RemovalHandler
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"qbittorrent_configured, is_private, client_type, protocol, expected",
|
|
[
|
|
(True, True, "qbittorrent", "torrent", "private_handling"),
|
|
(True, False, "qbittorrent", "torrent", "public_handling"),
|
|
(False, True, "qbittorrent", "torrent", "remove"),
|
|
(False, False, "qbittorrent", "torrent", "remove"),
|
|
(True, False, "transmission", "torrent", "remove"), # unsupported client
|
|
(True, False, "myusenetclient", "usenet", "remove"), # unsupported protocol
|
|
],
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_get_handling_method(
|
|
qbittorrent_configured,
|
|
is_private,
|
|
client_type,
|
|
protocol,
|
|
expected,
|
|
):
|
|
# Mock arr
|
|
arr = AsyncMock()
|
|
arr.tracker.private = ["A"] if is_private else []
|
|
|
|
# Mock settings and get_download_client_by_name
|
|
settings = MagicMock()
|
|
settings.download_clients.qbittorrent = ["dummy"] if qbittorrent_configured else []
|
|
|
|
# Simulate (client_name, client_type) return
|
|
settings.download_clients.get_download_client_by_name.return_value = (
|
|
"client_name",
|
|
client_type,
|
|
)
|
|
|
|
settings.general.private_tracker_handling = "private_handling"
|
|
settings.general.public_tracker_handling = "public_handling"
|
|
|
|
handler = RemovalHandler(arr=arr, settings=settings, job_name="test")
|
|
|
|
affected_download = {
|
|
"downloadClient": "qBittorrent",
|
|
"protocol": protocol,
|
|
}
|
|
|
|
result = await handler._get_handling_method( # pylint: disable=W0212
|
|
"A", affected_download
|
|
)
|
|
assert result == expected
|