Added black as pre-commit and applied it.

This commit is contained in:
Benjamin Harder
2024-09-08 18:47:57 +02:00
parent bba67f07c0
commit a926f10561
29 changed files with 1716 additions and 704 deletions

View File

@@ -1,5 +1,6 @@
import os
os.environ['IS_IN_PYTEST'] = 'true'
os.environ["IS_IN_PYTEST"] = "true"
import logging
import json
import pytest
@@ -8,45 +9,56 @@ from src.utils.shared import remove_download
from src.utils.trackers import Deleted_Downloads
# Utility function to load mock data
def load_mock_data(file_name):
with open(file_name, 'r') as file:
with open(file_name, "r") as file:
return json.load(file)
async def mock_rest_delete() -> None:
logger.debug(f"Mock rest_delete called with URL")
async def run_test(
settingsDict: Dict[str, Any],
settingsDict: Dict[str, Any],
expected_removal_messages: Set[str],
failType: str,
removeFromClient: bool,
mock_data_file: str,
monkeypatch: pytest.MonkeyPatch,
caplog: pytest.LogCaptureFixture
caplog: pytest.LogCaptureFixture,
) -> None:
# Load mock data
affectedItem = load_mock_data(mock_data_file)
# Mock the `rest_delete` function
monkeypatch.setattr('src.utils.shared.rest_delete', mock_rest_delete)
monkeypatch.setattr("src.utils.shared.rest_delete", mock_rest_delete)
# Call the function
with caplog.at_level(logging.INFO):
# Call the function and assert no exceptions
try:
deleted_downloads = Deleted_Downloads([])
await remove_download(settingsDict=settingsDict, BASE_URL='', API_KEY='', affectedItem=affectedItem, failType=failType, addToBlocklist=True, deleted_downloads=deleted_downloads, removeFromClient=removeFromClient)
try:
deleted_downloads = Deleted_Downloads([])
await remove_download(
settingsDict=settingsDict,
BASE_URL="",
API_KEY="",
affectedItem=affectedItem,
failType=failType,
addToBlocklist=True,
deleted_downloads=deleted_downloads,
removeFromClient=removeFromClient,
)
except Exception as e:
pytest.fail(f"remove_download raised an exception: {e}")
# Assertions:
# Check that expected log messages are in the captured log
log_messages = {record.message for record in caplog.records if record.levelname == 'INFO'}
log_messages = {
record.message for record in caplog.records if record.levelname == "INFO"
}
assert expected_removal_messages == log_messages
# Check that the affectedItem's downloadId was added to deleted_downloads
assert affectedItem['downloadId'] in deleted_downloads.dict
assert affectedItem["downloadId"] in deleted_downloads.dict