databricks-cli/libs/dyn/dynassert/assert_test.go

46 lines
1.1 KiB
Go

package dynassert
import (
"go/parser"
"go/token"
"io/fs"
"os"
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/require"
)
func TestThatThisTestPackageIsUsed(t *testing.T) {
var base = ".."
var files []string
err := fs.WalkDir(os.DirFS(base), ".", func(path string, d fs.DirEntry, err error) error {
if d.IsDir() {
// Filter this directory.
if filepath.Base(path) == "dynassert" {
return fs.SkipDir
}
}
if ok, _ := filepath.Match("*_test.go", d.Name()); ok {
files = append(files, filepath.Join(base, path))
}
return nil
})
require.NoError(t, err)
// Confirm that none of the test files under `libs/dyn` import the
// `testify/assert` package and instead import this package for asserts.
fset := token.NewFileSet()
for _, file := range files {
f, err := parser.ParseFile(fset, file, nil, parser.ParseComments)
require.NoError(t, err)
for _, imp := range f.Imports {
if strings.Contains(imp.Path.Value, `github.com/stretchr/testify/assert`) {
t.Errorf("File %s should not import github.com/stretchr/testify/assert", file)
}
}
}
}