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