137 lines
3.7 KiB
Go
137 lines
3.7 KiB
Go
package tengo_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/d5/tengo/v2"
|
|
"github.com/d5/tengo/v2/parser"
|
|
"github.com/d5/tengo/v2/require"
|
|
)
|
|
|
|
func TestInstructions_String(t *testing.T) {
|
|
assertInstructionString(t,
|
|
[][]byte{
|
|
tengo.MakeInstruction(parser.OpConstant, 1),
|
|
tengo.MakeInstruction(parser.OpConstant, 2),
|
|
tengo.MakeInstruction(parser.OpConstant, 65535),
|
|
},
|
|
`0000 CONST 1
|
|
0003 CONST 2
|
|
0006 CONST 65535`)
|
|
|
|
assertInstructionString(t,
|
|
[][]byte{
|
|
tengo.MakeInstruction(parser.OpBinaryOp, 11),
|
|
tengo.MakeInstruction(parser.OpConstant, 2),
|
|
tengo.MakeInstruction(parser.OpConstant, 65535),
|
|
},
|
|
`0000 BINARYOP 11
|
|
0002 CONST 2
|
|
0005 CONST 65535`)
|
|
|
|
assertInstructionString(t,
|
|
[][]byte{
|
|
tengo.MakeInstruction(parser.OpBinaryOp, 11),
|
|
tengo.MakeInstruction(parser.OpGetLocal, 1),
|
|
tengo.MakeInstruction(parser.OpConstant, 2),
|
|
tengo.MakeInstruction(parser.OpConstant, 65535),
|
|
},
|
|
`0000 BINARYOP 11
|
|
0002 GETL 1
|
|
0004 CONST 2
|
|
0007 CONST 65535`)
|
|
}
|
|
|
|
func TestMakeInstruction(t *testing.T) {
|
|
makeInstruction(t, []byte{parser.OpConstant, 0, 0},
|
|
parser.OpConstant, 0)
|
|
makeInstruction(t, []byte{parser.OpConstant, 0, 1},
|
|
parser.OpConstant, 1)
|
|
makeInstruction(t, []byte{parser.OpConstant, 255, 254},
|
|
parser.OpConstant, 65534)
|
|
makeInstruction(t, []byte{parser.OpPop}, parser.OpPop)
|
|
makeInstruction(t, []byte{parser.OpTrue}, parser.OpTrue)
|
|
makeInstruction(t, []byte{parser.OpFalse}, parser.OpFalse)
|
|
}
|
|
|
|
func TestNumObjects(t *testing.T) {
|
|
testCountObjects(t, &tengo.Array{}, 1)
|
|
testCountObjects(t, &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2},
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 3},
|
|
&tengo.Int{Value: 4},
|
|
&tengo.Int{Value: 5},
|
|
}},
|
|
}}, 7)
|
|
testCountObjects(t, tengo.TrueValue, 1)
|
|
testCountObjects(t, tengo.FalseValue, 1)
|
|
testCountObjects(t, &tengo.BuiltinFunction{}, 1)
|
|
testCountObjects(t, &tengo.Bytes{Value: []byte("foobar")}, 1)
|
|
testCountObjects(t, &tengo.Char{Value: '가'}, 1)
|
|
testCountObjects(t, &tengo.CompiledFunction{}, 1)
|
|
testCountObjects(t, &tengo.Error{Value: &tengo.Int{Value: 5}}, 2)
|
|
testCountObjects(t, &tengo.Float{Value: 19.84}, 1)
|
|
testCountObjects(t, &tengo.ImmutableArray{Value: []tengo.Object{
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2},
|
|
&tengo.ImmutableArray{Value: []tengo.Object{
|
|
&tengo.Int{Value: 3},
|
|
&tengo.Int{Value: 4},
|
|
&tengo.Int{Value: 5},
|
|
}},
|
|
}}, 7)
|
|
testCountObjects(t, &tengo.ImmutableMap{
|
|
Value: map[string]tengo.Object{
|
|
"k1": &tengo.Int{Value: 1},
|
|
"k2": &tengo.Int{Value: 2},
|
|
"k3": &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 3},
|
|
&tengo.Int{Value: 4},
|
|
&tengo.Int{Value: 5},
|
|
}},
|
|
}}, 7)
|
|
testCountObjects(t, &tengo.Int{Value: 1984}, 1)
|
|
testCountObjects(t, &tengo.Map{Value: map[string]tengo.Object{
|
|
"k1": &tengo.Int{Value: 1},
|
|
"k2": &tengo.Int{Value: 2},
|
|
"k3": &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 3},
|
|
&tengo.Int{Value: 4},
|
|
&tengo.Int{Value: 5},
|
|
}},
|
|
}}, 7)
|
|
testCountObjects(t, &tengo.String{Value: "foo bar"}, 1)
|
|
testCountObjects(t, &tengo.Time{Value: time.Now()}, 1)
|
|
testCountObjects(t, tengo.UndefinedValue, 1)
|
|
}
|
|
|
|
func testCountObjects(t *testing.T, o tengo.Object, expected int) {
|
|
require.Equal(t, expected, tengo.CountObjects(o))
|
|
}
|
|
|
|
func assertInstructionString(
|
|
t *testing.T,
|
|
instructions [][]byte,
|
|
expected string,
|
|
) {
|
|
concatted := make([]byte, 0)
|
|
for _, e := range instructions {
|
|
concatted = append(concatted, e...)
|
|
}
|
|
require.Equal(t, expected, strings.Join(
|
|
tengo.FormatInstructions(concatted, 0), "\n"))
|
|
}
|
|
|
|
func makeInstruction(
|
|
t *testing.T,
|
|
expected []byte,
|
|
opcode parser.Opcode,
|
|
operands ...int,
|
|
) {
|
|
inst := tengo.MakeInstruction(opcode, operands...)
|
|
require.Equal(t, expected, inst)
|
|
}
|