mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-29 23:33:37 +00:00
6b21ad9922
Everywhere including examples, external interop APIs, bindings generators
code and in other valuable places. A couple of `interface{}` usages are
intentionally left in the CHANGELOG.md, documentation and tests.
355 lines
9.2 KiB
Go
355 lines
9.2 KiB
Go
package management
|
|
|
|
import (
|
|
"errors"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/neorpc/result"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/nef"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testAct struct {
|
|
err error
|
|
res *result.Invoke
|
|
tx *transaction.Transaction
|
|
txh util.Uint256
|
|
vub uint32
|
|
}
|
|
|
|
func (t *testAct) Call(contract util.Uint160, operation string, params ...any) (*result.Invoke, error) {
|
|
return t.res, t.err
|
|
}
|
|
func (t *testAct) MakeCall(contract util.Uint160, method string, params ...any) (*transaction.Transaction, error) {
|
|
return t.tx, t.err
|
|
}
|
|
func (t *testAct) MakeUnsignedCall(contract util.Uint160, method string, attrs []transaction.Attribute, params ...any) (*transaction.Transaction, error) {
|
|
return t.tx, t.err
|
|
}
|
|
func (t *testAct) SendCall(contract util.Uint160, method string, params ...any) (util.Uint256, uint32, error) {
|
|
return t.txh, t.vub, t.err
|
|
}
|
|
func (t *testAct) MakeRun(script []byte) (*transaction.Transaction, error) {
|
|
return t.tx, t.err
|
|
}
|
|
func (t *testAct) MakeUnsignedRun(script []byte, attrs []transaction.Attribute) (*transaction.Transaction, error) {
|
|
return t.tx, t.err
|
|
}
|
|
func (t *testAct) SendRun(script []byte) (util.Uint256, uint32, error) {
|
|
return t.txh, t.vub, t.err
|
|
}
|
|
func (t *testAct) CallAndExpandIterator(contract util.Uint160, method string, maxItems int, params ...any) (*result.Invoke, error) {
|
|
return t.res, t.err
|
|
}
|
|
func (t *testAct) TerminateSession(sessionID uuid.UUID) error {
|
|
return t.err
|
|
}
|
|
func (t *testAct) TraverseIterator(sessionID uuid.UUID, iterator *result.Iterator, num int) ([]stackitem.Item, error) {
|
|
return t.res.Stack, t.err
|
|
}
|
|
|
|
func TestReader(t *testing.T) {
|
|
ta := new(testAct)
|
|
man := NewReader(ta)
|
|
|
|
ta.err = errors.New("")
|
|
_, err := man.GetContract(util.Uint160{1, 2, 3})
|
|
require.Error(t, err)
|
|
_, err = man.GetContractByID(1)
|
|
require.Error(t, err)
|
|
_, err = man.GetMinimumDeploymentFee()
|
|
require.Error(t, err)
|
|
_, err = man.HasMethod(util.Uint160{1, 2, 3}, "method", 0)
|
|
require.Error(t, err)
|
|
|
|
ta.err = nil
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make(42),
|
|
},
|
|
}
|
|
_, err = man.GetContract(util.Uint160{1, 2, 3})
|
|
require.Error(t, err)
|
|
_, err = man.GetContractByID(1)
|
|
require.Error(t, err)
|
|
fee, err := man.GetMinimumDeploymentFee()
|
|
require.NoError(t, err)
|
|
require.Equal(t, big.NewInt(42), fee)
|
|
hm, err := man.HasMethod(util.Uint160{1, 2, 3}, "method", 0)
|
|
require.NoError(t, err)
|
|
require.True(t, hm)
|
|
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make(false),
|
|
},
|
|
}
|
|
_, err = man.GetContract(util.Uint160{1, 2, 3})
|
|
require.Error(t, err)
|
|
_, err = man.GetContractByID(1)
|
|
require.Error(t, err)
|
|
hm, err = man.HasMethod(util.Uint160{1, 2, 3}, "method", 0)
|
|
require.NoError(t, err)
|
|
require.False(t, hm)
|
|
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make([]stackitem.Item{}),
|
|
},
|
|
}
|
|
_, err = man.GetContract(util.Uint160{1, 2, 3})
|
|
require.Error(t, err)
|
|
_, err = man.GetContractByID(1)
|
|
require.Error(t, err)
|
|
|
|
nefFile, _ := nef.NewFile([]byte{1, 2, 3})
|
|
nefBytes, _ := nefFile.Bytes()
|
|
manif := manifest.DefaultManifest("stack item")
|
|
manifItem, _ := manif.ToStackItem()
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make([]stackitem.Item{
|
|
stackitem.Make(1),
|
|
stackitem.Make(0),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
stackitem.Make(nefBytes),
|
|
manifItem,
|
|
}),
|
|
},
|
|
}
|
|
cs, err := man.GetContract(util.Uint160{1, 2, 3})
|
|
require.NoError(t, err)
|
|
require.Equal(t, int32(1), cs.ID)
|
|
require.Equal(t, uint16(0), cs.UpdateCounter)
|
|
require.Equal(t, util.Uint160{1, 2, 3}, cs.Hash)
|
|
cs2, err := man.GetContractByID(1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, cs, cs2)
|
|
}
|
|
|
|
func TestGetContractHashes(t *testing.T) {
|
|
ta := &testAct{}
|
|
man := NewReader(ta)
|
|
|
|
ta.err = errors.New("")
|
|
_, err := man.GetContractHashes()
|
|
require.Error(t, err)
|
|
_, err = man.GetContractHashesExpanded(5)
|
|
require.Error(t, err)
|
|
|
|
ta.err = nil
|
|
iid := uuid.New()
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.NewInterop(result.Iterator{
|
|
ID: &iid,
|
|
}),
|
|
},
|
|
}
|
|
_, err = man.GetContractHashes()
|
|
require.Error(t, err)
|
|
|
|
// Session-based iterator.
|
|
sid := uuid.New()
|
|
ta.res = &result.Invoke{
|
|
Session: sid,
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.NewInterop(result.Iterator{
|
|
ID: &iid,
|
|
}),
|
|
},
|
|
}
|
|
iter, err := man.GetContractHashes()
|
|
require.NoError(t, err)
|
|
|
|
ta.res = &result.Invoke{
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make([]stackitem.Item{
|
|
stackitem.Make([]byte{0, 0, 0, 1}),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
}),
|
|
},
|
|
}
|
|
vals, err := iter.Next(10)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(vals))
|
|
require.Equal(t, IDHash{
|
|
ID: 1,
|
|
Hash: util.Uint160{1, 2, 3},
|
|
}, vals[0])
|
|
|
|
ta.err = errors.New("")
|
|
_, err = iter.Next(1)
|
|
require.Error(t, err)
|
|
|
|
err = iter.Terminate()
|
|
require.Error(t, err)
|
|
|
|
// Value-based iterator.
|
|
ta.err = nil
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.NewInterop(result.Iterator{
|
|
Values: []stackitem.Item{stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.Make([]byte{0, 0, 0, 1}),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
})},
|
|
}),
|
|
},
|
|
}
|
|
iter, err = man.GetContractHashes()
|
|
require.NoError(t, err)
|
|
|
|
ta.err = errors.New("")
|
|
err = iter.Terminate()
|
|
require.NoError(t, err)
|
|
|
|
// Expanded
|
|
ta.err = nil
|
|
ta.res = &result.Invoke{
|
|
State: "HALT",
|
|
Stack: []stackitem.Item{
|
|
stackitem.Make([]stackitem.Item{stackitem.Make([]stackitem.Item{
|
|
stackitem.Make([]byte{0, 0, 0, 1}),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
})}),
|
|
},
|
|
}
|
|
vals, err = man.GetContractHashesExpanded(5)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(vals))
|
|
require.Equal(t, IDHash{
|
|
ID: 1,
|
|
Hash: util.Uint160{1, 2, 3},
|
|
}, vals[0])
|
|
}
|
|
|
|
func TestSetMinimumDeploymentFee(t *testing.T) {
|
|
ta := new(testAct)
|
|
man := New(ta)
|
|
|
|
ta.err = errors.New("")
|
|
_, _, err := man.SetMinimumDeploymentFee(big.NewInt(42))
|
|
require.Error(t, err)
|
|
|
|
for _, m := range []func(*big.Int) (*transaction.Transaction, error){
|
|
man.SetMinimumDeploymentFeeTransaction,
|
|
man.SetMinimumDeploymentFeeUnsigned,
|
|
} {
|
|
_, err = m(big.NewInt(100))
|
|
require.Error(t, err)
|
|
}
|
|
|
|
ta.err = nil
|
|
ta.txh = util.Uint256{1, 2, 3}
|
|
ta.vub = 42
|
|
|
|
h, vub, err := man.SetMinimumDeploymentFee(big.NewInt(42))
|
|
require.NoError(t, err)
|
|
require.Equal(t, ta.txh, h)
|
|
require.Equal(t, ta.vub, vub)
|
|
|
|
ta.tx = transaction.New([]byte{1, 2, 3}, 100500)
|
|
for _, m := range []func(*big.Int) (*transaction.Transaction, error){
|
|
man.SetMinimumDeploymentFeeTransaction,
|
|
man.SetMinimumDeploymentFeeUnsigned,
|
|
} {
|
|
tx, err := m(big.NewInt(100))
|
|
require.NoError(t, err)
|
|
require.Equal(t, ta.tx, tx)
|
|
}
|
|
}
|
|
|
|
func TestDeploy(t *testing.T) {
|
|
ta := new(testAct)
|
|
man := New(ta)
|
|
nefFile, _ := nef.NewFile([]byte{1, 2, 3})
|
|
manif := manifest.DefaultManifest("stack item")
|
|
|
|
ta.err = errors.New("")
|
|
_, _, err := man.Deploy(nefFile, manif, nil)
|
|
require.Error(t, err)
|
|
|
|
for _, m := range []func(exe *nef.File, manif *manifest.Manifest, data any) (*transaction.Transaction, error){
|
|
man.DeployTransaction,
|
|
man.DeployUnsigned,
|
|
} {
|
|
_, err = m(nefFile, manif, nil)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
ta.err = nil
|
|
ta.txh = util.Uint256{1, 2, 3}
|
|
ta.vub = 42
|
|
|
|
h, vub, err := man.Deploy(nefFile, manif, nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, ta.txh, h)
|
|
require.Equal(t, ta.vub, vub)
|
|
|
|
ta.tx = transaction.New([]byte{1, 2, 3}, 100500)
|
|
for _, m := range []func(exe *nef.File, manif *manifest.Manifest, data any) (*transaction.Transaction, error){
|
|
man.DeployTransaction,
|
|
man.DeployUnsigned,
|
|
} {
|
|
tx, err := m(nefFile, manif, nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, ta.tx, tx)
|
|
|
|
_, err = m(nefFile, manif, map[int]int{})
|
|
require.Error(t, err)
|
|
}
|
|
|
|
_, _, err = man.Deploy(nefFile, manif, map[int]int{})
|
|
require.Error(t, err)
|
|
|
|
_, _, err = man.Deploy(nefFile, manif, 100500)
|
|
require.NoError(t, err)
|
|
|
|
nefFile.Compiler = "intentionally very long compiler string that will make NEF code explode on encoding"
|
|
_, _, err = man.Deploy(nefFile, manif, nil)
|
|
require.Error(t, err)
|
|
|
|
// Unfortunately, manifest _always_ marshals successfully (or panics).
|
|
}
|
|
|
|
func TestItemsToIDHashesErrors(t *testing.T) {
|
|
for name, input := range map[string][]stackitem.Item{
|
|
"not a struct": {stackitem.Make(1)},
|
|
"wrong length": {stackitem.Make([]stackitem.Item{})},
|
|
"wrong id": {stackitem.Make([]stackitem.Item{
|
|
stackitem.Make([]stackitem.Item{}),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
})},
|
|
"lengthy id": {stackitem.Make([]stackitem.Item{
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
stackitem.Make(util.Uint160{1, 2, 3}.BytesBE()),
|
|
})},
|
|
"not a good hash": {stackitem.Make([]stackitem.Item{
|
|
stackitem.Make([]byte{0, 0, 0, 1}),
|
|
stackitem.Make([]stackitem.Item{}),
|
|
})},
|
|
"not a good u160 hash": {stackitem.Make([]stackitem.Item{
|
|
stackitem.Make([]byte{0, 0, 0, 1}),
|
|
stackitem.Make(util.Uint256{1, 2, 3}.BytesBE()),
|
|
})},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
_, err := itemsToIDHashes(input)
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
}
|