/
tengo_test.go
137 lines (127 loc) · 3.73 KB
/
tengo_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
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)
}