2021-06-11 17:26:25 +00:00
|
|
|
package kit2
|
2021-06-10 17:25:02 +00:00
|
|
|
|
|
|
|
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()
|
|
|
|
}
|
|
|
|
|
2021-06-11 17:26:25 +00:00
|
|
|
// AssertFilesEqual compares two files by blake2b hash equality and
|
|
|
|
// fails the test if unequal.
|
|
|
|
func AssertFilesEqual(t *testing.T, left, right string) {
|
2021-06-10 17:25:02 +00:00
|
|
|
// 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)
|
|
|
|
|
2021-06-11 17:26:25 +00:00
|
|
|
require.True(t, bytes.Equal(leftD, rightD))
|
2021-06-10 17:25:02 +00:00
|
|
|
}
|