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