Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Properly include dialect after changes to column generation

Browse source
This commit is contained in:
Eli Ribble 2026-05-14 20:56:35 +00:00
parent 82c1b9891b
commit e007342206
No known key found for this signature in database
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/postgres/generator_template_test.go
View file

@ -464,11 +464,11 @@ func TestGeneratorTemplate_SQLBuilder_ChangeColumnTypes(t *testing.T) {
UseSQLBuilder(template.DefaultSQLBuilder(). UseSQLBuilder(template.DefaultSQLBuilder().
UseTable(func(table metadata.Table) template.TableSQLBuilder { UseTable(func(table metadata.Table) template.TableSQLBuilder {
return template.DefaultTableSQLBuilder(table). return template.DefaultTableSQLBuilder(table).
UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn { UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
defaultColumn := template.DefaultTableSQLBuilderColumn(column) defaultColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
if defaultColumn.Name == "ActorID" { if defaultColumn.Name == "ActorID" {
defaultColumn.Type = "String" defaultColumn.Type = "ColumnString"
} }
return defaultColumn return defaultColumn
@ -494,8 +494,8 @@ func TestGeneratorTemplate_Model_SQLBuilder_SkipColumn(t *testing.T) {
UseSQLBuilder(template.DefaultSQLBuilder(). UseSQLBuilder(template.DefaultSQLBuilder().
UseTable(func(table metadata.Table) template.TableSQLBuilder { UseTable(func(table metadata.Table) template.TableSQLBuilder {
return template.DefaultTableSQLBuilder(table). return template.DefaultTableSQLBuilder(table).
UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn { UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
defaultColumn := template.DefaultTableSQLBuilderColumn(column) defaultColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
if defaultColumn.Name == "FirstName" { if defaultColumn.Name == "FirstName" {
defaultColumn.Skip = true defaultColumn.Skip = true
@ -610,8 +610,8 @@ func TestGeneratorTemplate_Model_SqlBuilder_RenameStructFieldNames(t *testing.T)
UseSQLBuilder(template.DefaultSQLBuilder(). UseSQLBuilder(template.DefaultSQLBuilder().
UseTable(func(table metadata.Table) template.TableSQLBuilder { UseTable(func(table metadata.Table) template.TableSQLBuilder {
return template.DefaultTableSQLBuilder(table). return template.DefaultTableSQLBuilder(table).
UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn { UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
defaultSqlBuilderColumn := template.DefaultTableSQLBuilderColumn(column) defaultSqlBuilderColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
if table.Name == "payment" && column.Name == "amount" { if table.Name == "payment" && column.Name == "amount" {
defaultSqlBuilderColumn.Name = "AmountInCents" defaultSqlBuilderColumn.Name = "AmountInCents"