mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-05 03:58:23 +00:00
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package compiler_test
|
|
|
|
import (
|
|
"fmt"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPanic(t *testing.T) {
|
|
t.Run("no panic", func(t *testing.T) {
|
|
src := getPanicSource(false, `"execution fault"`)
|
|
eval(t, src, big.NewInt(7))
|
|
})
|
|
|
|
t.Run("panic with message", func(t *testing.T) {
|
|
var logs []string
|
|
src := getPanicSource(true, `"execution fault"`)
|
|
v := vmAndCompile(t, src)
|
|
v.RegisterInteropGetter(logGetter(&logs))
|
|
|
|
require.Error(t, v.Run())
|
|
require.True(t, v.HasFailed())
|
|
require.Equal(t, 1, len(logs))
|
|
require.Equal(t, "execution fault", logs[0])
|
|
})
|
|
|
|
t.Run("panic with nil", func(t *testing.T) {
|
|
var logs []string
|
|
src := getPanicSource(true, `nil`)
|
|
v := vmAndCompile(t, src)
|
|
v.RegisterInteropGetter(logGetter(&logs))
|
|
|
|
require.Error(t, v.Run())
|
|
require.True(t, v.HasFailed())
|
|
require.Equal(t, 0, len(logs))
|
|
})
|
|
}
|
|
|
|
func getPanicSource(need bool, message string) string {
|
|
return fmt.Sprintf(`
|
|
package main
|
|
func Main() int {
|
|
needPanic := %#v
|
|
if needPanic {
|
|
panic(%s)
|
|
return 5
|
|
}
|
|
return 7
|
|
}
|
|
`, need, message)
|
|
}
|
|
|
|
func logGetter(logs *[]string) vm.InteropGetterFunc {
|
|
logID := vm.InteropNameToID([]byte("Neo.Runtime.Log"))
|
|
return func(id uint32) *vm.InteropFuncPrice {
|
|
if id != logID {
|
|
return nil
|
|
}
|
|
|
|
return &vm.InteropFuncPrice{
|
|
Func: func(v *vm.VM) error {
|
|
msg := string(v.Estack().Pop().Bytes())
|
|
*logs = append(*logs, msg)
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
}
|