tests: update status for go tests

Author: Hoblovski

.gitignore

@@ -69,6 +69,7 @@ __pycache__
 rust-analyzer-x86_64-unknown-linux-gnu
 
 testdata/test
+testdata/repos
 testdata/jsons
 
 src/lang/testdata
@@ -77,4 +78,4 @@ src/lang/testdata
 tools
 abcoder
 
-!testdata/asts/*.json
+!testdata/asts/*.json

lang/golang/parser/pkg_test.go

@@ -16,21 +16,23 @@ package parser
 
 import (
 	"encoding/json"
-	"fmt"
 	"go/ast"
 	"go/parser"
 	"go/token"
 	"os"
-	"path/filepath"
 	"testing"
 
+	"github.com/cloudwego/abcoder/lang/testutils"
 	. "github.com/cloudwego/abcoder/lang/uniast"
 )
 
+const localSessURL = "github.com/cloudwego/localsession"
+
 func Test_goParser_ParseRepo(t *testing.T) {
 	type fields struct {
 		modName     string
 		homePageDir string
+		id          Identity
 	}
 	tests := []struct {
 		name   string
@@ -39,40 +41,31 @@ func Test_goParser_ParseRepo(t *testing.T) {
 		{
 			name: "test",
 			fields: fields{
-				modName:     "github.com/cloudwego/localsession",
-				homePageDir: "../../../tmp/localsession",
+				modName:     localSessURL,
+				homePageDir: "localsession",
+				id:          NewIdentity(localSessURL, localSessURL+"/backup", "RecoverCtxOnDemands"),
 			},
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			abs, _ := filepath.Abs(tt.fields.homePageDir)
-			println(abs)
-			p := newGoParser(tt.fields.modName, tt.fields.homePageDir, Options{
+			repoDir, err := testutils.GitCloneFast(tt.fields.modName, tt.fields.homePageDir, "main")
+			if err != nil {
+				t.Fatalf("failed to clone repo %s", err)
+			}
+			p := newGoParser(tt.fields.modName, repoDir, Options{
 				ReferCodeDepth: 1,
 				NeedTest:       true,
 			})
 			r, err := p.ParseRepo()
 			if err != nil {
-				t.Fatal(err)
+				t.Fatalf("failed to parse repo %s", err)
 			}
 			r.BuildGraph()
-			// spew.Dump(p)
-			pj, err := json.MarshalIndent(r, "", "  ")
+			_, err = p.getNode(tt.fields.id)
 			if err != nil {
-				t.Fatal(err)
+				t.Fatalf("failed to get node %s", err)
 			}
-			_ = pj
-			_ = os.WriteFile("ast.json", pj, 0644)
-			n, err := p.getNode(NewIdentity("github.com/cloudwego/localsession", "github.com/cloudwego/localsession/backup", "RecoverCtxOnDemands"))
-			if err != nil {
-				t.Fatal(err)
-			}
-			jf, err := json.MarshalIndent(n, "", "  ")
-			if err != nil {
-				t.Fatalf("json.Marshal() error = %v", err)
-			}
-			os.WriteFile("node.json", jf, 0644)
 		})
 	}
 }
@@ -93,7 +86,7 @@ func Test_goParser_ParseDirs(t *testing.T) {
 		{
 			name: "test",
 			args: args{
-				homePageDir: "../../../testdata/golang",
+				homePageDir: testutils.FirstTest("go"),
 				modName:     "a.b/c",
 				pkg:         "a.b/c/cmd",
 				opts: Options{
@@ -137,7 +130,6 @@ type Struct struct {
 		t.Fatal(err)
 	}
 	ast.Inspect(node, func(n ast.Node) bool {
-		fmt.Printf("%#v\n", n)
 		if sel, ok := n.(*ast.SelectorExpr); ok {
 			println("selector:", string(GetRawContent(fset, []byte(src), sel, false)))
 		}
@@ -157,6 +149,10 @@ func Test_goParser_ParseNode(t *testing.T) {
 		pkgPath string
 		name    string
 	}
+	localSessionDir, err := testutils.GitCloneFast(localSessURL, "localsession", "main")
+	if err != nil {
+		t.Fatalf("failed to clone repo %s", err)
+	}
 	tests := []struct {
 		name    string
 		fields  fields
@@ -167,7 +163,7 @@ func Test_goParser_ParseNode(t *testing.T) {
 			name: "test",
 			fields: fields{
 				modName:     "github.com/cloudwego/localsession",
-				homePageDir: "../../../tmp/localsession",
+				homePageDir: localSessionDir,
 			},
 			args: args{
 				pkgPath: "github.com/modern-go/gls",

lang/golang/writer/write_test.go

@@ -22,11 +22,13 @@ import (
 	"reflect"
 	"testing"
 
+	"github.com/cloudwego/abcoder/lang/testutils"
 	"github.com/cloudwego/abcoder/lang/uniast"
 )
 
 func TestWriter_WriteRepo(t *testing.T) {
-	repo, err := uniast.LoadRepo("../../../../tmp_compress/localsession.json")
+	astFile := testutils.GetTestAstFile("localsession")
+	repo, err := uniast.LoadRepo(astFile)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -46,17 +48,18 @@ func TestWriter_WriteRepo(t *testing.T) {
 			name: "test",
 			fields: fields{
 				Options: Options{
-					CompilerPath: "1.18",
+					CompilerPath: "true", // DO NOT RUN go mod tidy
 				},
 			},
 			args:    args{repo: repo},
 			wantErr: false,
 		},
 	}
+	tmproot := testutils.MakeTmpTestdir(true)
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			w := NewWriter(tt.fields.Options)
-			if err := w.WriteRepo(tt.args.repo, "../../../../tmp/localsession2"); (err != nil) != tt.wantErr {
+			if err := w.WriteRepo(tt.args.repo, tmproot); (err != nil) != tt.wantErr {
 				t.Errorf("Writer.WriteRepo() error = %v, wantErr %v", err, tt.wantErr)
 			}
 		})

lang/patch/lib_test.go

@@ -17,8 +17,6 @@
 package patch
 
 import (
-	"os"
-	"os/exec"
 	"testing"
 
 	"github.com/cloudwego/abcoder/lang/testutils"
@@ -36,22 +34,18 @@ func TestPatcher(t *testing.T) {
 	}
 
 	// Load repo from git
-	tmproot := testutils.MakeTmpTestdir(true)
 	repoURL := "github.com/cloudwego/localsession"
-	repoDir := tmproot + "/localsession"
-	t.Logf("Cloning repo %s to %s...", repoURL, repoDir)
-	cmd := exec.Command("git", "clone", "--depth", "1", "--branch", "main", "https://"+repoURL, repoDir)
-	cmd.Stdout = os.Stdout
-	cmd.Stderr = os.Stderr
-	if err := cmd.Run(); err != nil {
-		t.Fatalf("git clone failed: %v", err)
+	repoDir, err := testutils.GitCloneFast(repoURL, "localsession", "main")
+	if err != nil {
+		t.Fatalf("failed to clone repo: %v", err)
 	}
 
 	// Create patcher with options
+	tmpRoot := testutils.MakeTmpTestdir(true)
 	patcher := NewPatcher(repo, Options{
-		RepoDir:        repoDir,
-		OutDir:         tmproot + "/localsession2",
-		DefaultLanuage: uniast.Golang,
+		RepoDir:         repoDir,
+		OutDir:          tmpRoot + "/localsession2",
+		DefaultLanguage: uniast.Golang,
 	})
 
 	// Create a test patch

lang/testutils/testutils.go

@@ -18,6 +18,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"runtime"
 	"sort"
@@ -69,6 +70,29 @@ func MakeTmpTestdir(reset bool) string {
 	return tmpDir
 }
 
+func GitCloneFast(repoURL, dir, branch string) (string, error) {
+	rootDir := GetTestDataRoot()
+	repoDir := filepath.Join(rootDir, "repos", dir)
+	if _, err := os.Stat(repoDir); !os.IsNotExist(err) {
+		cmd := exec.Command("git", "-C", repoDir, "status")
+		if err := cmd.Run(); err == nil {
+			return repoDir, nil
+		} else {
+			return "", fmt.Errorf("bad existing repo %s: %w", repoDir, err)
+		}
+	}
+	if err := os.MkdirAll(repoDir, 0755); err != nil {
+		return "", fmt.Errorf("failed to create repo directory: %w", err)
+	}
+	cmd := exec.Command("git", "clone", "--depth", "1", "--branch", branch, "https://"+repoURL, repoDir)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Run(); err != nil {
+		return "", fmt.Errorf("git clone failed: %w", err)
+	}
+	return repoDir, nil
+}
+
 func GetTestAstFile(name string) string {
 	rootDir := GetTestDataRoot()
 	astFile := filepath.Join(rootDir, "asts", name+".json")