milvus/internal/util/funcutil/func_test.go
dragondriver 6796c7bfe7 Add unittest to function util
Signed-off-by: dragondriver <jiquan.long@zilliz.com>
2021-03-31 18:22:17 +08:00

44 lines
913 B
Go

package funcutil
import (
"encoding/json"
"strconv"
"testing"
"github.com/stretchr/testify/assert"
)
func TestCheckPortAvailable(t *testing.T) {
num := 10
for i := 0; i < num; i++ {
port := GetAvailablePort()
assert.Equal(t, CheckPortAvailable(port), true)
}
}
func TestParseIndexParamsMap(t *testing.T) {
num := 10
keys := make([]string, 0)
values := make([]string, 0)
params := make(map[string]string)
for i := 0; i < num; i++ {
keys = append(keys, "key"+strconv.Itoa(i))
values = append(values, "value"+strconv.Itoa(i))
params[keys[i]] = values[i]
}
paramsBytes, err := json.Marshal(params)
assert.Equal(t, err, nil)
paramsStr := string(paramsBytes)
parsedParams, err := ParseIndexParamsMap(paramsStr)
assert.Equal(t, err, nil)
assert.Equal(t, parsedParams, params)
invalidStr := "invalid string"
_, err = ParseIndexParamsMap(invalidStr)
assert.NotEqual(t, err, nil)
}