gitea-admin/gh-christianlouis-docuelevate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Apply Black formatting to test file

Browse Source 
Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-02-07 17:54:29 +00:00
parent 77678f6368
commit 4c8ec85bfa
1 changed files with 70 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_bulk_operations.py
+39 -46
View File
@@ -1,6 +1,7 @@
""" """
Tests for bulk file operations (delete and reprocess). Tests for bulk file operations (delete and reprocess).
""" """
import pytest import pytest
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from app.models import FileRecord, ProcessingLog from app.models import FileRecord, ProcessingLog
@@ -20,7 +21,7 @@ class TestSingleFileOperations:
original_filename="test.pdf", original_filename="test.pdf",
local_filename="/tmp/test.pdf", local_filename="/tmp/test.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.commit() db_session.commit()
@@ -34,7 +35,9 @@ class TestSingleFileOperations:
assert f"File record {file_id} deleted successfully" in data["message"] assert f"File record {file_id} deleted successfully" in data["message"]
# Verify file is deleted # Verify file is deleted
file_record = db_session.query(FileRecord).filter(FileRecord.id == file_id).first() file_record = (
db_session.query(FileRecord).filter(FileRecord.id == file_id).first()
)
assert file_record is None assert file_record is None
def test_single_file_delete_nonexistent(self, client: TestClient, db_session): def test_single_file_delete_nonexistent(self, client: TestClient, db_session):
@@ -60,7 +63,7 @@ class TestBulkOperations:
original_filename=f"test{i}.pdf", original_filename=f"test{i}.pdf",
local_filename=f"/tmp/test{i}.pdf", local_filename=f"/tmp/test{i}.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -68,10 +71,7 @@ class TestBulkOperations:
db_session.commit() db_session.commit()
# Bulk delete # Bulk delete
response = client.post( response = client.post("/api/files/bulk-delete", json=file_ids)
"/api/files/bulk-delete",
json=file_ids
)
assert response.status_code == 200 assert response.status_code == 200
data = response.json() data = response.json()
assert data["status"] == "success" assert data["status"] == "success"
@@ -79,29 +79,27 @@ class TestBulkOperations:
# Verify files are deleted # Verify files are deleted
for file_id in file_ids: for file_id in file_ids:
file_record = db_session.query(FileRecord).filter(FileRecord.id == file_id).first() file_record = (
db_session.query(FileRecord).filter(FileRecord.id == file_id).first()
)
assert file_record is None assert file_record is None
def test_bulk_delete_empty_list(self, client: TestClient, db_session): def test_bulk_delete_empty_list(self, client: TestClient, db_session):
"""Test bulk deletion with empty list.""" """Test bulk deletion with empty list."""
response = client.post( response = client.post("/api/files/bulk-delete", json=[])
"/api/files/bulk-delete",
json=[]
)
assert response.status_code == 404 assert response.status_code == 404
data = response.json() data = response.json()
assert "No files found" in data["detail"] assert "No files found" in data["detail"]
def test_bulk_delete_nonexistent_files(self, client: TestClient, db_session): def test_bulk_delete_nonexistent_files(self, client: TestClient, db_session):
"""Test bulk deletion of non-existent files.""" """Test bulk deletion of non-existent files."""
response = client.post( response = client.post("/api/files/bulk-delete", json=[9999, 9998])
"/api/files/bulk-delete",
json=[9999, 9998]
)
assert response.status_code == 404 assert response.status_code == 404
@patch('app.api.files.process_document') @patch("app.api.files.process_document")
def test_bulk_reprocess_success(self, mock_process_document, client: TestClient, db_session): def test_bulk_reprocess_success(
self, mock_process_document, client: TestClient, db_session
):
"""Test bulk reprocessing of files.""" """Test bulk reprocessing of files."""
# Setup mock # Setup mock
mock_task = MagicMock() mock_task = MagicMock()
@@ -116,7 +114,7 @@ class TestBulkOperations:
original_filename=f"test{i}.pdf", original_filename=f"test{i}.pdf",
local_filename=f"/tmp/test{i}.pdf", local_filename=f"/tmp/test{i}.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -124,11 +122,8 @@ class TestBulkOperations:
db_session.commit() db_session.commit()
# Mock os.path.exists to return True # Mock os.path.exists to return True
with patch('os.path.exists', return_value=True): with patch("os.path.exists", return_value=True):
response = client.post( response = client.post("/api/files/bulk-reprocess", json=file_ids)
"/api/files/bulk-reprocess",
json=file_ids
)
assert response.status_code == 200 assert response.status_code == 200
data = response.json() data = response.json()
@@ -136,8 +131,10 @@ class TestBulkOperations:
assert len(data["processed_files"]) == 2 assert len(data["processed_files"]) == 2
assert len(data["task_ids"]) == 2 assert len(data["task_ids"]) == 2
@patch('app.api.files.process_document') @patch("app.api.files.process_document")
def test_bulk_reprocess_missing_files(self, mock_process_document, client: TestClient, db_session): def test_bulk_reprocess_missing_files(
self, mock_process_document, client: TestClient, db_session
):
"""Test bulk reprocessing when some local files are missing.""" """Test bulk reprocessing when some local files are missing."""
# Setup mock # Setup mock
mock_task = MagicMock() mock_task = MagicMock()
@@ -150,9 +147,11 @@ class TestBulkOperations:
file_record = FileRecord( file_record = FileRecord(
filehash=f"hash{i}", filehash=f"hash{i}",
original_filename=f"test{i}.pdf", original_filename=f"test{i}.pdf",
local_filename=f"/tmp/test{i}.pdf" if i == 0 else None, # Second file has no local file local_filename=(
f"/tmp/test{i}.pdf" if i == 0 else None
), # Second file has no local file
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -160,11 +159,8 @@ class TestBulkOperations:
db_session.commit() db_session.commit()
# Mock os.path.exists to return False for missing file # Mock os.path.exists to return False for missing file
with patch('os.path.exists', return_value=False): with patch("os.path.exists", return_value=False):
response = client.post( response = client.post("/api/files/bulk-reprocess", json=file_ids)
"/api/files/bulk-reprocess",
json=file_ids
)
assert response.status_code == 200 assert response.status_code == 200
data = response.json() data = response.json()
@@ -174,10 +170,7 @@ class TestBulkOperations:
def test_bulk_reprocess_nonexistent_files(self, client: TestClient, db_session): def test_bulk_reprocess_nonexistent_files(self, client: TestClient, db_session):
"""Test bulk reprocessing of non-existent files.""" """Test bulk reprocessing of non-existent files."""
response = client.post( response = client.post("/api/files/bulk-reprocess", json=[9999, 9998])
"/api/files/bulk-reprocess",
json=[9999, 9998]
)
assert response.status_code == 404 assert response.status_code == 404
@@ -195,7 +188,7 @@ class TestStatusFilter:
original_filename="pending.pdf", original_filename="pending.pdf",
local_filename="/tmp/pending.pdf", local_filename="/tmp/pending.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file1) db_session.add(file1)
@@ -205,7 +198,7 @@ class TestStatusFilter:
original_filename="processing.pdf", original_filename="processing.pdf",
local_filename="/tmp/processing.pdf", local_filename="/tmp/processing.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file2) db_session.add(file2)
db_session.flush() db_session.flush()
@@ -215,7 +208,7 @@ class TestStatusFilter:
task_id="task2", task_id="task2",
step_name="OCR", step_name="OCR",
status="in_progress", status="in_progress",
message="Processing..." message="Processing...",
) )
db_session.add(log2) db_session.add(log2)
db_session.commit() db_session.commit()
@@ -235,7 +228,7 @@ class TestStatusFilter:
original_filename="processing.pdf", original_filename="processing.pdf",
local_filename="/tmp/processing.pdf", local_filename="/tmp/processing.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -245,7 +238,7 @@ class TestStatusFilter:
task_id="task1", task_id="task1",
step_name="OCR", step_name="OCR",
status="in_progress", status="in_progress",
message="Processing..." message="Processing...",
) )
db_session.add(log) db_session.add(log)
db_session.commit() db_session.commit()
@@ -263,7 +256,7 @@ class TestStatusFilter:
original_filename="completed.pdf", original_filename="completed.pdf",
local_filename="/tmp/completed.pdf", local_filename="/tmp/completed.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -273,7 +266,7 @@ class TestStatusFilter:
task_id="task1", task_id="task1",
step_name="OCR", step_name="OCR",
status="success", status="success",
message="Completed" message="Completed",
) )
db_session.add(log) db_session.add(log)
db_session.commit() db_session.commit()
@@ -291,7 +284,7 @@ class TestStatusFilter:
original_filename="failed.pdf", original_filename="failed.pdf",
local_filename="/tmp/failed.pdf", local_filename="/tmp/failed.pdf",
file_size=1024, file_size=1024,
mime_type="application/pdf" mime_type="application/pdf",
) )
db_session.add(file_record) db_session.add(file_record)
db_session.flush() db_session.flush()
@@ -301,7 +294,7 @@ class TestStatusFilter:
task_id="task1", task_id="task1",
step_name="OCR", step_name="OCR",
status="failure", status="failure",
message="Failed" message="Failed",
) )
db_session.add(log) db_session.add(log)
db_session.commit() db_session.commit()