mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-13 15:45:01 +00:00
45 lines
820 B
Go
45 lines
820 B
Go
|
package stack
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/binary"
|
||
|
"math/big"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
// helper functions
|
||
|
func testPeakInteger(t *testing.T, tStack *RandomAccess, n uint16) *Int {
|
||
|
stackElement, err := tStack.Peek(n)
|
||
|
assert.Nil(t, err)
|
||
|
item, err := stackElement.Integer()
|
||
|
if err != nil {
|
||
|
t.Fail()
|
||
|
}
|
||
|
return item
|
||
|
}
|
||
|
|
||
|
func testPopInteger(t *testing.T, tStack *RandomAccess) *Int {
|
||
|
stackElement, err := tStack.Pop()
|
||
|
assert.Nil(t, err)
|
||
|
item, err := stackElement.Integer()
|
||
|
if err != nil {
|
||
|
t.Fail()
|
||
|
}
|
||
|
return item
|
||
|
}
|
||
|
|
||
|
func testMakeStackInt(t *testing.T, num int64) *Int {
|
||
|
a, err := NewInt(big.NewInt(num))
|
||
|
assert.Nil(t, err)
|
||
|
return a
|
||
|
}
|
||
|
|
||
|
func testReadInt64(data []byte) int64 {
|
||
|
var ret int64
|
||
|
buf := bytes.NewBuffer(data)
|
||
|
binary.Read(buf, binary.LittleEndian, &ret)
|
||
|
return ret
|
||
|
}
|