milvus/pkg/util/expr/expr_test.go
SimFG dd9c61831d
enhance: Support to get the param value in the runtime (#29297)
/kind improvement
issue: #29299

Signed-off-by: SimFG <bang.fu@zilliz.com>
2023-12-22 18:36:44 +08:00

64 lines
1.6 KiB
Go

/*
* Licensed to the LF AI & Data foundation under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 expr
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/milvus-io/milvus/pkg/util/paramtable"
)
func TestExec(t *testing.T) {
paramtable.Init()
t.Run("not init", func(t *testing.T) {
_, err := Exec("1+1", "by-dev")
assert.Error(t, err)
})
Init()
Register("foo", "hello")
t.Run("empty code", func(t *testing.T) {
_, err := Exec("", "by-dev")
assert.Error(t, err)
})
t.Run("empty auth", func(t *testing.T) {
_, err := Exec("1+1", "")
assert.Error(t, err)
})
t.Run("invalid auth", func(t *testing.T) {
_, err := Exec("1+1", "000")
assert.Error(t, err)
})
t.Run("invalid code", func(t *testing.T) {
_, err := Exec("1+", "by-dev")
assert.Error(t, err)
})
t.Run("valid code", func(t *testing.T) {
out, err := Exec("foo", "by-dev")
assert.NoError(t, err)
assert.Equal(t, "hello", out)
})
}