mirror of
https://github.com/golang/go
synced 2024-11-13 20:10:32 -07:00
877c1892bb
* add -diff command line option * use scoping information in refersTo, isPkgDot, isPtrPkgDot. * add new scoping-based helpers countUses, rewriteUses, assignsTo, isTopName. * rename rewrite to walk, add walkBeforeAfter. * add toy typechecker, a placeholder for go/types R=gri CC=golang-dev https://golang.org/cl/4285053
127 lines
2.6 KiB
Go
127 lines
2.6 KiB
Go
// Copyright 2011 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"go/ast"
|
|
"go/parser"
|
|
"go/printer"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type testCase struct {
|
|
Name string
|
|
Fn func(*ast.File) bool
|
|
In string
|
|
Out string
|
|
}
|
|
|
|
var testCases []testCase
|
|
|
|
func addTestCases(t []testCase) {
|
|
testCases = append(testCases, t...)
|
|
}
|
|
|
|
func fnop(*ast.File) bool { return false }
|
|
|
|
func parseFixPrint(t *testing.T, fn func(*ast.File) bool, desc, in string) (out string, fixed, ok bool) {
|
|
file, err := parser.ParseFile(fset, desc, in, parserMode)
|
|
if err != nil {
|
|
t.Errorf("%s: parsing: %v", desc, err)
|
|
return
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
buf.Reset()
|
|
_, err = (&printer.Config{printerMode, tabWidth}).Fprint(&buf, fset, file)
|
|
if err != nil {
|
|
t.Errorf("%s: printing: %v", desc, err)
|
|
return
|
|
}
|
|
if s := buf.String(); in != s && fn != fnop {
|
|
t.Errorf("%s: not gofmt-formatted.\n--- %s\n%s\n--- %s | gofmt\n%s",
|
|
desc, desc, in, desc, s)
|
|
tdiff(t, in, s)
|
|
return
|
|
}
|
|
|
|
if fn == nil {
|
|
for _, fix := range fixes {
|
|
if fix.f(file) {
|
|
fixed = true
|
|
}
|
|
}
|
|
} else {
|
|
fixed = fn(file)
|
|
}
|
|
|
|
buf.Reset()
|
|
_, err = (&printer.Config{printerMode, tabWidth}).Fprint(&buf, fset, file)
|
|
if err != nil {
|
|
t.Errorf("%s: printing: %v", desc, err)
|
|
return
|
|
}
|
|
|
|
return buf.String(), fixed, true
|
|
}
|
|
|
|
func TestRewrite(t *testing.T) {
|
|
for _, tt := range testCases {
|
|
// Apply fix: should get tt.Out.
|
|
out, fixed, ok := parseFixPrint(t, tt.Fn, tt.Name, tt.In)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
// reformat to get printing right
|
|
out, _, ok = parseFixPrint(t, fnop, tt.Name, out)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if out != tt.Out {
|
|
t.Errorf("%s: incorrect output.\n", tt.Name)
|
|
if !strings.HasPrefix(tt.Name, "testdata/") {
|
|
t.Errorf("--- have\n%s\n--- want\n%s", out, tt.Out)
|
|
}
|
|
tdiff(t, out, tt.Out)
|
|
continue
|
|
}
|
|
|
|
if changed := out != tt.In; changed != fixed {
|
|
t.Errorf("%s: changed=%v != fixed=%v", tt.Name, changed, fixed)
|
|
continue
|
|
}
|
|
|
|
// Should not change if run again.
|
|
out2, fixed2, ok := parseFixPrint(t, tt.Fn, tt.Name+" output", out)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if fixed2 {
|
|
t.Errorf("%s: applied fixes during second round", tt.Name)
|
|
continue
|
|
}
|
|
|
|
if out2 != out {
|
|
t.Errorf("%s: changed output after second round of fixes.\n--- output after first round\n%s\n--- output after second round\n%s",
|
|
tt.Name, out, out2)
|
|
tdiff(t, out, out2)
|
|
}
|
|
}
|
|
}
|
|
|
|
func tdiff(t *testing.T, a, b string) {
|
|
data, err := diff([]byte(a), []byte(b))
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
t.Error(string(data))
|
|
}
|