frostfs-sdk-go/object/fmt_test.go
Denis Kirillov 2806d90089 [#93] Remove golang.org/x/crypto dependency
Signed-off-by: Denis Kirillov <denis@nspcc.ru>
2021-12-14 15:27:04 +03:00

81 lines
1.4 KiB
Go

package object
import (
"crypto/rand"
"testing"
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
"github.com/stretchr/testify/require"
)
func TestVerificationFields(t *testing.T) {
obj := NewRaw()
payload := make([]byte, 10)
_, _ = rand.Read(payload)
obj.SetPayload(payload)
obj.SetPayloadSize(uint64(len(payload)))
p, err := keys.NewPrivateKey()
require.NoError(t, err)
require.NoError(t, SetVerificationFields(&p.PrivateKey, obj))
require.NoError(t, CheckVerificationFields(obj.Object()))
items := []struct {
corrupt func()
restore func()
}{
{
corrupt: func() {
payload[0]++
},
restore: func() {
payload[0]--
},
},
{
corrupt: func() {
obj.SetPayloadSize(obj.PayloadSize() + 1)
},
restore: func() {
obj.SetPayloadSize(obj.PayloadSize() - 1)
},
},
{
corrupt: func() {
obj.ID().ToV2().GetValue()[0]++
},
restore: func() {
obj.ID().ToV2().GetValue()[0]--
},
},
{
corrupt: func() {
obj.Signature().Key()[0]++
},
restore: func() {
obj.Signature().Key()[0]--
},
},
{
corrupt: func() {
obj.Signature().Sign()[0]++
},
restore: func() {
obj.Signature().Sign()[0]--
},
},
}
for _, item := range items {
item.corrupt()
require.Error(t, CheckVerificationFields(obj.Object()))
item.restore()
require.NoError(t, CheckVerificationFields(obj.Object()))
}
}