mirror of
https://github.com/golang/go
synced 2024-11-18 19:54:44 -07:00
35c670923e
This is mostly to set things up to use golden files the same way for other commands. Change-Id: I7fcc7165706763e655b0e46f0790b367fe5d3d59 Reviewed-on: https://go-review.googlesource.com/c/tools/+/174018 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Rebecca Stambler <rstambler@golang.org>
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
// Copyright 2019 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 cmd_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io/ioutil"
|
|
"os/exec"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
"golang.org/x/tools/internal/lsp/cmd"
|
|
"golang.org/x/tools/internal/lsp/tests"
|
|
"golang.org/x/tools/internal/tool"
|
|
)
|
|
|
|
var formatModes = [][]string{
|
|
[]string{},
|
|
[]string{"-d"},
|
|
}
|
|
|
|
func (r *runner) Format(t *testing.T, data tests.Formats) {
|
|
for _, spn := range data {
|
|
for _, mode := range formatModes {
|
|
tag := "gofmt" + strings.Join(mode, "")
|
|
uri := spn.URI()
|
|
filename, err := uri.Filename()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
args := append(mode, filename)
|
|
expect := string(r.data.Golden(tag, filename, func(golden string) error {
|
|
cmd := exec.Command("gofmt", args...)
|
|
buf := &bytes.Buffer{}
|
|
cmd.Stdout = buf
|
|
cmd.Run() // ignore error, sometimes we have intentionally ungofmt-able files
|
|
contents := r.normalizePaths(fixFileHeader(buf.String()))
|
|
return ioutil.WriteFile(golden, []byte(contents), 0666)
|
|
}))
|
|
if expect == "" {
|
|
//TODO: our error handling differs, for now just skip unformattable files
|
|
continue
|
|
}
|
|
app := &cmd.Application{}
|
|
app.Config = r.data.Config
|
|
got := captureStdOut(t, func() {
|
|
tool.Main(context.Background(), app, append([]string{"format"}, args...))
|
|
})
|
|
got = r.normalizePaths(got)
|
|
// check the first two lines are the expected file header
|
|
if expect != got {
|
|
t.Errorf("format failed with %#v expected:\n%s\ngot:\n%s", args, expect, got)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var unifiedHeader = regexp.MustCompile(`^diff -u.*\n(---\s+\S+\.go\.orig)\s+[\d-:. ]+(\n\+\+\+\s+\S+\.go)\s+[\d-:. ]+(\n@@)`)
|
|
|
|
func fixFileHeader(s string) string {
|
|
match := unifiedHeader.FindStringSubmatch(s)
|
|
if match == nil {
|
|
return s
|
|
}
|
|
return strings.Join(append(match[1:], s[len(match[0]):]), "")
|
|
}
|