add test case for #65

xnuinside · Jan 17, 2025 · b74839e · b74839e
1 parent fcac01d
commit b74839e
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/tests/functional/generator/test_dataclasses.py b/tests/functional/generator/test_dataclasses.py
@@ -203,3 +203,30 @@ class Material:
 """
     result = create_models(ddl, models_type="dataclass")["code"]
     assert expected == result
+
+
+def test_default_boolean():
+    expected = """from dataclasses import dataclass
+
+
+@dataclass
+class Material:
+
+    id: int
+    name: str
+    edited: bool = False
+    deleted: bool = True
+"""
+
+    ddl = """
+    CREATE TABLE Material
+    (
+        id integer,
+        name      string,
+        edited   boolean DEFAULT false,
+        deleted  boolean DEFAULT true,
+        PRIMARY KEY (id, name)
+    );
+    """
+    result = create_models(ddl, models_type="dataclass")["code"]
+    assert expected == result