fork github.com/doug-martin

2025-03-22 23:02:05 +08:00
commit f14642a736
131 changed files with 34555 additions and 0 deletions
@@ -0,0 +1,81 @@
+package exp_test
+
+import (
+	"testing"
+
+	"git.fsdpf.net/go/db/v2/exp"
+	"github.com/stretchr/testify/suite"
+)
+
+type sqlFunctionExpressionSuite struct {
+	suite.Suite
+	fn exp.SQLFunctionExpression
+}
+
+func TestSQLFunctionExpressionSuite(t *testing.T) {
+	suite.Run(t, &sqlFunctionExpressionSuite{
+		fn: exp.NewSQLFunctionExpression("COUNT", exp.Star()),
+	})
+}
+
+func (sfes *sqlFunctionExpressionSuite) TestClone() {
+	sfes.Equal(sfes.fn, sfes.fn.Clone())
+}
+
+func (sfes *sqlFunctionExpressionSuite) TestExpression() {
+	sfes.Equal(sfes.fn, sfes.fn.Expression())
+}
+
+func (sfes *sqlFunctionExpressionSuite) TestArgs() {
+	sfes.Equal([]interface{}{exp.Star()}, sfes.fn.Args())
+}
+
+func (sfes *sqlFunctionExpressionSuite) TestName() {
+	sfes.Equal("COUNT", sfes.fn.Name())
+}
+
+func (sfes *sqlFunctionExpressionSuite) TestAllOthers() {
+	fn := sfes.fn
+
+	rv := exp.NewRangeVal(1, 2)
+	pattern := "func like%"
+	inVals := []interface{}{1, 2}
+	testCases := []struct {
+		Ex       exp.Expression
+		Expected exp.Expression
+	}{
+		{Ex: fn.As("a"), Expected: exp.NewAliasExpression(fn, "a")},
+		{Ex: fn.Eq(1), Expected: exp.NewBooleanExpression(exp.EqOp, fn, 1)},
+		{Ex: fn.Neq(1), Expected: exp.NewBooleanExpression(exp.NeqOp, fn, 1)},
+		{Ex: fn.Gt(1), Expected: exp.NewBooleanExpression(exp.GtOp, fn, 1)},
+		{Ex: fn.Gte(1), Expected: exp.NewBooleanExpression(exp.GteOp, fn, 1)},
+		{Ex: fn.Lt(1), Expected: exp.NewBooleanExpression(exp.LtOp, fn, 1)},
+		{Ex: fn.Lte(1), Expected: exp.NewBooleanExpression(exp.LteOp, fn, 1)},
+		{Ex: fn.Between(rv), Expected: exp.NewRangeExpression(exp.BetweenOp, fn, rv)},
+		{Ex: fn.NotBetween(rv), Expected: exp.NewRangeExpression(exp.NotBetweenOp, fn, rv)},
+		{Ex: fn.Like(pattern), Expected: exp.NewBooleanExpression(exp.LikeOp, fn, pattern)},
+		{Ex: fn.NotLike(pattern), Expected: exp.NewBooleanExpression(exp.NotLikeOp, fn, pattern)},
+		{Ex: fn.ILike(pattern), Expected: exp.NewBooleanExpression(exp.ILikeOp, fn, pattern)},
+		{Ex: fn.NotILike(pattern), Expected: exp.NewBooleanExpression(exp.NotILikeOp, fn, pattern)},
+		{Ex: fn.RegexpLike(pattern), Expected: exp.NewBooleanExpression(exp.RegexpLikeOp, fn, pattern)},
+		{Ex: fn.RegexpNotLike(pattern), Expected: exp.NewBooleanExpression(exp.RegexpNotLikeOp, fn, pattern)},
+		{Ex: fn.RegexpILike(pattern), Expected: exp.NewBooleanExpression(exp.RegexpILikeOp, fn, pattern)},
+		{Ex: fn.RegexpNotILike(pattern), Expected: exp.NewBooleanExpression(exp.RegexpNotILikeOp, fn, pattern)},
+		{Ex: fn.In(inVals), Expected: exp.NewBooleanExpression(exp.InOp, fn, inVals)},
+		{Ex: fn.NotIn(inVals), Expected: exp.NewBooleanExpression(exp.NotInOp, fn, inVals)},
+		{Ex: fn.Is(true), Expected: exp.NewBooleanExpression(exp.IsOp, fn, true)},
+		{Ex: fn.IsNot(true), Expected: exp.NewBooleanExpression(exp.IsNotOp, fn, true)},
+		{Ex: fn.IsNull(), Expected: exp.NewBooleanExpression(exp.IsOp, fn, nil)},
+		{Ex: fn.IsNotNull(), Expected: exp.NewBooleanExpression(exp.IsNotOp, fn, nil)},
+		{Ex: fn.IsTrue(), Expected: exp.NewBooleanExpression(exp.IsOp, fn, true)},
+		{Ex: fn.IsNotTrue(), Expected: exp.NewBooleanExpression(exp.IsNotOp, fn, true)},
+		{Ex: fn.IsFalse(), Expected: exp.NewBooleanExpression(exp.IsOp, fn, false)},
+		{Ex: fn.IsNotFalse(), Expected: exp.NewBooleanExpression(exp.IsNotOp, fn, false)},
+		{Ex: fn.Desc(), Expected: exp.NewOrderedExpression(fn, exp.DescSortDir, exp.NoNullsSortType)},
+		{Ex: fn.Asc(), Expected: exp.NewOrderedExpression(fn, exp.AscDir, exp.NoNullsSortType)},
+	}
+
+	for _, tc := range testCases {
+		sfes.Equal(tc.Expected, tc.Ex)
+	}
+}