Ignore invalid editor draft payloads (#1533)
This commit is contained in:
@@ -67,6 +67,14 @@ def _summary(d: EditorDraft) -> Dict[str, Any]:
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _load_payload(raw: Optional[str]) -> Dict[str, Any]:
|
||||||
|
try:
|
||||||
|
payload = json.loads(raw) if raw else {}
|
||||||
|
except Exception:
|
||||||
|
return {}
|
||||||
|
return payload if isinstance(payload, dict) else {}
|
||||||
|
|
||||||
|
|
||||||
def setup_editor_draft_routes() -> APIRouter:
|
def setup_editor_draft_routes() -> APIRouter:
|
||||||
router = APIRouter(tags=["editor-drafts"])
|
router = APIRouter(tags=["editor-drafts"])
|
||||||
|
|
||||||
@@ -93,13 +101,9 @@ def setup_editor_draft_routes() -> APIRouter:
|
|||||||
).first()
|
).first()
|
||||||
if not d or not _owns(d, user):
|
if not d or not _owns(d, user):
|
||||||
raise HTTPException(404, "Draft not found")
|
raise HTTPException(404, "Draft not found")
|
||||||
try:
|
|
||||||
payload = json.loads(d.payload) if d.payload else {}
|
|
||||||
except Exception:
|
|
||||||
payload = {}
|
|
||||||
return {
|
return {
|
||||||
**_summary(d),
|
**_summary(d),
|
||||||
"payload": payload,
|
"payload": _load_payload(d.payload),
|
||||||
}
|
}
|
||||||
finally:
|
finally:
|
||||||
db.close()
|
db.close()
|
||||||
|
|||||||
24
tests/test_editor_draft_payload.py
Normal file
24
tests/test_editor_draft_payload.py
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
import sys
|
||||||
|
import types
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
|
|
||||||
|
def _load_module(monkeypatch):
|
||||||
|
db_stub = types.ModuleType("core.database")
|
||||||
|
db_stub.EditorDraft = MagicMock()
|
||||||
|
db_stub.SessionLocal = MagicMock()
|
||||||
|
monkeypatch.setitem(sys.modules, "core.database", db_stub)
|
||||||
|
monkeypatch.delitem(sys.modules, "routes.editor_draft_routes", raising=False)
|
||||||
|
|
||||||
|
import routes.editor_draft_routes as mod
|
||||||
|
|
||||||
|
return mod
|
||||||
|
|
||||||
|
|
||||||
|
def test_load_payload_rejects_non_object_json(monkeypatch):
|
||||||
|
mod = _load_module(monkeypatch)
|
||||||
|
|
||||||
|
assert mod._load_payload("[]") == {}
|
||||||
|
assert mod._load_payload('"draft"') == {}
|
||||||
|
assert mod._load_payload("{bad json") == {}
|
||||||
|
assert mod._load_payload('{"layers": []}') == {"layers": []}
|
||||||
Reference in New Issue
Block a user