wehub-resource-sync
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
"""
|
|
Tests for max_length on SourceCreate.notebooks/transformations
|
|
(api/models.py).
|
|
|
|
Both are iterated with a per-item DB lookup (Notebook.get()/
|
|
Transformation.get()) in api/routers/sources.py's create_source() - an
|
|
unbounded array let a caller amplify a single request into an arbitrarily
|
|
large number of sequential DB round trips.
|
|
"""
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from api.models import SourceCreate
|
|
|
|
|
|
def make_ids(n, prefix):
|
|
return [f"{prefix}:{i}" for i in range(n)]
|
|
|
|
|
|
class TestNotebooksMaxLength:
|
|
def test_accepts_up_to_50_notebooks(self):
|
|
request = SourceCreate(type="text", content="hi", notebooks=make_ids(50, "notebook"))
|
|
assert request.notebooks is not None
|
|
assert len(request.notebooks) == 50
|
|
|
|
def test_rejects_51_notebooks(self):
|
|
with pytest.raises(ValidationError):
|
|
SourceCreate(type="text", content="hi", notebooks=make_ids(51, "notebook"))
|
|
|
|
def test_none_notebooks_still_allowed(self):
|
|
# Pre-existing behavior (validate_notebook_fields): None normalizes
|
|
# to an empty list, unrelated to the max_length addition.
|
|
request = SourceCreate(type="text", content="hi", notebooks=None)
|
|
assert request.notebooks == []
|
|
|
|
def test_empty_list_still_allowed(self):
|
|
request = SourceCreate(type="text", content="hi", notebooks=[])
|
|
assert request.notebooks == []
|
|
|
|
|
|
class TestTransformationsMaxLength:
|
|
def test_accepts_up_to_50_transformations(self):
|
|
request = SourceCreate(
|
|
type="text", content="hi", transformations=make_ids(50, "transformation")
|
|
)
|
|
assert request.transformations is not None
|
|
assert len(request.transformations) == 50
|
|
|
|
def test_rejects_51_transformations(self):
|
|
with pytest.raises(ValidationError):
|
|
SourceCreate(
|
|
type="text", content="hi", transformations=make_ids(51, "transformation")
|
|
)
|
|
|
|
def test_default_is_empty_list(self):
|
|
request = SourceCreate(type="text", content="hi")
|
|
assert request.transformations == []
|
|
|
|
|
|
class TestFormParsingReturns422:
|
|
"""The multipart form path builds SourceCreate manually in
|
|
parse_source_form_data(), so pydantic's ValidationError doesn't go
|
|
through FastAPI's request-validation handler — without an explicit
|
|
catch it surfaced as a 500. These hit the real endpoint and assert
|
|
the client gets a clean 422 instead (found in v1.11 release testing).
|
|
"""
|
|
|
|
@pytest.fixture
|
|
def client(self):
|
|
from fastapi.testclient import TestClient
|
|
|
|
from api.main import app
|
|
|
|
return TestClient(app)
|
|
|
|
def test_51_notebooks_via_form_returns_422(self, client):
|
|
import json as _json
|
|
|
|
response = client.post(
|
|
"/api/sources",
|
|
data={
|
|
"type": "text",
|
|
"content": "probe",
|
|
"notebooks": _json.dumps(make_ids(51, "notebook")),
|
|
},
|
|
)
|
|
assert response.status_code == 422
|
|
assert "Invalid source data" in response.json()["detail"]
|
|
|
|
def test_invalid_notebooks_json_returns_422(self, client):
|
|
response = client.post(
|
|
"/api/sources",
|
|
data={"type": "text", "content": "probe", "notebooks": "not-json["},
|
|
)
|
|
assert response.status_code == 422
|
|
assert "notebooks" in response.json()["detail"]
|
|
|
|
def test_invalid_transformations_json_returns_422(self, client):
|
|
response = client.post(
|
|
"/api/sources",
|
|
data={"type": "text", "content": "probe", "transformations": "]bad"},
|
|
)
|
|
assert response.status_code == 422
|
|
assert "transformations" in response.json()["detail"]
|