58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
|
package kit
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"math/rand"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/minio/blake2b-simd"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CreateRandomFile creates a random file with the provided seed and the
|
||
|
|
// provided size.
|
||
|
|
func CreateRandomFile(t *testing.T, rseed, size int) (path string) {
|
||
|
|
if size == 0 {
|
||
|
|
size = 1600
|
||
|
|
}
|
||
|
|
|
||
|
|
source := io.LimitReader(rand.New(rand.NewSource(int64(rseed))), int64(size))
|
||
|
|
|
||
|
|
file, err := os.CreateTemp(t.TempDir(), "sourcefile.dat")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
n, err := io.Copy(file, source)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.EqualValues(t, n, size)
|
||
|
|
|
||
|
|
return file.Name()
|
||
|
|
}
|
||
|
|
|
||
|
|
// FilesEqual compares two files by blake2b hash equality.
|
||
|
|
func FilesEqual(t *testing.T, left, right string) bool {
|
||
|
|
// initialize hashes.
|
||
|
|
leftH, rightH := blake2b.New256(), blake2b.New256()
|
||
|
|
|
||
|
|
// open files.
|
||
|
|
leftF, err := os.Open(left)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
rightF, err := os.Open(right)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// feed hash functions.
|
||
|
|
_, err = io.Copy(leftH, leftF)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = io.Copy(rightH, rightF)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// compute digests.
|
||
|
|
leftD, rightD := leftH.Sum(nil), rightH.Sum(nil)
|
||
|
|
|
||
|
|
return bytes.Equal(leftD, rightD)
|
||
|
|
}
|