expr/optimizer/sum_map_test.go

package optimizer_test

import (
	"testing"

	"github.com/hanzoai/expr/internal/testify/assert"
	"github.com/hanzoai/expr/internal/testify/require"

	"github.com/hanzoai/expr/ast"
	"github.com/hanzoai/expr/optimizer"
	"github.com/hanzoai/expr/parser"
)

func TestOptimize_sum_map(t *testing.T) {
	tree, err := parser.Parse(`sum(map(users, {.Age}))`)
	require.NoError(t, err)

	err = optimizer.Optimize(&tree.Node, nil)
	require.NoError(t, err)

	expected := &ast.BuiltinNode{
		Name: "sum",
		Arguments: []ast.Node{
			&ast.IdentifierNode{Value: "users"},
			&ast.PredicateNode{
				Node: &ast.MemberNode{
					Node:     &ast.PointerNode{},
					Property: &ast.StringNode{Value: "Age"},
				},
			},
		},
	}

	assert.Equal(t, ast.Dump(expected), ast.Dump(tree.Node))
}