mirror of
https://github.com/chaisql/chai.git
synced 2025-10-06 16:18:14 +08:00
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package functions_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/chaisql/chai/internal/expr"
|
|
"github.com/chaisql/chai/internal/expr/functions"
|
|
"github.com/chaisql/chai/internal/object"
|
|
"github.com/chaisql/chai/internal/testutil/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDefinitions(t *testing.T) {
|
|
packages := functions.DefaultPackages()
|
|
def, err := packages.GetFunc("", "count")
|
|
assert.NoError(t, err)
|
|
|
|
t.Run("String()", func(t *testing.T) {
|
|
require.Equal(t, "count(arg1)", def.String())
|
|
})
|
|
|
|
t.Run("Function()", func(t *testing.T) {
|
|
fexpr, err := def.Function(expr.Path(object.NewPath("a")))
|
|
assert.NoError(t, err)
|
|
require.NotNil(t, fexpr)
|
|
})
|
|
|
|
t.Run("Arity()", func(t *testing.T) {
|
|
require.Equal(t, 1, def.Arity())
|
|
})
|
|
}
|
|
|
|
func TestPackages(t *testing.T) {
|
|
table := functions.DefaultPackages()
|
|
|
|
t.Run("OK GetFunc()", func(t *testing.T) {
|
|
def, err := table.GetFunc("math", "floor")
|
|
assert.NoError(t, err)
|
|
require.Equal(t, "floor", def.Name())
|
|
def, err = table.GetFunc("", "count")
|
|
assert.NoError(t, err)
|
|
require.Equal(t, "count", def.Name())
|
|
})
|
|
|
|
t.Run("NOK GetFunc() missing func", func(t *testing.T) {
|
|
def, err := table.GetFunc("math", "foobar")
|
|
assert.Error(t, err)
|
|
require.Nil(t, def)
|
|
def, err = table.GetFunc("", "foobar")
|
|
assert.Error(t, err)
|
|
require.Nil(t, def)
|
|
})
|
|
|
|
t.Run("NOK GetFunc() missing package", func(t *testing.T) {
|
|
def, err := table.GetFunc("foobar", "foobar")
|
|
assert.Error(t, err)
|
|
require.Nil(t, def)
|
|
})
|
|
}
|