Added test for getGolangFuncs()

Author: msoap

ast.go

@@ -6,7 +6,7 @@ import (
 	"go/token"
 )
 
-// Func - one go function in source with return information
+// Func - one go function in source
 type Func struct {
 	Name       string
 	Begin, End int

ast_test.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"reflect"
+	"testing"
+)
+
+const testGolangSrc = `package somepkg
+
+import "fmt"
+
+type T int
+
+func (r T) String() string {
+	return fmt.Sprintf("%v", r)
+}
+
+func fn() string {
+	return "Hello"
+}
+`
+
+func Test_getGolangFuncs(t *testing.T) {
+	tests := []struct {
+		name        string
+		fileContent []byte
+		wantResult  []Func
+		wantErr     bool
+	}{
+		{
+			name:        "without error",
+			fileContent: []byte(testGolangSrc),
+			wantResult: []Func{
+				{Name: "String", Begin: 44, End: 103},
+				{Name: "fn", Begin: 105, End: 141},
+			},
+			wantErr: false,
+		},
+		{
+			name:        "with error",
+			fileContent: []byte("..."),
+			wantResult:  nil,
+			wantErr:     true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotResult, err := getGolangFuncs(tt.fileContent)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("getGolangFuncs() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(gotResult, tt.wantResult) {
+				t.Errorf("getGolangFuncs() = %#v, want %#v", gotResult, tt.wantResult)
+			}
+		})
+	}
+}