diff --git a/generator/template/model_template_test.go b/generator/template/model_template_test.go index a7bbe28..0fce487 100644 --- a/generator/template/model_template_test.go +++ b/generator/template/model_template_test.go @@ -1,9 +1,11 @@ package template import ( - "github.com/go-jet/jet/v2/generator/metadata" - "github.com/stretchr/testify/require" "testing" + + "github.com/stretchr/testify/require" + + "github.com/go-jet/jet/v2/generator/metadata" ) func Test_TableModelField(t *testing.T) { @@ -43,3 +45,29 @@ func Test_TableModelField(t *testing.T) { Tags: nil, }) } + +func Test_Model_ShouldSkip(t *testing.T) { + tests := []struct { + name string + initial Model + skip bool + }{ + { + name: "True", + initial: Model{Skip: false}, + skip: true, + }, + { + name: "False", + initial: Model{Skip: true}, + skip: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + updatedModel := tt.initial.ShouldSkip(tt.skip) + require.Equal(t, tt.skip, updatedModel.Skip) + }) + } +} diff --git a/generator/template/sql_builder_template_test.go b/generator/template/sql_builder_template_test.go index b980a60..9d6dcdb 100644 --- a/generator/template/sql_builder_template_test.go +++ b/generator/template/sql_builder_template_test.go @@ -1,9 +1,11 @@ package template import ( - "github.com/go-jet/jet/v2/generator/metadata" - "github.com/stretchr/testify/require" "testing" + + "github.com/stretchr/testify/require" + + "github.com/go-jet/jet/v2/generator/metadata" ) func TestToGoEnumValueIdentifier(t *testing.T) { @@ -35,3 +37,29 @@ func TestColumnRenameReserved(t *testing.T) { }) } } + +func Test_SQLBuilder_ShouldSkip(t *testing.T) { + tests := []struct { + name string + initial SQLBuilder + skip bool + }{ + { + name: "True", + initial: SQLBuilder{Skip: false}, + skip: true, + }, + { + name: "False", + initial: SQLBuilder{Skip: true}, + skip: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + updatedBuilder := tt.initial.ShouldSkip(tt.skip) + require.Equal(t, tt.skip, updatedBuilder.Skip) + }) + } +}