forked from TrueCloudLab/neoneo-go
smartcontract: add ExpandParameterToEmitable method
It'll help to convert Parameter to a atandard type which then can be emitted ass an array item.
This commit is contained in:
parent
c77997a357
commit
28b74cb647
2 changed files with 111 additions and 0 deletions
|
@ -14,6 +14,7 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
"unicode/utf8"
|
"unicode/utf8"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/io"
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/util"
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
||||||
)
|
)
|
||||||
|
@ -437,3 +438,28 @@ func NewParameterFromString(in string) (*Parameter, error) {
|
||||||
}
|
}
|
||||||
return res, nil
|
return res, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ExpandParameterToEmitable converts parameter to a type which can be handled as
|
||||||
|
// an array item by emit.Array. It correlates with the way RPC server handles
|
||||||
|
// FuncParams for invoke* calls inside the request.ExpandArrayIntoScript function.
|
||||||
|
func ExpandParameterToEmitable(param Parameter) (interface{}, error) {
|
||||||
|
var err error
|
||||||
|
switch t := param.Type; t {
|
||||||
|
case PublicKeyType:
|
||||||
|
return param.Value.(*keys.PublicKey).Bytes(), nil
|
||||||
|
case ArrayType:
|
||||||
|
arr := param.Value.([]Parameter)
|
||||||
|
res := make([]interface{}, len(arr))
|
||||||
|
for i := range arr {
|
||||||
|
res[i], err = ExpandParameterToEmitable(arr[i])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
case MapType, InteropInterfaceType, UnknownType, AnyType, VoidType:
|
||||||
|
return nil, fmt.Errorf("unsupported parameter type: %s", t.String())
|
||||||
|
default:
|
||||||
|
return param.Value, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -9,7 +9,10 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neo-go/internal/testserdes"
|
"github.com/nspcc-dev/neo-go/internal/testserdes"
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/util"
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
@ -517,3 +520,85 @@ func hexToBase64(s string) string {
|
||||||
b, _ := hex.DecodeString(s)
|
b, _ := hex.DecodeString(s)
|
||||||
return base64.StdEncoding.EncodeToString(b)
|
return base64.StdEncoding.EncodeToString(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestExpandParameterToEmitable(t *testing.T) {
|
||||||
|
pk, _ := keys.NewPrivateKey()
|
||||||
|
testCases := []struct {
|
||||||
|
In Parameter
|
||||||
|
Expected interface{}
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
In: Parameter{Type: BoolType, Value: true},
|
||||||
|
Expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: IntegerType, Value: int64(123)},
|
||||||
|
Expected: int64(123),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: ByteArrayType, Value: []byte{1, 2, 3}},
|
||||||
|
Expected: []byte{1, 2, 3},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: StringType, Value: "writing's on the wall"},
|
||||||
|
Expected: "writing's on the wall",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: Hash160Type, Value: util.Uint160{1, 2, 3}},
|
||||||
|
Expected: util.Uint160{1, 2, 3},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: Hash256Type, Value: util.Uint256{1, 2, 3}},
|
||||||
|
Expected: util.Uint256{1, 2, 3},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: PublicKeyType, Value: pk.PublicKey()},
|
||||||
|
Expected: pk.PublicKey().Bytes(),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: SignatureType, Value: []byte{1, 2, 3}},
|
||||||
|
Expected: []byte{1, 2, 3},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
In: Parameter{Type: ArrayType, Value: []Parameter{
|
||||||
|
{
|
||||||
|
Type: IntegerType,
|
||||||
|
Value: int64(123),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Type: ByteArrayType,
|
||||||
|
Value: []byte{1, 2, 3},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Type: ArrayType,
|
||||||
|
Value: []Parameter{
|
||||||
|
{
|
||||||
|
Type: BoolType,
|
||||||
|
Value: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}},
|
||||||
|
Expected: []interface{}{int64(123), []byte{1, 2, 3}, []interface{}{true}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
bw := io.NewBufBinWriter()
|
||||||
|
for _, testCase := range testCases {
|
||||||
|
actual, err := ExpandParameterToEmitable(testCase.In)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, testCase.Expected, actual)
|
||||||
|
|
||||||
|
emit.Array(bw.BinWriter, actual)
|
||||||
|
require.NoError(t, bw.Err)
|
||||||
|
}
|
||||||
|
errCases := []Parameter{
|
||||||
|
{Type: AnyType},
|
||||||
|
{Type: UnknownType},
|
||||||
|
{Type: MapType},
|
||||||
|
{Type: InteropInterfaceType},
|
||||||
|
}
|
||||||
|
for _, errCase := range errCases {
|
||||||
|
_, err := ExpandParameterToEmitable(errCase)
|
||||||
|
require.Error(t, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue