Merge pull request #780 from nspcc-dev/fix/776

vm: allow to push integer-like items to stack
This commit is contained in:
Roman Khimov 2020-03-18 13:27:31 +03:00 committed by GitHub
commit b37c5939f8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 0 deletions

View file

@ -87,6 +87,11 @@ func makeStackItem(v interface{}) StackItem {
}
return makeStackItem(a)
default:
i64T := reflect.TypeOf(int64(0))
if reflect.TypeOf(val).ConvertibleTo(i64T) {
i64Val := reflect.ValueOf(val).Convert(i64T).Interface()
return makeStackItem(i64Val)
}
panic(
fmt.Sprintf(
"invalid stack item type: %v (%v)",

View file

@ -1,9 +1,11 @@
package vm
import (
"math/big"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestPushElement(t *testing.T) {
@ -20,6 +22,20 @@ func TestPushElement(t *testing.T) {
}
}
func TestStack_PushVal(t *testing.T) {
type (
i32 int32
testByte uint8
)
s := NewStack("test")
require.NotPanics(t, func() { s.PushVal(i32(123)) })
require.NotPanics(t, func() { s.PushVal(testByte(42)) })
require.Equal(t, 2, s.Len())
require.Equal(t, big.NewInt(42), s.Pop().Value())
require.Equal(t, big.NewInt(123), s.Pop().Value())
}
func TestPopElement(t *testing.T) {
var (
s = NewStack("test")