.config/coc/extensions/coc-go-data/tools/pkg/mod/golang.org/x/tools@v0.1.1-0.20210319172145-bda8f5cee399/go/analysis/internal/checker/checker_test.go

   1 // Copyright 2019 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 package checker_test
   6
   7 import (
   8         "fmt"
   9         "go/ast"
  10         "io/ioutil"
  11         "path/filepath"
  12         "testing"
  13
  14         "golang.org/x/tools/go/analysis"
  15         "golang.org/x/tools/go/analysis/analysistest"
  16         "golang.org/x/tools/go/analysis/internal/checker"
  17         "golang.org/x/tools/go/analysis/passes/inspect"
  18         "golang.org/x/tools/go/ast/inspector"
  19         "golang.org/x/tools/internal/testenv"
  20 )
  21
  22 var from, to string
  23
  24 func TestApplyFixes(t *testing.T) {
  25         testenv.NeedsGoPackages(t)
  26
  27         from = "bar"
  28         to = "baz"
  29
  30         files := map[string]string{
  31                 "rename/test.go": `package rename
  32
  33 func Foo() {
  34         bar := 12
  35         _ = bar
  36 }
  37
  38 // the end
  39 `}
  40         want := `package rename
  41
  42 func Foo() {
  43         baz := 12
  44         _ = baz
  45 }
  46
  47 // the end
  48 `
  49
  50         testdata, cleanup, err := analysistest.WriteFiles(files)
  51         if err != nil {
  52                 t.Fatal(err)
  53         }
  54         path := filepath.Join(testdata, "src/rename/test.go")
  55         checker.Fix = true
  56         checker.Run([]string{"file=" + path}, []*analysis.Analyzer{analyzer})
  57
  58         contents, err := ioutil.ReadFile(path)
  59         if err != nil {
  60                 t.Fatal(err)
  61         }
  62
  63         got := string(contents)
  64         if got != want {
  65                 t.Errorf("contents of rewritten file\ngot: %s\nwant: %s", got, want)
  66         }
  67
  68         defer cleanup()
  69 }
  70
  71 var analyzer = &analysis.Analyzer{
  72         Name:     "rename",
  73         Requires: []*analysis.Analyzer{inspect.Analyzer},
  74         Run:      run,
  75 }
  76
  77 func run(pass *analysis.Pass) (interface{}, error) {
  78         inspect := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)
  79         nodeFilter := []ast.Node{(*ast.Ident)(nil)}
  80         inspect.Preorder(nodeFilter, func(n ast.Node) {
  81                 ident := n.(*ast.Ident)
  82                 if ident.Name == from {
  83                         msg := fmt.Sprintf("renaming %q to %q", from, to)
  84                         pass.Report(analysis.Diagnostic{
  85                                 Pos:     ident.Pos(),
  86                                 End:     ident.End(),
  87                                 Message: msg,
  88                                 SuggestedFixes: []analysis.SuggestedFix{{
  89                                         Message: msg,
  90                                         TextEdits: []analysis.TextEdit{{
  91                                                 Pos:     ident.Pos(),
  92                                                 End:     ident.End(),
  93                                                 NewText: []byte(to),
  94                                         }},
  95                                 }},
  96                         })
  97                 }
  98         })
  99
 100         return nil, nil
 101 }