mirror of
https://github.com/jaypyles/Scraperr.git
synced 2025-12-04 06:45:56 +00:00
1b8c8c779a
* feat: allow downloading in MD format * fix: unit tests * fix: deployments [skip ci] * fix: deployment
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from unittest.mock import AsyncMock, patch
|
|
from api.backend.app import app
|
|
from api.backend.models import DownloadJob
|
|
from api.backend.tests.factories.job_factory import create_completed_job
|
|
|
|
client = TestClient(app)
|
|
|
|
mocked_job = create_completed_job().model_dump()
|
|
mock_results = [mocked_job]
|
|
mocked_random_int = 123456
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@patch("api.backend.routers.job_router.query")
|
|
@patch("api.backend.routers.job_router.random.randint")
|
|
async def test_download(mock_randint: AsyncMock, mock_query: AsyncMock):
|
|
# Ensure the mock returns immediately
|
|
mock_query.return_value = mock_results
|
|
mock_randint.return_value = mocked_random_int
|
|
|
|
# Create a DownloadJob instance
|
|
download_job = DownloadJob(ids=[mocked_job["id"]], job_format="csv")
|
|
|
|
# Make a POST request to the /download endpoint
|
|
response = client.post("/download", json=download_job.model_dump())
|
|
|
|
# Assertions
|
|
assert response.status_code == 200
|
|
assert response.headers["Content-Disposition"] == "attachment; filename=export.csv"
|
|
|
|
# Check the content of the CSV
|
|
csv_content = response.content.decode("utf-8")
|
|
expected_csv = (
|
|
f'"id","url","element_name","xpath","text","user","time_created"\r\n'
|
|
f'"{mocked_job["id"]}-{mocked_random_int}","https://example.com","element_name","//div","example",'
|
|
f'"{mocked_job["user"]}","{mocked_job["time_created"]}"\r\n'
|
|
)
|
|
assert csv_content == expected_csv
|