feat: added dataclasses tests

Signed-off-by: QuentinN42 <quentin@lieumont.fr>
sqlc-dev · QuentinN42 · Jul 25, 2025 · Jul 25, 2025 · Jul 25, 2025 · Jul 25, 2025
commit 068fd251596b60ba31d51fb8ffd3a5b057413269
diff --git a/...ydantic_models_with_fields/db/__init__.py → ...ses_with_reserved_keywords/db/__init__.py b/...ydantic_models_with_fields/db/__init__.py → ...ses_with_reserved_keywords/db/__init__.py
diff --git a/internal/endtoend/testdata/dataclasses_with_reserved_keywords/db/models.py b/internal/endtoend/testdata/dataclasses_with_reserved_keywords/db/models.py
@@ -0,0 +1,10 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.29.0
+import dataclasses
+
+
+@dataclasses.dataclass()
+class Author:
+    id: int
+    class_: str
diff --git a/...t_pydantic_models_with_fields/db/query.py → ...lasses_with_reserved_keywords/db/query.py b/...t_pydantic_models_with_fields/db/query.py → ...lasses_with_reserved_keywords/db/query.py
diff --git a/...mit_pydantic_models_with_fields/query.sql → ...aclasses_with_reserved_keywords/query.sql b/...mit_pydantic_models_with_fields/query.sql → ...aclasses_with_reserved_keywords/query.sql
diff --git a/...it_pydantic_models_with_fields/schema.sql → ...classes_with_reserved_keywords/schema.sql b/...it_pydantic_models_with_fields/schema.sql → ...classes_with_reserved_keywords/schema.sql
diff --git a/internal/endtoend/testdata/dataclasses_with_reserved_keywords/sqlc.yaml b/internal/endtoend/testdata/dataclasses_with_reserved_keywords/sqlc.yaml
@@ -0,0 +1,18 @@
+version: "2"
+plugins:
+  - name: py
+    wasm:
+      url: file://../../../../bin/sqlc-gen-python.wasm
+      sha256: "24b0da217e85c9b952a4c746476aa761e9b293a4a68bef8409d97edc1c003016"
+sql:
+  - schema: schema.sql
+    queries: query.sql
+    engine: postgresql
+    codegen:
+      - plugin: py
+        out: db
+        options:
+          package: db
+          emit_sync_querier: true
+          emit_async_querier: true
+          emit_pydantic_models: false
diff --git a/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/db/__init__.py b/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/db/__init__.py
diff --git a/..._pydantic_models_with_fields/db/models.py → ...odels_with_reserved_keywords/db/models.py b/..._pydantic_models_with_fields/db/models.py → ...odels_with_reserved_keywords/db/models.py
diff --git a/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/db/query.py b/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/db/query.py
@@ -0,0 +1,44 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.29.0
+# source: query.sql
+from typing import Optional
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from db import models
+
+
+GET_AUTHOR = """-- name: get_author \\:one
+SELECT id, class FROM authors
+WHERE id = :p1 LIMIT 1
+"""
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    def get_author(self, *, id: int) -> Optional[models.Author]:
+        row = self._conn.execute(sqlalchemy.text(GET_AUTHOR), {"p1": id}).first()
+        if row is None:
+            return None
+        return models.Author(
+            id=row[0],
+            class_=row[1],
+        )
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    async def get_author(self, *, id: int) -> Optional[models.Author]:
+        row = (await self._conn.execute(sqlalchemy.text(GET_AUTHOR), {"p1": id})).first()
+        if row is None:
+            return None
+        return models.Author(
+            id=row[0],
+            class_=row[1],
+        )
diff --git a/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/query.sql b/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/query.sql
@@ -0,0 +1,3 @@
+-- name: GetAuthor :one
+SELECT * FROM authors
+WHERE id = $1 LIMIT 1;
diff --git a/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/schema.sql b/internal/endtoend/testdata/emit_pydantic_models_with_reserved_keywords/schema.sql
@@ -0,0 +1,4 @@
+CREATE TABLE authors (
+          id    BIGSERIAL PRIMARY KEY,
+          class text      NOT NULL
+);
diff --git a/...mit_pydantic_models_with_fields/sqlc.yaml → ...c_models_with_reserved_keywords/sqlc.yaml b/...mit_pydantic_models_with_fields/sqlc.yaml → ...c_models_with_reserved_keywords/sqlc.yaml