prysm-pulse/testing/bls/utils/utils.go

package utils

import (
	"io/ioutil"
	"path"
	"strings"
	"testing"

	"github.com/bazelbuild/rules_go/go/tools/bazel"
	"github.com/prysmaticlabs/prysm/shared/fileutil"
	"github.com/prysmaticlabs/prysm/shared/testutil/require"
)

func RetrieveFiles(name string, t *testing.T) ([]string, [][]byte) {
	filepath, err := bazel.Runfile(name)
	require.NoError(t, err)
	testFiles, err := ioutil.ReadDir(filepath)
	require.NoError(t, err)

	fileNames := []string{}
	fileContent := [][]byte{}
	require.Equal(t, false, len(testFiles) == 0, "no files exist in directory")
	for _, f := range testFiles {
		// Remove .yml suffix
		fName := strings.TrimSuffix(f.Name(), ".yaml")
		fileNames = append(fileNames, fName)
		data, err := fileutil.ReadFileAsBytes(path.Join(filepath, f.Name()))
		require.NoError(t, err)
		fileContent = append(fileContent, data)
	}
	return fileNames, fileContent
}