From 5c4068dec3260cafd050345869f86682c19c3794 Mon Sep 17 00:00:00 2001
From: Josh Kaplinsky <37640086+joshkaplinsky@users.noreply.github.com>
Date: Fri, 18 Apr 2025 08:21:02 -0400
Subject: [PATCH] chore: add ShouldSkip to generator template tests

---
 tests/postgres/generator_template_test.go | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

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