41 lines
797 B
Go
41 lines
797 B
Go
package datagen
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFixedPayload(t *testing.T) {
|
|
const size = 123
|
|
data := make([]byte, size)
|
|
_, err := rand.Read(data)
|
|
require.NoError(t, err)
|
|
|
|
p := NewFixedPayload(data)
|
|
require.Equal(t, size, p.Size())
|
|
|
|
actual, err := io.ReadAll(p.Reader())
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, actual)
|
|
|
|
h := sha256.Sum256(data)
|
|
require.Equal(t, hex.EncodeToString(h[:]), p.Hash())
|
|
}
|
|
|
|
func TestStreamingPayload(t *testing.T) {
|
|
const size = 123
|
|
|
|
p := NewStreamPayload(size, 0, "")
|
|
require.Equal(t, size, p.Size())
|
|
|
|
actual, err := io.ReadAll(p.Reader())
|
|
require.NoError(t, err)
|
|
require.Equal(t, size, len(actual))
|
|
require.Equal(t, sha256.Size*2, len(p.Hash()))
|
|
}
|