fix: raw struct unit test (#188)

cloudwego · Apr 9, 2024 · 9463514 · 9463514
1 parent 91a39db
commit 9463514
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/generator/golang/templates/constant.go b/generator/golang/templates/constant.go
@@ -23,7 +23,7 @@ const (
 	{{- range $Consts}}
 	{{InsertionPoint "constant" .Name}}
 	{{- if and Features.ReserveComments .ReservedComments}}{{.ReservedComments}}{{end}}
-	{{.GoName}} {{.GoTypeName}} = {{.Initialization}}
+	{{.GoName}} = {{.Initialization}}
 	{{- end}}{{/* range $Consts */}}
 	{{InsertionPoint "constants"}}
 )

diff --git a/generator/golang/templates/init_test.go b/generator/golang/templates/init_test.go
@@ -23,6 +23,7 @@ import (
 	"github.com/cloudwego/thriftgo/generator/backend"
 	"github.com/cloudwego/thriftgo/generator/golang"
 	"github.com/cloudwego/thriftgo/generator/golang/templates"
+	"github.com/cloudwego/thriftgo/generator/golang/templates/raw_struct"
 )
 
 // forceSingleDefinition panics if the given template contains multiple definition.
@@ -69,7 +70,7 @@ func TestDefinitionNumber(t *testing.T) {
 	}
 	for pkg, tpls := range templates.Alternative() {
 		for _, tpl := range tpls {
-			if tpl != templates.File {
+			if !(tpl == templates.File || tpl == raw_struct.File) { // raw_struct rewrite 'template.File'
 				forceSingleDefinition(pkg, tpl, funcs)
 			}
 		}

diff --git a/version/version.go b/version/version.go
@@ -14,4 +14,4 @@
 
 package version
 
-const ThriftgoVersion = "0.3.9"
+const ThriftgoVersion = "0.3.10"