chore: add ShouldSkip to generator template tests

This commit is contained in:
Josh Kaplinsky 2025-04-18 08:21:02 -04:00
parent 0d361c8863
commit 5c4068dec3
No known key found for this signature in database
GPG key ID: 5D5721AE9886811F

View file

@ -3,6 +3,11 @@ package postgres
import ( import (
"database/sql" "database/sql"
"fmt" "fmt"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"github.com/go-jet/jet/v2/generator/metadata" "github.com/go-jet/jet/v2/generator/metadata"
"github.com/go-jet/jet/v2/generator/postgres" "github.com/go-jet/jet/v2/generator/postgres"
"github.com/go-jet/jet/v2/generator/template" "github.com/go-jet/jet/v2/generator/template"
@ -12,9 +17,6 @@ import (
postgres2 "github.com/go-jet/jet/v2/postgres" postgres2 "github.com/go-jet/jet/v2/postgres"
"github.com/go-jet/jet/v2/tests/dbconfig" "github.com/go-jet/jet/v2/tests/dbconfig"
file2 "github.com/go-jet/jet/v2/tests/internal/utils/file" file2 "github.com/go-jet/jet/v2/tests/internal/utils/file"
"github.com/stretchr/testify/require"
"path/filepath"
"testing"
) )
const tempTestDir = "./.tempTestDir" const tempTestDir = "./.tempTestDir"
@ -62,9 +64,7 @@ func TestGeneratorTemplate_Model_SkipGeneration(t *testing.T) {
template.Default(postgres2.Dialect). template.Default(postgres2.Dialect).
UseSchema(func(schemaMetaData metadata.Schema) template.Schema { UseSchema(func(schemaMetaData metadata.Schema) template.Schema {
return template.DefaultSchema(schemaMetaData). return template.DefaultSchema(schemaMetaData).
UseModel(template.Model{ UseModel(template.DefaultModel().ShouldSkip(true))
Skip: true,
})
}), }),
) )
@ -80,9 +80,7 @@ func TestGeneratorTemplate_SQLBuilder_SkipGeneration(t *testing.T) {
template.Default(postgres2.Dialect). template.Default(postgres2.Dialect).
UseSchema(func(schemaMetaData metadata.Schema) template.Schema { UseSchema(func(schemaMetaData metadata.Schema) template.Schema {
return template.DefaultSchema(schemaMetaData). return template.DefaultSchema(schemaMetaData).
UseSQLBuilder(template.SQLBuilder{ UseSQLBuilder(template.DefaultSQLBuilder().ShouldSkip(true))
Skip: true,
})
}), }),
) )