81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package checksum
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
|
"github.com/nspcc-dev/tzhash/tz"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestChecksum(t *testing.T) {
|
|
var c Checksum
|
|
|
|
cSHA256 := [sha256.Size]byte{}
|
|
_, _ = rand.Read(cSHA256[:])
|
|
|
|
c.SetSHA256(cSHA256)
|
|
|
|
require.Equal(t, SHA256, c.Type())
|
|
require.Equal(t, cSHA256[:], c.Value())
|
|
|
|
var cV2 refs.Checksum
|
|
c.WriteToV2(&cV2)
|
|
|
|
require.Equal(t, refs.SHA256, cV2.GetType())
|
|
require.Equal(t, cSHA256[:], cV2.GetSum())
|
|
|
|
cTZ := [tz.Size]byte{}
|
|
_, _ = rand.Read(cSHA256[:])
|
|
|
|
c.SetTillichZemor(cTZ)
|
|
|
|
require.Equal(t, TZ, c.Type())
|
|
require.Equal(t, cTZ[:], c.Value())
|
|
|
|
c.WriteToV2(&cV2)
|
|
|
|
require.Equal(t, refs.TillichZemor, cV2.GetType())
|
|
require.Equal(t, cTZ[:], cV2.GetSum())
|
|
}
|
|
|
|
func TestNewChecksum(t *testing.T) {
|
|
t.Run("default values", func(t *testing.T) {
|
|
var chs Checksum
|
|
|
|
// check initial values
|
|
require.Equal(t, Unknown, chs.Type())
|
|
require.Nil(t, chs.Value())
|
|
|
|
// convert to v2 message
|
|
var chsV2 refs.Checksum
|
|
chs.WriteToV2(&chsV2)
|
|
|
|
require.Equal(t, refs.UnknownChecksum, chsV2.GetType())
|
|
require.Nil(t, chsV2.GetSum())
|
|
})
|
|
}
|
|
|
|
func TestCalculation(t *testing.T) {
|
|
var c Checksum
|
|
payload := []byte{0, 1, 2, 3, 4, 5}
|
|
|
|
t.Run("SHA256", func(t *testing.T) {
|
|
orig := sha256.Sum256(payload)
|
|
|
|
Calculate(&c, SHA256, payload)
|
|
|
|
require.Equal(t, orig[:], c.Value())
|
|
})
|
|
|
|
t.Run("TZ", func(t *testing.T) {
|
|
orig := tz.Sum(payload)
|
|
|
|
Calculate(&c, TZ, payload)
|
|
|
|
require.Equal(t, orig[:], c.Value())
|
|
})
|
|
}
|