copilot-swe-agent[bot]
0f5e676fd9
Celery's __wrapped__ attribute strips the self parameter for bind=True tasks. Updated error-case tests to call __wrapped__(file_path, ...) instead of __wrapped__(mock_self, file_path, ...) to avoid 'multiple values' TypeError. Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
270 lines
12 KiB
Python
270 lines
12 KiB
Python
"""
|
|
Tests for app/tasks/upload_to_nextcloud.py module.
|
|
|
|
Covers the upload_to_nextcloud Celery task including configuration validation,
|
|
WebDAV upload, directory creation, and error handling.
|
|
"""
|
|
|
|
import os
|
|
from unittest.mock import MagicMock, Mock, call, patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestUploadToNextcloud:
|
|
"""Tests for upload_to_nextcloud Celery task."""
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
def test_file_not_found(self, mock_log):
|
|
"""Test that missing file raises FileNotFoundError."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
with pytest.raises(FileNotFoundError):
|
|
upload_to_nextcloud.__wrapped__("/nonexistent/file.pdf", file_id=1)
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_skipped_when_not_configured(self, mock_settings, mock_log, tmp_path):
|
|
"""Test upload skipped when Nextcloud URL not configured."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = None
|
|
mock_settings.nextcloud_username = None
|
|
mock_settings.nextcloud_password = None
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Skipped"
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_skipped_when_username_missing(self, mock_settings, mock_log, tmp_path):
|
|
"""Test upload skipped when username is missing."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav"
|
|
mock_settings.nextcloud_username = None
|
|
mock_settings.nextcloud_password = "password" # noqa: S105
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Skipped"
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_successful_upload_201(self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path):
|
|
"""Test successful file upload with 201 response."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav/"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = ""
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "test.pdf"
|
|
mock_unique.return_value = "test.pdf"
|
|
|
|
# Mock the PUT response for file upload
|
|
mock_put_response = Mock()
|
|
mock_put_response.status_code = 201
|
|
mock_requests.put.return_value = mock_put_response
|
|
|
|
# Mock PROPFIND for existence check
|
|
mock_propfind_response = Mock()
|
|
mock_propfind_response.text = ""
|
|
mock_requests.request.return_value = mock_propfind_response
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Completed"
|
|
assert result["response_code"] == 201
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_successful_upload_204(self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path):
|
|
"""Test successful file upload with 204 (No Content / overwrite) response."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav/"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = ""
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "test.pdf"
|
|
mock_unique.return_value = "test.pdf"
|
|
|
|
mock_put_response = Mock()
|
|
mock_put_response.status_code = 204
|
|
mock_requests.put.return_value = mock_put_response
|
|
|
|
mock_propfind_response = Mock()
|
|
mock_propfind_response.text = ""
|
|
mock_requests.request.return_value = mock_propfind_response
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Completed"
|
|
assert result["response_code"] == 204
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_upload_failure_status_code(
|
|
self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path
|
|
):
|
|
"""Test upload failure with non-success status code."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav/"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = ""
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "test.pdf"
|
|
mock_unique.return_value = "test.pdf"
|
|
|
|
mock_put_response = Mock()
|
|
mock_put_response.status_code = 500
|
|
mock_put_response.text = "Internal Server Error"
|
|
mock_requests.put.return_value = mock_put_response
|
|
|
|
mock_propfind_response = Mock()
|
|
mock_propfind_response.text = ""
|
|
mock_requests.request.return_value = mock_propfind_response
|
|
|
|
with pytest.raises(Exception, match="Failed to upload"):
|
|
upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_creates_parent_directories(
|
|
self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path
|
|
):
|
|
"""Test that parent directories are created via MKCOL."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav/"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = "documents"
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "documents/subfolder/test.pdf"
|
|
mock_unique.return_value = "documents/subfolder/test.pdf"
|
|
|
|
mock_put_response = Mock()
|
|
mock_put_response.status_code = 201
|
|
mock_requests.put.return_value = mock_put_response
|
|
|
|
mock_request_response = Mock()
|
|
mock_request_response.text = ""
|
|
mock_requests.request.return_value = mock_request_response
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Completed"
|
|
# Verify MKCOL calls were made for parent directories
|
|
mkcol_calls = [c for c in mock_requests.request.call_args_list if c[0][0] == "MKCOL"]
|
|
assert len(mkcol_calls) >= 1
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_connection_error(self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path):
|
|
"""Test handling of connection errors during upload."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav/"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = ""
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "test.pdf"
|
|
mock_unique.return_value = "test.pdf"
|
|
|
|
mock_requests.put.side_effect = Exception("Connection refused")
|
|
mock_requests.request.return_value = Mock(text="")
|
|
|
|
with pytest.raises(Exception, match="Failed to upload"):
|
|
upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
@patch("app.tasks.upload_to_nextcloud.get_unique_filename")
|
|
@patch("app.tasks.upload_to_nextcloud.extract_remote_path")
|
|
@patch("app.tasks.upload_to_nextcloud.requests")
|
|
@patch("app.tasks.upload_to_nextcloud.log_task_progress")
|
|
@patch("app.tasks.upload_to_nextcloud.settings")
|
|
def test_url_trailing_slash_normalization(
|
|
self, mock_settings, mock_log, mock_requests, mock_extract, mock_unique, tmp_path
|
|
):
|
|
"""Test that URLs without trailing slashes are handled."""
|
|
from app.tasks.upload_to_nextcloud import upload_to_nextcloud
|
|
|
|
mock_settings.nextcloud_upload_url = "https://nextcloud.example.com/remote.php/dav"
|
|
mock_settings.nextcloud_username = "user"
|
|
mock_settings.nextcloud_password = "pass" # noqa: S105
|
|
mock_settings.nextcloud_folder = ""
|
|
mock_settings.workdir = str(tmp_path)
|
|
mock_settings.http_request_timeout = 30
|
|
|
|
test_file = tmp_path / "test.pdf"
|
|
test_file.write_bytes(b"test content")
|
|
|
|
mock_extract.return_value = "test.pdf"
|
|
mock_unique.return_value = "test.pdf"
|
|
|
|
mock_put_response = Mock()
|
|
mock_put_response.status_code = 201
|
|
mock_requests.put.return_value = mock_put_response
|
|
|
|
mock_request_response = Mock()
|
|
mock_request_response.text = ""
|
|
mock_requests.request.return_value = mock_request_response
|
|
|
|
result = upload_to_nextcloud.apply(args=[str(test_file)], kwargs={"file_id": 1}).get()
|
|
|
|
assert result["status"] == "Completed"
|