forked from TrueCloudLab/frostfs-node
[#452] morph/event: Support reputationPut
event
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
b9a1aaec23
commit
335b31206f
2 changed files with 140 additions and 0 deletions
68
pkg/morph/event/reputation/put.go
Normal file
68
pkg/morph/event/reputation/put.go
Normal file
|
@ -0,0 +1,68 @@
|
||||||
|
package reputation
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Put structure of reputation.reputationPut notification from
|
||||||
|
// morph chain.
|
||||||
|
type Put struct {
|
||||||
|
epoch uint64
|
||||||
|
peerID reputation.PeerID
|
||||||
|
value []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// MorphEvent implements Neo:Morph Event interface.
|
||||||
|
func (Put) MorphEvent() {}
|
||||||
|
|
||||||
|
// Epoch returns epoch value of reputation data.
|
||||||
|
func (p Put) Epoch() uint64 {
|
||||||
|
return p.epoch
|
||||||
|
}
|
||||||
|
|
||||||
|
// PeerID returns peer id of reputation data.
|
||||||
|
func (p Put) PeerID() reputation.PeerID {
|
||||||
|
return p.peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
// Value returns reputation structure.
|
||||||
|
func (p Put) Value() []byte {
|
||||||
|
return p.value // consider returning parsed structure
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParsePut from notification into reputation event structure.
|
||||||
|
func ParsePut(prms []stackitem.Item) (event.Event, error) {
|
||||||
|
var (
|
||||||
|
ev Put
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
|
||||||
|
if ln := len(prms); ln != 3 {
|
||||||
|
return nil, event.WrongNumberOfParameters(3, ln)
|
||||||
|
}
|
||||||
|
|
||||||
|
epoch, err := client.IntFromStackItem(prms[0])
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "could not get integer epoch number")
|
||||||
|
}
|
||||||
|
|
||||||
|
ev.epoch = uint64(epoch)
|
||||||
|
|
||||||
|
peerID, err := client.BytesFromStackItem(prms[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "could not get peer ID value")
|
||||||
|
}
|
||||||
|
|
||||||
|
ev.peerID = reputation.PeerIDFromBytes(peerID)
|
||||||
|
|
||||||
|
ev.value, err = client.BytesFromStackItem(prms[2])
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "could not get reputation value")
|
||||||
|
}
|
||||||
|
|
||||||
|
return ev, nil
|
||||||
|
}
|
72
pkg/morph/event/reputation/put_test.go
Normal file
72
pkg/morph/event/reputation/put_test.go
Normal file
|
@ -0,0 +1,72 @@
|
||||||
|
package reputation
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/big"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestParsePut(t *testing.T) {
|
||||||
|
var (
|
||||||
|
epoch uint64 = 42
|
||||||
|
|
||||||
|
peerID = reputation.PeerIDFromBytes([]byte("peerID"))
|
||||||
|
value = []byte("There should be marshalled structure")
|
||||||
|
)
|
||||||
|
|
||||||
|
t.Run("wrong number of parameters", func(t *testing.T) {
|
||||||
|
prms := []stackitem.Item{
|
||||||
|
stackitem.NewMap(),
|
||||||
|
stackitem.NewMap(),
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := ParsePut(prms)
|
||||||
|
require.EqualError(t, err, event.WrongNumberOfParameters(3, len(prms)).Error())
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("wrong epoch parameter", func(t *testing.T) {
|
||||||
|
_, err := ParsePut([]stackitem.Item{
|
||||||
|
stackitem.NewMap(),
|
||||||
|
})
|
||||||
|
|
||||||
|
require.Error(t, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("wrong peerID parameter", func(t *testing.T) {
|
||||||
|
_, err := ParsePut([]stackitem.Item{
|
||||||
|
stackitem.NewBigInteger(new(big.Int).SetUint64(epoch)),
|
||||||
|
stackitem.NewMap(),
|
||||||
|
})
|
||||||
|
|
||||||
|
require.Error(t, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("wrong value parameter", func(t *testing.T) {
|
||||||
|
_, err := ParsePut([]stackitem.Item{
|
||||||
|
stackitem.NewBigInteger(new(big.Int).SetUint64(epoch)),
|
||||||
|
stackitem.NewByteArray(peerID.Bytes()),
|
||||||
|
stackitem.NewMap(),
|
||||||
|
})
|
||||||
|
|
||||||
|
require.Error(t, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("correct behavior", func(t *testing.T) {
|
||||||
|
ev, err := ParsePut([]stackitem.Item{
|
||||||
|
stackitem.NewBigInteger(new(big.Int).SetUint64(epoch)),
|
||||||
|
stackitem.NewByteArray(peerID.Bytes()),
|
||||||
|
stackitem.NewByteArray(value),
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Equal(t, Put{
|
||||||
|
epoch: epoch,
|
||||||
|
peerID: peerID,
|
||||||
|
value: value,
|
||||||
|
}, ev)
|
||||||
|
})
|
||||||
|
}
|
Loading…
Reference in a new issue