go-libs/pkg/query/expression_test.go

package query

import (
	"fmt"
	"math/big"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestParseExpression(t *testing.T) {
	t.Parallel()
	json := `{
	"$not": {
		"$and": [
			{
				"$match": {
					"account": "accounts::pending"
				}
			},
			{
				"$or": [
					{
						"$gte": {
							"balance": 288230376151711747
						}
					},
					{
						"$match": {
							"metadata[category]": "gold"
						}
					}
				]
			},
			{
				"$in": {
					"account": ["A", "B"]
				}
			}
		]
	}
}`
	expr, err := ParseJSON(json)
	require.NoError(t, err)

	q, args, err := expr.Build(ContextFn(func(key, operator string, value any) (string, []any, error) {
		return fmt.Sprintf("%s %s ?", key, DefaultComparisonOperatorsMapping[operator]), []any{value}, nil
	}))
	require.NoError(t, err)
	require.Equal(t, "not ((account = ?) and ((balance >= ?) or (metadata[category] = ?)) and (account IN ?))", q)

	var expectedBalance big.Int
	expectedBalance.SetString("288230376151711747", 10)
	require.Equal(t, []any{
		"accounts::pending",
		&expectedBalance,
		"gold",
		[]any{"A", "B"},
	}, args)
}