go/internal/lsp/cmd/format_test.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"
	"fmt"
	"io/ioutil"
	"os/exec"
	"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 {
			isDiff := false
			tag := "gofmt"
			for _, arg := range mode {
				tag += arg
				if arg == "-d" {
					isDiff = true
				}
			}
			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 := buf.String()
				// strip the unwanted diff line
				if isDiff {
					if strings.HasPrefix(contents, "diff -u") {
						if i := strings.IndexRune(contents, '\n'); i >= 0 && i < len(contents)-1 {
							contents = contents[i+1:]
						}
					}
					contents, _ = stripFileHeader(contents)
				}
				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...))
			})
			if isDiff {
				got, err = stripFileHeader(got)
				if err != nil {
					t.Errorf("%v: got: %v\n%v", filename, err, got)
					continue
				}
			}
			// 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)
			}
		}
	}
}

func stripFileHeader(s string) (string, error) {
	s = strings.TrimSpace(s)
	if !strings.HasPrefix(s, "---") {
		return s, fmt.Errorf("missing original")
	}
	if i := strings.IndexRune(s, '\n'); i >= 0 && i < len(s)-1 {
		s = s[i+1:]
	} else {
		return s, fmt.Errorf("no EOL for original")
	}
	if !strings.HasPrefix(s, "+++") {
		return s, fmt.Errorf("missing output")
	}
	if i := strings.IndexRune(s, '\n'); i >= 0 && i < len(s)-1 {
		s = s[i+1:]
	} else {
		return s, fmt.Errorf("no EOL for output")
	}
	return s, nil
}