create test template for using gotests

Makefile

@@ -268,6 +268,11 @@ tensorflow/install: /usr/local/lib/libtensorflow.so
 	rm -f libtensorflow-cpu-linux-x86_64-$(TENSORFLOW_C_VERSION).tar.gz
 	ldconfig
 
+.PHONY: gentest
+## gentest
+gentest:
+	$(call gen-test)
+
 .PHONY: test
 ## run tests
 test:

Makefile.d/functions.mk

@@ -89,3 +89,27 @@ define telepresence
 	    ## will be available after telepresence 0.105 released
 	    ## --deployment-type "$(SWAP_DEPLOYMENT_TYPE)"
 endef
+
+define gen-test
+	find . -type d | \
+	grep "./cmd\|./hack\|./internal\|./pkg" | \
+	grep -v "./cmd/cli\| \
+			 ./hack/benchmark/internal/client/ngtd\| \
+			 ./hack/benchmark/internal/starter/agent\| \
+			 ./hack/benchmark/internal/starter/external\| \
+			 ./hack/benchmark/internal/starter/gateway\| \
+			 ./hack/license\| \
+			 ./hack/swagger\| \
+			 ./hack/tools"| \
+	while read dir; do \
+		files=`find $${dir} -type f -maxdepth 1 -name "*.go" -not -name '*_test.go' -not -name 'doc.go'`; \
+		for file in $${files}; do \
+			path='./assets/test/templates/common';     \
+			if [[ $${file} =~ .*options?\.go ]] ; then \
+				path='./assets/test/templates/option'; \
+			fi; \
+			echo start generate test code: $${file}; \
+			gotests -w -template_dir $${path} -all $${file}; \
+		done; \
+	done
+endef

assets/test/templates/common/call.tmpl

@@ -0,0 +1 @@
+{{define "call"}}{{with .Receiver}}{{if not .IsStruct}}test.{{end}}{{Receiver .}}.{{end}}{{.Name}}({{range $i, $el := .Parameters}}{{if $i}}, {{end}}{{if not .IsWriter}}test.args.{{end}}{{Param .}}{{if .Type.IsVariadic}}...{{end}}{{end}}){{end}}

assets/test/templates/common/fill.tmpl

@@ -0,0 +1,28 @@
+{{define "fill" }}
+        {{- range . }}
+                {{ Param . }}: {{ $typ := .Type.String -}}
+                {{- if eq $typ "int8" -}} 0,
+                {{- else if eq $typ "uint8" -}} 0,
+                {{- else if eq $typ "uint16" -}} 0,
+                {{- else if eq $typ "uint32" -}} 0,
+                {{- else if eq $typ "uint64" -}} 0,
+                {{- else if eq $typ "uint" -}} 0,
+                {{- else if eq $typ "uintptr" -}} 0,
+                {{- else if eq $typ "int8" -}} 0,
+                {{- else if eq $typ "int16" -}} 0,
+                {{- else if eq $typ "int32" -}} 0,
+                {{- else if eq $typ "int64" -}} 0,
+                {{- else if eq $typ "int" -}} 0,
+                {{- else if eq $typ "float32" -}} 0,
+                {{- else if eq $typ "float64" -}} 0,
+                {{- else if eq $typ "complex64" -}} 0+0i,
+                {{- else if eq $typ "complex128" -}} 0+0i,
+                {{- else if eq $typ "byte" -}} 0,
+                {{- else if eq $typ "rune" -}} 0,
+                {{- else if eq $typ "string" -}} "",
+                {{- else if eq $typ "bool" -}} false,
+                {{- else if .IsStruct -}} {{.Type.Value}}{},
+                {{- else -}} nil,
+                {{- end -}}
+        {{ end -}}
+{{ end }}

assets/test/templates/common/function.tmpl

@@ -0,0 +1,154 @@
+{{define "function"}}
+{{- $f := . }}
+
+func {{ .TestName }}(t *testing.T) {
+    {{- if .TestParameters }}
+    type args struct {
+		{{- range .TestParameters }}
+				{{ Param . }} {{ .Type }}
+		{{- end}}
+    }
+    {{- end }}
+    {{- $hasFields := false -}}
+    {{- with .Receiver }}
+        {{- if .IsStruct }}
+            {{- if .Fields}}
+                type fields struct { {{ $hasFields = true }}
+                    {{- range .Fields }}
+                        {{ Field . }} {{ .Type }}
+                    {{- end }}
+                }
+            {{- end }}
+        {{- end }}
+    {{- end }}
+    type want struct {
+        {{- range .TestResults }}
+            {{ Want . }} {{ .Type }}
+        {{- end }}
+        {{- if .ReturnsError }}
+            err error
+        {{- end }}
+    }
+    type test struct {
+        name string
+        {{- if .TestParameters }}
+            args args
+        {{- end }}
+        {{- with .Receiver}}
+            {{- if and .IsStruct .Fields}}
+                fields fields
+            {{- else}}
+                {{Receiver .}} {{.Type}}
+            {{- end}}
+        {{- end}}
+        want want
+        checkFunc func(want,
+                    {{- range .TestResults }} {{ .Type }}, {{- end }}
+                    {{- if .ReturnsError }} error {{- end }} ) error
+        beforeFunc func({{- if .TestParameters }} args {{- end}})
+        afterFunc func({{- if .TestParameters }} args {{- end}})
+    }
+    defaultCheckFunc := func(w want,
+        {{- range .TestResults }} {{ Got . }} {{ .Type }}, {{- end }}
+        {{- if .ReturnsError }} err error {{- end }} ) error {
+
+        {{- if .ReturnsError }}
+            if !errors.Is(err, w.err) {
+                return errors.Errorf("got error = %v, want %v", err, w.err)
+            }
+        {{- end }}
+        {{- range .TestResults }} {{ $want := Want . }} {{ $got := Got . }}
+            if !reflect.DeepEqual({{ $got }}, w.{{ $want }}) {
+                return errors.Errorf("got = %v, want %v", {{ $got }}, w.{{ $want }})
+            }
+        {{- end }}
+        return nil
+    }
+    tests := []test{
+        // TODO test cases
+        /*
+        {
+            name: "test_case_1",
+            {{- if .TestParameters }}
+            args: args {
+                {{- template "fill" .TestParameters }}
+            },
+            {{- end }}
+            {{- if $hasFields }}
+            fields: fields {
+                {{- template "fill" .Receiver.Fields }}
+            },
+            {{- end }}
+            want: want{},
+            checkFunc: defaultCheckFunc,
+        },
+        */
+
+        // TODO test cases
+        /*
+        func() test {
+            return test {
+                name: "test_case_2",
+                {{- if .TestParameters }}
+                args: args {
+                    {{- template "fill" .TestParameters }}
+                },
+                {{- end }}
+                {{- if $hasFields }}
+                fields: fields {
+                    {{- template "fill" .Receiver.Fields }}
+                },
+                {{- end }}
+                want: want{},
+                checkFunc: defaultCheckFunc,
+            }
+        }(),
+        */
+    }
+
+    for {{- if (or .Subtests (not .IsNaked)) }} _, test := {{- end }} range tests {
+        {{- if .Subtests }}
+            t.Run(test.name, func(tt *testing.T) {
+                if test.beforeFunc != nil {
+                    test.beforeFunc({{- if .TestParameters }} test.args {{- end }})
+                }
+                if test.afterFunc != nil {
+                    defer test.afterFunc({{- if .TestParameters }} test.args {{- end }})
+                }
+                if test.checkFunc == nil {
+                    test.checkFunc = defaultCheckFunc
+                }
+
+                {{- with .Receiver }}
+                    {{- if .IsStruct }}
+                        {{ Receiver . }} := {{- if .Type.IsStar }}&{{- end }}{{ .Type.Value }} {
+                        {{- range .Fields }}
+                            {{ Field . }}: test.fields.{{ Field . }},
+                        {{- end }}
+                        }
+                    {{- end }}
+                {{- end }}
+
+                {{- range .Parameters }}
+                    {{- if .IsWriter }}
+                        {{ Param . }} := &bytes.Buffer{}
+                    {{- end }}
+                {{- end }}
+
+                {{ $len := len .Results }}
+                {{- if or (ge $len 1) (.OnlyReturnsOneValue) (.OnlyReturnsError) }}
+                    {{ template "results" $f }} := {{ template "call" $f }}
+                    if err := test.checkFunc(test.want, {{ template "results" $f }} ); err != nil {
+                        tt.Errorf("error = %v", err)
+                    }
+                {{ else }}
+                    {{ template "call" $f }}
+                    if err := test.checkFunc(test.want); err != nil {
+                        tt.Errorf("error = %v", err)
+                    }
+                {{- end }}
+            })
+        {{- end }}
+    }
+}
+{{- end }}

assets/test/templates/common/header.tmpl

@@ -0,0 +1,10 @@
+{{define "header"}}
+{{range .Comments}}{{.}}
+{{end -}}
+package {{.Package}}
+
+import (
+{{range .Imports}}{{.Name}} {{.Path}}
+{{end}}
+)
+{{end}}

assets/test/templates/common/results.tmpl

@@ -0,0 +1,2 @@
+{{define "results"}}{{- if .OnlyReturnsError }}err{{ else }}{{range $i, $el := .Results}}{{if $i}}, {{end}}{{Got .}}{{end}}{{if .ReturnsError}}, err{{end}}
+{{- end }}{{end}}

assets/test/templates/option/call.tmpl

@@ -0,0 +1 @@
+../common/call.tmpl

assets/test/templates/option/fill.tmpl

@@ -0,0 +1 @@
+../common/fill.tmpl