ast_expr_test.go
1.9 KB
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
package domain
import (
"github.com/linmadan/egglib-go/utils/json"
"github.com/stretchr/testify/assert"
"testing"
)
func TestAstExprUnmarshalJSON(t *testing.T) {
inputs := []struct {
data ExprAST
}{
{
data: &FunCallerExprAST{
Name: "if",
ExprType: TypeFunCallerExprAST,
Args: []ExprAST{
&BinaryExprAST{
ExprType: TypeBinaryExprAST,
Op: "<",
Lhs: &FieldExprAST{
ExprType: TypeFieldExprAST,
Str: "table.count",
},
Rhs: &FieldExprAST{
ExprType: TypeFieldExprAST,
Str: "100",
},
},
&FieldExprAST{
ExprType: TypeFieldExprAST,
Str: "200",
},
&FieldExprAST{
ExprType: TypeFieldExprAST,
Str: "300",
},
},
},
},
{
data: &FunCallerExprAST{
Name: "sum",
ExprType: TypeFunCallerExprAST,
Args: []ExprAST{
&BinaryExprAST{
ExprType: TypeBinaryExprAST,
Op: "/",
Lhs: &ValueExprAST{
ExprType: TypeValueExprAST,
Str: "1",
},
Rhs: &FunCallerExprAST{
ExprType: TypeFunCallerExprAST,
Name: "countifs",
Args: []ExprAST{
&FieldExprAST{
ExprType: TypeFieldExprAST,
Str: "业绩1",
Field: &TableField{
TableId: 1,
TableName: "测试ABC",
TableSqlName: "table_abc_test",
FieldName: "业绩",
FieldSqlName: "ye_ji_1",
FieldSQLType: "Float",
},
},
&ValueExprAST{
ExprType: TypeValueExprAST,
Str: "业绩2",
},
},
},
},
},
},
},
}
for _, input := range inputs {
data, err := json.Marshal(input.data)
if err != nil {
t.Fatal(err)
}
v, err := AstExprUnmarshalJSON(data)
if err != nil {
t.Fatal(err)
}
if v != nil {
}
assert.Equal(t, input.data, v)
}
}