11"""Tests for Drizzle ORM parser and generator."""
2+
23from __future__ import annotations
34
45import sys
@@ -135,9 +136,20 @@ def test_generate_simple_table(self):
135136
136137 schema = Schema ()
137138 table = Table (name = "users" )
138- table .columns .append (Column (name = "id" , type = ColumnType .INTEGER , primary_key = True ))
139- table .columns .append (Column (name = "name" , type = ColumnType .STRING , nullable = False , type_args = {"length" : 255 }))
140- table .columns .append (Column (name = "email" , type = ColumnType .TEXT , nullable = False , unique = True ))
139+ table .columns .append (
140+ Column (name = "id" , type = ColumnType .INTEGER , primary_key = True )
141+ )
142+ table .columns .append (
143+ Column (
144+ name = "name" ,
145+ type = ColumnType .STRING ,
146+ nullable = False ,
147+ type_args = {"length" : 255 },
148+ )
149+ )
150+ table .columns .append (
151+ Column (name = "email" , type = ColumnType .TEXT , nullable = False , unique = True )
152+ )
141153 schema .tables .append (table )
142154
143155 gen = DrizzleGenerator (dialect = "pg" )
@@ -157,7 +169,9 @@ def test_generate_with_enum(self):
157169 schema = Schema ()
158170 schema .enums .append (EnumType (name = "role" , values = ["admin" , "user" ]))
159171 table = Table (name = "users" )
160- table .columns .append (Column (name = "id" , type = ColumnType .INTEGER , primary_key = True ))
172+ table .columns .append (
173+ Column (name = "id" , type = ColumnType .INTEGER , primary_key = True )
174+ )
161175 schema .tables .append (table )
162176
163177 gen = DrizzleGenerator (dialect = "pg" )
@@ -172,7 +186,9 @@ def test_generate_mysql_dialect(self):
172186
173187 schema = Schema ()
174188 table = Table (name = "products" )
175- table .columns .append (Column (name = "id" , type = ColumnType .INTEGER , primary_key = True ))
189+ table .columns .append (
190+ Column (name = "id" , type = ColumnType .INTEGER , primary_key = True )
191+ )
176192 schema .tables .append (table )
177193
178194 gen = DrizzleGenerator (dialect = "mysql" )
@@ -186,7 +202,9 @@ def test_generate_default_now(self):
186202
187203 schema = Schema ()
188204 table = Table (name = "logs" )
189- table .columns .append (Column (name = "createdAt" , type = ColumnType .DATETIME , default = "now()" ))
205+ table .columns .append (
206+ Column (name = "createdAt" , type = ColumnType .DATETIME , default = "now()" )
207+ )
190208 schema .tables .append (table )
191209
192210 gen = DrizzleGenerator (dialect = "pg" )
0 commit comments