Function TestConvertAST

common/ast/conversion_test.go:38–282

(t *testing.T)

)

func TestConvertAST(t *testing.T) {
	fac := ast.NewExprFactory()
	tests := []struct {
		name  string
		goAST *ast.AST
		pbAST *exprpb.CheckedExpr
	}{
		{
			name: "simple ast",
			goAST: ast.NewCheckedAST(ast.NewAST(nil, nil),
				map[int64]*types.Type{
					1: types.BoolType,
					2: types.DynType,
				},
				map[int64]*ast.ReferenceInfo{
					1: ast.NewFunctionReference(overloads.LogicalNot),
					2: ast.NewIdentReference("TRUE", types.True),
				},
			),
			pbAST: &exprpb.CheckedExpr{
				Expr:       &exprpb.Expr{},
				SourceInfo: &exprpb.SourceInfo{},
				TypeMap: map[int64]*exprpb.Type{
					1: chkdecls.Bool,
					2: chkdecls.Dyn,
				},
				ReferenceMap: map[int64]*exprpb.Reference{
					1: {OverloadId: []string{overloads.LogicalNot}},
					2: {
						Name: "TRUE",
						Value: &exprpb.Constant{
							ConstantKind: &exprpb.Constant_BoolValue{BoolValue: true},
						},
					},
				},
			},
		},
		{
			name: "comprehension ast",
			goAST: ast.NewAST(
				fac.NewComprehensionTwoVar(1,
					fac.NewIdent(2, "data"),
					"i",
					"v",
					"__result__",
					fac.NewList(3, []ast.Expr{}, []int32{}),
					fac.NewLiteral(4, types.True),
					fac.NewCall(8, operators.Add,
						fac.NewAccuIdent(9),
						fac.NewCall(5, operators.Add,
							fac.NewIdent(6, "i"),
							fac.NewIdent(7, "v"),
						)),
					fac.NewAccuIdent(10),
				), nil),
			pbAST: &exprpb.CheckedExpr{
				Expr: &exprpb.Expr{
					Id: 1,