forked from TrueCloudLab/frostfs-node
[#7] Add put container event parser for morph
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
9f51c85054
commit
5f3978b66c
2 changed files with 167 additions and 0 deletions
67
pkg/morph/event/container/put.go
Normal file
67
pkg/morph/event/container/put.go
Normal file
|
@ -0,0 +1,67 @@
|
|||
package container
|
||||
|
||||
import (
|
||||
"crypto/elliptic"
|
||||
|
||||
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// Put structure of container.Put notification from morph chain.
|
||||
type Put struct {
|
||||
rawContainer []byte
|
||||
signature []byte
|
||||
publicKey *keys.PublicKey
|
||||
}
|
||||
|
||||
// MorphEvent implements Neo:Morph Event interface.
|
||||
func (Put) MorphEvent() {}
|
||||
|
||||
// Container is a marshalled container structure, defined in API.
|
||||
func (p Put) Container() []byte { return p.rawContainer }
|
||||
|
||||
// Signature of marshalled container by container owner.
|
||||
func (p Put) Signature() []byte { return p.signature }
|
||||
|
||||
// PublicKey of container owner.
|
||||
func (p Put) PublicKey() *keys.PublicKey { return p.publicKey }
|
||||
|
||||
// ParsePut from notification into container event structure.
|
||||
func ParsePut(params []smartcontract.Parameter) (event.Event, error) {
|
||||
var (
|
||||
ev Put
|
||||
err error
|
||||
)
|
||||
|
||||
if ln := len(params); ln != 3 {
|
||||
return nil, event.WrongNumberOfParameters(3, ln)
|
||||
}
|
||||
|
||||
// parse container
|
||||
ev.rawContainer, err = client.BytesFromStackParameter(params[0])
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get container")
|
||||
}
|
||||
|
||||
// parse signature
|
||||
ev.signature, err = client.BytesFromStackParameter(params[1])
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get signature")
|
||||
}
|
||||
|
||||
// parse public key
|
||||
key, err := client.BytesFromStackParameter(params[2])
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get public key")
|
||||
}
|
||||
|
||||
ev.publicKey, err = keys.NewPublicKeyFromBytes(key, elliptic.P256())
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not parse public key")
|
||||
}
|
||||
|
||||
return ev, nil
|
||||
}
|
100
pkg/morph/event/container/put_test.go
Normal file
100
pkg/morph/event/container/put_test.go
Normal file
|
@ -0,0 +1,100 @@
|
|||
package container
|
||||
|
||||
import (
|
||||
"crypto/elliptic"
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
||||
crypto "github.com/nspcc-dev/neofs-crypto"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/util/test"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestParsePut(t *testing.T) {
|
||||
var (
|
||||
containerData = []byte("containerData")
|
||||
signature = []byte("signature")
|
||||
publicKey = &test.DecodeKey(-1).PublicKey
|
||||
)
|
||||
|
||||
t.Run("wrong number of parameters", func(t *testing.T) {
|
||||
prms := []smartcontract.Parameter{
|
||||
{},
|
||||
{},
|
||||
}
|
||||
|
||||
_, err := ParsePut(prms)
|
||||
require.EqualError(t, err, event.WrongNumberOfParameters(3, len(prms)).Error())
|
||||
})
|
||||
|
||||
t.Run("wrong container parameter", func(t *testing.T) {
|
||||
_, err := ParsePut([]smartcontract.Parameter{
|
||||
{
|
||||
Type: smartcontract.ArrayType,
|
||||
},
|
||||
})
|
||||
|
||||
require.Error(t, err)
|
||||
})
|
||||
|
||||
t.Run("wrong signature parameter", func(t *testing.T) {
|
||||
_, err := ParsePut([]smartcontract.Parameter{
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: containerData,
|
||||
},
|
||||
{
|
||||
Type: smartcontract.ArrayType,
|
||||
},
|
||||
})
|
||||
|
||||
require.Error(t, err)
|
||||
})
|
||||
|
||||
t.Run("wrong key parameter", func(t *testing.T) {
|
||||
_, err := ParsePut([]smartcontract.Parameter{
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: containerData,
|
||||
},
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: signature,
|
||||
},
|
||||
{
|
||||
Type: smartcontract.ArrayType,
|
||||
},
|
||||
})
|
||||
|
||||
require.Error(t, err)
|
||||
})
|
||||
|
||||
t.Run("correct behavior", func(t *testing.T) {
|
||||
ev, err := ParsePut([]smartcontract.Parameter{
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: containerData,
|
||||
},
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: signature,
|
||||
},
|
||||
{
|
||||
Type: smartcontract.ByteArrayType,
|
||||
Value: crypto.MarshalPublicKey(publicKey),
|
||||
},
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
expectedKey, err := keys.NewPublicKeyFromBytes(crypto.MarshalPublicKey(publicKey), elliptic.P256())
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Equal(t, Put{
|
||||
rawContainer: containerData,
|
||||
signature: signature,
|
||||
publicKey: expectedKey,
|
||||
}, ev)
|
||||
})
|
||||
}
|
Loading…
Reference in a new issue