70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package testhelper
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func RequireLog(t *testing.T, dir string, bs *bytes.Buffer) {
|
|
t.Helper()
|
|
|
|
// Get the caller pkg used for instruction on rerunning the specific test
|
|
pc, _, _, _ := runtime.Caller(1)
|
|
funcName := runtime.FuncForPC(pc).Name()
|
|
lastSlash := strings.LastIndexByte(funcName, '/')
|
|
if lastSlash < 0 {
|
|
lastSlash = 0
|
|
}
|
|
firstDot := strings.IndexByte(funcName[lastSlash:], '.') + lastSlash
|
|
callerPkg := funcName[:firstDot]
|
|
|
|
testNameComponents := strings.Split(t.Name(), "/")
|
|
testBaseName := strings.ToLower(
|
|
strings.ReplaceAll(
|
|
testNameComponents[len(testNameComponents)-1],
|
|
" ",
|
|
"_",
|
|
),
|
|
)
|
|
wantLogFileDir := filepath.Join("testdata", dir)
|
|
wantLogFile := filepath.Join(wantLogFileDir, testBaseName)
|
|
|
|
if os.Getenv("HELMFILE_UPDATE_SNAPSHOT") != "" {
|
|
if err := os.MkdirAll(wantLogFileDir, 0755); err != nil {
|
|
t.Fatalf("unable to create directory %q: %v", wantLogFileDir, err)
|
|
}
|
|
if err := os.WriteFile(wantLogFile, bs.Bytes(), 0644); err != nil {
|
|
t.Fatalf("unable to update lint log snapshot: %v", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
wantLogData, err := os.ReadFile(wantLogFile)
|
|
if err != nil {
|
|
t.Fatalf(
|
|
"Snapshot file %q does not exist. Rerun this test with `HELMFILE_UPDATE_SNAPSHOT=1 go test -v -run %s %s` to create the snapshot",
|
|
wantLogFile,
|
|
t.Name(),
|
|
callerPkg,
|
|
)
|
|
}
|
|
|
|
wantLog := string(wantLogData)
|
|
gotLog := bs.String()
|
|
|
|
assert.Equalf(
|
|
t,
|
|
wantLog,
|
|
gotLog,
|
|
"If you think this is due to the snapshot file being outdated, rerun this test with `HELMFILE_UPDATE_SNAPSHOT=1 go test -v -run %s %s` to update the snapshot",
|
|
t.Name(),
|
|
callerPkg,
|
|
)
|
|
}
|