package golden import ( "flag" "io/ioutil" "os" "path/filepath" "testing" "github.com/stretchr/testify/require" ) // update allows golden files to be updated based on the current output. var update = flag.Bool("update", false, "update golden files") // Get reads the expected value from the file at filename and returns the value. // filename is relative to the ./testdata directory. // // If the `-update` flag is used with `go test`, the golden file will be updated // to the value of actual. func Get(t *testing.T, actual, filename string) string { t.Helper() path := filepath.Join("testdata", filename) if *update { if dir := filepath.Dir(path); dir != "." { require.NoError(t, os.MkdirAll(dir, 0755)) } err := ioutil.WriteFile(path, []byte(actual), 0644) require.NoError(t, err) } expected, err := ioutil.ReadFile(path) require.NoError(t, err) return string(expected) }