chore: update unit test

main_test.go

@@ -1,10 +1,11 @@
 package main
 
 import (
-	"github.com/stretchr/testify/assert"
 	"go/ast"
 	"go/token"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func Test_generate(t *testing.T) {
@@ -25,12 +26,18 @@ func Test_generate(t *testing.T) {
 				Type: "int",
 			},
 		},
+		BuildTags: []string{
+			"//go:build example",
+			"// +build example",
+		},
 	}
 
 	result, err := generate(genConfig)
 	assert.NoError(t, err)
 
-	expected := []byte(`// Code generated by go-genbuilder. DO NOT EDIT.
+	expected := `// Code generated by go-genbuilder. DO NOT EDIT.
+//go:build example
+// +build example
 
 package testpackage
 
@@ -62,9 +69,9 @@ func (builder *TestStructBuilder) SetTestOtherField(testotherfield int) *TestStr
 func (builder *TestStructBuilder) Build() *TestStruct {
 	return builder.teststruct
 }
-`)
+`
 
-	assert.Equal(t, expected, result)
+	assert.Equal(t, expected, string(result))
 }
 
 func TestParseFile(t *testing.T) {
@@ -118,6 +125,10 @@ func TestParseFile(t *testing.T) {
 						Type: "int",
 					},
 				},
+				BuildTags: []string{
+					"//go:build example",
+					"// +build example",
+				},
 			}
 			assert.Equal(t, expected, actual)
 		})