fix gomodcheck test

Author: wwcchh0123

internal/linters/go/gomodcheck/gomodcheck.go

@@ -48,17 +48,17 @@ func goModCheckHandler(ctx context.Context, a lint.Agent) error {
 func goModCheckOutput(log *xlog.Logger, a lint.Agent) (map[string][]lint.LinterOutput, error) {
 	output := make(map[string][]lint.LinterOutput)
 	extensions := []string{".mod"}
-	modFiles, err := util.FindFileWithExt(".", extensions)
+	modFiles, err := util.FindFileWithExt(a.RepoDir, extensions)
 	if err != nil {
 		return output, err
 	}
 	for _, file := range modFiles {
-		fName := file
+		fName := strings.TrimPrefix(file, a.RepoDir+"/")
 		if !strings.HasSuffix(fName, "go.mod") {
 			continue
 		}
 
-		goModPath := fName
+		goModPath := file
 		file, err := os.ReadFile(goModPath)
 		if err != nil {
 			log.Errorf("Error opening %s: %s", goModPath, err)

internal/linters/go/gomodcheck/gomodcheck_test.go

@@ -17,7 +17,6 @@
 package gomodcheck
 
 import (
-	"context"
 	"os"
 	"path/filepath"
 	"reflect"
@@ -108,14 +107,10 @@ func TestGoModCheck(t *testing.T) {
 
 	for _, tc := range tcs {
 		t.Run(tc.id, func(t *testing.T) {
-			p, err := lint.NewGithubProvider(context.TODO(), nil, github.PullRequestEvent{}, lint.WithPullRequestChangedFiles(tc.input))
-			if err != nil {
-				t.Errorf("Error creating github provider: %v", err)
-				return
-			}
+
 			// prepare go.mod files
-			for _, file := range p.GetFiles(nil) {
-				filename := file
+			for _, file := range tc.input {
+				filename := *file.Filename
 				dir := filepath.Dir(filename)
 				err := os.MkdirAll(dir, 0o755)
 				if err != nil {
@@ -131,9 +126,12 @@ func TestGoModCheck(t *testing.T) {
 				}
 			}
 
-			output, err := goModCheckOutput(&xlog.Logger{}, lint.Agent{
-				Provider: p,
-			})
+			repoDir, err := os.Getwd()
+			if err != nil {
+				t.Errorf("Error get current working directory: %v", err)
+				return
+			}
+			output, err := goModCheckOutput(&xlog.Logger{}, lint.Agent{RepoDir: repoDir})
 			if err != nil {
 				t.Errorf("Error execute goModCheckOutput : %v", err)
 			}

server.go

@@ -430,6 +430,17 @@ func (s *Server) handleCodeRequestEvent(ctx context.Context, info *codeRequestIn
 
 func processForCLI(ctx context.Context, path string, o cliOptions) error {
 	log := util.FromContext(ctx)
+	// 如果传入的是目录，则运行目录下的所有文件
+	//  todo 如果传入的是文件，提取后缀，并只跑对应的linter
+	info, err := os.Stat(path)
+	if err != nil {
+		return err
+	}
+	if !info.IsDir() {
+		path = filepath.Dir(path)
+		// ext := filepath.Ext(path)
+	}
+
 	for name, fn := range lint.TotalPullRequestHandlers() {
 		// check if linter is installed
 		_, err := exec.LookPath(name)