mirror of
https://github.com/calibrain/shelfmark.git
synced 2026-09-24 20:40:27 +01:00
Alex
554f5fcbe7
- Add admin config for self-settings options visibility. Remove delivery preferences or notifications from the view. - Add option to use Booklore's Bookdrop API destination instead of a specific library - Add download path options for all torrent clients
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from shelfmark.download.outputs.booklore import BookloreConfig, booklore_upload_file
|
|
|
|
|
|
def _booklore_config(upload_to_bookdrop: bool) -> BookloreConfig:
|
|
return BookloreConfig(
|
|
base_url="http://booklore:6060",
|
|
username="admin",
|
|
password="secret",
|
|
library_id=7,
|
|
path_id=21,
|
|
upload_to_bookdrop=upload_to_bookdrop,
|
|
refresh_after_upload=not upload_to_bookdrop,
|
|
)
|
|
|
|
|
|
def test_booklore_upload_file_uses_library_endpoint_with_query_params(tmp_path):
|
|
file_path = tmp_path / "book.epub"
|
|
file_path.write_bytes(b"content")
|
|
response = MagicMock()
|
|
response.raise_for_status.return_value = None
|
|
|
|
with patch(
|
|
"shelfmark.download.outputs.booklore.requests.post",
|
|
return_value=response,
|
|
) as mock_post:
|
|
booklore_upload_file(_booklore_config(upload_to_bookdrop=False), "token", file_path)
|
|
|
|
assert mock_post.call_count == 1
|
|
args, kwargs = mock_post.call_args
|
|
assert args[0] == "http://booklore:6060/api/v1/files/upload"
|
|
assert kwargs["params"] == {"libraryId": 7, "pathId": 21}
|
|
assert kwargs["headers"] == {"Authorization": "Bearer token"}
|
|
|
|
|
|
def test_booklore_upload_file_uses_bookdrop_endpoint_without_query_params(tmp_path):
|
|
file_path = tmp_path / "book.epub"
|
|
file_path.write_bytes(b"content")
|
|
response = MagicMock()
|
|
response.raise_for_status.return_value = None
|
|
|
|
with patch(
|
|
"shelfmark.download.outputs.booklore.requests.post",
|
|
return_value=response,
|
|
) as mock_post:
|
|
booklore_upload_file(_booklore_config(upload_to_bookdrop=True), "token", file_path)
|
|
|
|
assert mock_post.call_count == 1
|
|
args, kwargs = mock_post.call_args
|
|
assert args[0] == "http://booklore:6060/api/v1/files/upload/bookdrop"
|
|
assert kwargs["params"] is None
|
|
assert kwargs["headers"] == {"Authorization": "Bearer token"}
|