github.com/dolthub/go-mysql-server@v0.18.0/sql/rowexec/bindvar_test.go

// Copyright 2020-2021 Dolthub, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package rowexec

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/dolthub/go-mysql-server/sql"
	"github.com/dolthub/go-mysql-server/sql/expression"
	"github.com/dolthub/go-mysql-server/sql/plan"
	"github.com/dolthub/go-mysql-server/sql/types"
)

func TestApplyBindings(t *testing.T) {
	type tc struct {
		Name     string
		Node     sql.Node
		Bindings map[string]sql.Expression
		Expected sql.Node
	}
	cases := []tc{
		tc{
			"SingleV1",
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewEquals(
						expression.NewUnresolvedColumn("foo"),
						expression.NewBindVar("v1"),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
			map[string]sql.Expression{
				"v1": expression.NewLiteral("Four score and seven years ago...", types.LongText),
			},
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewEquals(
						expression.NewUnresolvedColumn("foo"),
						expression.NewLiteral("Four score and seven years ago...", types.LongText),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
		},
		tc{
			"VarNotBound",
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewEquals(
						expression.NewUnresolvedColumn("foo"),
						expression.NewBindVar("v1"),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
			map[string]sql.Expression{},
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewEquals(
						expression.NewUnresolvedColumn("foo"),
						expression.NewBindVar("v1"),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
		},
		tc{
			"SameVarMultipleTimes",
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewOr(
						expression.NewAnd(
							expression.NewEquals(
								expression.NewUnresolvedColumn("foo"),
								expression.NewBindVar("strvar"),
							),
							expression.NewEquals(
								expression.NewUnresolvedColumn("bar"),
								expression.NewBindVar("strvar"),
							),
						),
						expression.NewLessThan(
							expression.NewUnresolvedColumn("icol"),
							expression.NewBindVar("intvar"),
						),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
			map[string]sql.Expression{
				"strvar": expression.NewLiteral("Four score and seven years ago...", types.LongText),
				"intvar": expression.NewLiteral(int8(10), types.Int8),
			},
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewFilter(
					expression.NewOr(
						expression.NewAnd(
							expression.NewEquals(
								expression.NewUnresolvedColumn("foo"),
								expression.NewLiteral("Four score and seven years ago...", types.LongText),
							),
							expression.NewEquals(
								expression.NewUnresolvedColumn("bar"),
								expression.NewLiteral("Four score and seven years ago...", types.LongText),
							),
						),
						expression.NewLessThan(
							expression.NewUnresolvedColumn("icol"),
							expression.NewLiteral(int8(10), types.Int8),
						),
					),
					plan.NewUnresolvedTable("t1", ""),
				),
			),
		},
		tc{
			"Subquery",
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewSubqueryAlias(
					"a",
					"select * from foo where bar = :v1",
					plan.NewProject(
						[]sql.Expression{
							expression.NewStar(),
						},
						plan.NewFilter(
							expression.NewEquals(
								expression.NewUnresolvedColumn("bar"),
								expression.NewBindVar("v1"),
							),
							plan.NewUnresolvedTable("foo", ""),
						),
					),
				),
			),
			map[string]sql.Expression{
				"v1": expression.NewLiteral("Four score and seven years ago...", types.LongText),
			},
			plan.NewProject(
				[]sql.Expression{
					expression.NewStar(),
				},
				plan.NewSubqueryAlias(
					"a",
					"select * from foo where bar = :v1",
					plan.NewProject(
						[]sql.Expression{
							expression.NewStar(),
						},
						plan.NewFilter(
							expression.NewEquals(
								expression.NewUnresolvedColumn("bar"),
								expression.NewLiteral("Four score and seven years ago...", types.LongText),
							),
							plan.NewUnresolvedTable("foo", ""),
						),
					),
				),
			),
		},
	}

	for _, c := range cases {
		t.Run(c.Name, func(t *testing.T) {
			res, _, err := plan.ApplyBindings(c.Node, c.Bindings)
			if assert.NoError(t, err) {
				assert.Equal(t, res, c.Expected)
			}
		})
	}
}