Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 20 additions & 4 deletions backend/app/schemas.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,11 @@ class IndexRedundancyOut(BaseModel):
class DiagramViewCreateIn(BaseModel):
"""Request body for saving an ERD canvas view."""

name: str = Field(min_length=1, max_length=200)
name: str = Field(
min_length=1,
max_length=200,
pattern=r"^[^\x00-\x1F\x7F]+$",
)
# Opaque client layout (node positions, hidden tables, viewport). The API
# bounds the serialized size in the endpoint to prevent abuse.
layout_json: dict
Expand All @@ -214,8 +218,16 @@ class DiagramViewDetailOut(DiagramViewOut):
class TableAnnotationUpsertIn(BaseModel):
"""Request body for creating/updating a table annotation."""

schema_name: str = Field(min_length=1, max_length=255)
relation_name: str = Field(min_length=1, max_length=255)
schema_name: str = Field(
min_length=1,
max_length=255,
pattern=r"^[^\x00-\x1F\x7F]+$",
)
relation_name: str = Field(
min_length=1,
max_length=255,
pattern=r"^[^\x00-\x1F\x7F]+$",
)
body: str = Field(min_length=1, max_length=10_000)


Expand Down Expand Up @@ -302,7 +314,11 @@ class DbmlConvertOut(BaseModel):
class ApiKeyCreateIn(BaseModel):
"""Request body for creating an API key."""

key_name: str = Field(min_length=1, max_length=128)
key_name: str = Field(
min_length=1,
max_length=128,
pattern=r"^[^\x00-\x1F\x7F]+$",
)


class ApiKeyOut(BaseModel):
Expand Down
37 changes: 36 additions & 1 deletion backend/tests/test_schema_validation.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,14 @@
import pytest
from pydantic import ValidationError

from app.schemas import ConnectionCreateIn, ProjectCreateIn, ProjectMemberAddIn
from app.schemas import (
ApiKeyCreateIn,
ConnectionCreateIn,
DiagramViewCreateIn,
ProjectCreateIn,
ProjectMemberAddIn,
TableAnnotationUpsertIn,
)


def test_project_name_length_is_bounded() -> None:
Expand Down Expand Up @@ -37,3 +44,31 @@ def test_conn_name_rejects_control_characters() -> None:
ConnectionCreateIn(conn_name="my\x00conn", dsn="postgresql://localhost/db")
with pytest.raises(ValidationError):
ConnectionCreateIn(conn_name="my\nconn", dsn="postgresql://localhost/db")


def test_diagram_view_name_rejects_control_characters() -> None:
with pytest.raises(ValidationError):
DiagramViewCreateIn(name="my\x00view", layout_json={})
with pytest.raises(ValidationError):
DiagramViewCreateIn(name="my\nview", layout_json={})


def test_api_key_name_rejects_control_characters() -> None:
with pytest.raises(ValidationError):
ApiKeyCreateIn(key_name="my\x00key")
with pytest.raises(ValidationError):
ApiKeyCreateIn(key_name="my\nkey")


def test_table_annotation_schema_name_rejects_control_characters() -> None:
with pytest.raises(ValidationError):
TableAnnotationUpsertIn(schema_name="public\x00", relation_name="users", body="test")
with pytest.raises(ValidationError):
TableAnnotationUpsertIn(schema_name="public\n", relation_name="users", body="test")


def test_table_annotation_relation_name_rejects_control_characters() -> None:
with pytest.raises(ValidationError):
TableAnnotationUpsertIn(schema_name="public", relation_name="users\x00", body="test")
with pytest.raises(ValidationError):
TableAnnotationUpsertIn(schema_name="public", relation_name="users\n", body="test")
Loading