forked from TrueCloudLab/frostfs-node
c167ae26f9
Parsers should have original notification structure to be able to construct internal event structure that contains necessary for unique nonce calculation information. So notification parsers take raw notification structure instead of slice of stack items. Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package balance
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpc/response/result/subscriptions"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestParseLock(t *testing.T) {
|
|
var (
|
|
id = []byte("Hello World")
|
|
user = util.Uint160{0x1, 0x2, 0x3}
|
|
lock = util.Uint160{0x3, 0x2, 0x1}
|
|
|
|
amount int64 = 10
|
|
until int64 = 20
|
|
)
|
|
|
|
t.Run("wrong number of parameters", func(t *testing.T) {
|
|
prms := []stackitem.Item{
|
|
stackitem.NewMap(),
|
|
stackitem.NewMap(),
|
|
}
|
|
|
|
_, err := ParseLock(createNotifyEventFromItems(prms))
|
|
require.EqualError(t, err, event.WrongNumberOfParameters(5, len(prms)).Error())
|
|
})
|
|
|
|
t.Run("wrong id parameter", func(t *testing.T) {
|
|
_, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewMap(),
|
|
}))
|
|
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("wrong from parameter", func(t *testing.T) {
|
|
_, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewByteArray(id),
|
|
stackitem.NewMap(),
|
|
}))
|
|
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("wrong lock parameter", func(t *testing.T) {
|
|
_, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewByteArray(id),
|
|
stackitem.NewByteArray(user.BytesBE()),
|
|
stackitem.NewMap(),
|
|
}))
|
|
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("wrong amount parameter", func(t *testing.T) {
|
|
_, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewByteArray(id),
|
|
stackitem.NewByteArray(user.BytesBE()),
|
|
stackitem.NewByteArray(lock.BytesBE()),
|
|
stackitem.NewMap(),
|
|
}))
|
|
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("wrong until parameter", func(t *testing.T) {
|
|
_, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewByteArray(id),
|
|
stackitem.NewByteArray(user.BytesBE()),
|
|
stackitem.NewByteArray(lock.BytesBE()),
|
|
stackitem.NewBigInteger(new(big.Int).SetInt64(amount)),
|
|
stackitem.NewMap(),
|
|
}))
|
|
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("correct behavior", func(t *testing.T) {
|
|
ev, err := ParseLock(createNotifyEventFromItems([]stackitem.Item{
|
|
stackitem.NewByteArray(id),
|
|
stackitem.NewByteArray(user.BytesBE()),
|
|
stackitem.NewByteArray(lock.BytesBE()),
|
|
stackitem.NewBigInteger(new(big.Int).SetInt64(amount)),
|
|
stackitem.NewBigInteger(new(big.Int).SetInt64(until)),
|
|
}))
|
|
|
|
require.NoError(t, err)
|
|
require.Equal(t, Lock{
|
|
id: id,
|
|
user: user,
|
|
lock: lock,
|
|
amount: amount,
|
|
until: until,
|
|
}, ev)
|
|
})
|
|
}
|
|
|
|
func createNotifyEventFromItems(items []stackitem.Item) *subscriptions.NotificationEvent {
|
|
return &subscriptions.NotificationEvent{
|
|
NotificationEvent: state.NotificationEvent{
|
|
Item: stackitem.NewArray(items),
|
|
},
|
|
}
|
|
}
|