mirror of https://github.com/docker/cli.git
143 lines
4.1 KiB
Go
143 lines
4.1 KiB
Go
/*Package golden provides tools for comparing large mutli-line strings.
|
|
|
|
Golden files are files in the ./testdata/ subdirectory of the package under test.
|
|
*/
|
|
package golden // import "gotest.tools/golden"
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
|
|
"gotest.tools/assert"
|
|
"gotest.tools/assert/cmp"
|
|
"gotest.tools/internal/format"
|
|
)
|
|
|
|
var flagUpdate = flag.Bool("test.update-golden", false, "update golden file")
|
|
|
|
type helperT interface {
|
|
Helper()
|
|
}
|
|
|
|
// Get returns the contents of the file in ./testdata
|
|
func Get(t assert.TestingT, filename string) []byte {
|
|
if ht, ok := t.(helperT); ok {
|
|
ht.Helper()
|
|
}
|
|
expected, err := ioutil.ReadFile(Path(filename))
|
|
assert.NilError(t, err)
|
|
return expected
|
|
}
|
|
|
|
// Path returns the full path to a file in ./testdata
|
|
func Path(filename string) string {
|
|
if filepath.IsAbs(filename) {
|
|
return filename
|
|
}
|
|
return filepath.Join("testdata", filename)
|
|
}
|
|
|
|
func update(filename string, actual []byte, normalize normalize) error {
|
|
if *flagUpdate {
|
|
return ioutil.WriteFile(Path(filename), normalize(actual), 0644)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type normalize func([]byte) []byte
|
|
|
|
func removeCarriageReturn(in []byte) []byte {
|
|
return bytes.Replace(in, []byte("\r\n"), []byte("\n"), -1)
|
|
}
|
|
|
|
func exactBytes(in []byte) []byte {
|
|
return in
|
|
}
|
|
|
|
// Assert compares the actual content to the expected content in the golden file.
|
|
// If the `-test.update-golden` flag is set then the actual content is written
|
|
// to the golden file.
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
// This is equivalent to assert.Check(t, String(actual, filename))
|
|
func Assert(t assert.TestingT, actual string, filename string, msgAndArgs ...interface{}) {
|
|
if ht, ok := t.(helperT); ok {
|
|
ht.Helper()
|
|
}
|
|
assert.Assert(t, String(actual, filename), msgAndArgs...)
|
|
}
|
|
|
|
// String compares actual to the contents of filename and returns success
|
|
// if the strings are equal.
|
|
// If the `-test.update-golden` flag is set then the actual content is written
|
|
// to the golden file.
|
|
//
|
|
// Any \r\n substrings in actual are converted to a single \n character
|
|
// before comparing it to the expected string. When updating the golden file the
|
|
// normalized version will be written to the file. This allows Windows to use
|
|
// the same golden files as other operating systems.
|
|
func String(actual string, filename string) cmp.Comparison {
|
|
return func() cmp.Result {
|
|
actualBytes := removeCarriageReturn([]byte(actual))
|
|
result, expected := compare(actualBytes, filename, removeCarriageReturn)
|
|
if result != nil {
|
|
return result
|
|
}
|
|
diff := format.UnifiedDiff(format.DiffConfig{
|
|
A: string(expected),
|
|
B: string(actualBytes),
|
|
From: "expected",
|
|
To: "actual",
|
|
})
|
|
return cmp.ResultFailure("\n" + diff)
|
|
}
|
|
}
|
|
|
|
// AssertBytes compares the actual result to the expected result in the golden
|
|
// file. If the `-test.update-golden` flag is set then the actual content is
|
|
// written to the golden file.
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
// This is equivalent to assert.Check(t, Bytes(actual, filename))
|
|
func AssertBytes(
|
|
t assert.TestingT,
|
|
actual []byte,
|
|
filename string,
|
|
msgAndArgs ...interface{},
|
|
) {
|
|
if ht, ok := t.(helperT); ok {
|
|
ht.Helper()
|
|
}
|
|
assert.Assert(t, Bytes(actual, filename), msgAndArgs...)
|
|
}
|
|
|
|
// Bytes compares actual to the contents of filename and returns success
|
|
// if the bytes are equal.
|
|
// If the `-test.update-golden` flag is set then the actual content is written
|
|
// to the golden file.
|
|
func Bytes(actual []byte, filename string) cmp.Comparison {
|
|
return func() cmp.Result {
|
|
result, expected := compare(actual, filename, exactBytes)
|
|
if result != nil {
|
|
return result
|
|
}
|
|
msg := fmt.Sprintf("%v (actual) != %v (expected)", actual, expected)
|
|
return cmp.ResultFailure(msg)
|
|
}
|
|
}
|
|
|
|
func compare(actual []byte, filename string, normalize normalize) (cmp.Result, []byte) {
|
|
if err := update(filename, actual, normalize); err != nil {
|
|
return cmp.ResultFromError(err), nil
|
|
}
|
|
expected, err := ioutil.ReadFile(Path(filename))
|
|
if err != nil {
|
|
return cmp.ResultFromError(err), nil
|
|
}
|
|
if bytes.Equal(expected, actual) {
|
|
return cmp.ResultSuccess, nil
|
|
}
|
|
return nil, expected
|
|
}
|