mirror of https://github.com/milvus-io/milvus.git
91 lines
2.8 KiB
Go
91 lines
2.8 KiB
Go
package testcases
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
client "github.com/milvus-io/milvus/client/v2/milvusclient"
|
|
"github.com/milvus-io/milvus/tests/go_client/base"
|
|
"github.com/milvus-io/milvus/tests/go_client/common"
|
|
hp "github.com/milvus-io/milvus/tests/go_client/testcases/helper"
|
|
)
|
|
|
|
// test connect and close, connect again
|
|
func TestConnectClose(t *testing.T) {
|
|
// connect
|
|
ctx := hp.CreateContext(t, time.Second*common.DefaultTimeout)
|
|
mc, errConnect := base.NewMilvusClient(ctx, hp.GetDefaultClientConfig())
|
|
common.CheckErr(t, errConnect, true)
|
|
|
|
// verify that connect success
|
|
listOpt := client.NewListCollectionOption()
|
|
_, errList := mc.ListCollections(ctx, listOpt)
|
|
common.CheckErr(t, errList, true)
|
|
|
|
// close connect and verify
|
|
err := mc.Close(ctx)
|
|
common.CheckErr(t, err, true)
|
|
_, errList2 := mc.ListCollections(ctx, listOpt)
|
|
common.CheckErr(t, errList2, false, "service not ready[SDK=0]: not connected")
|
|
|
|
// connect again
|
|
mc, errConnect2 := base.NewMilvusClient(ctx, hp.GetDefaultClientConfig())
|
|
common.CheckErr(t, errConnect2, true)
|
|
_, errList3 := mc.ListCollections(ctx, listOpt)
|
|
common.CheckErr(t, errList3, true)
|
|
}
|
|
|
|
func genInvalidClientConfig() []client.ClientConfig {
|
|
invalidClientConfigs := []client.ClientConfig{
|
|
{Address: "aaa"}, // not exist address
|
|
{Address: strings.Split(hp.GetAddr(), ":")[0]}, // Address=localhost
|
|
{Address: strings.Split(hp.GetAddr(), ":")[1]}, // Address=19530
|
|
{Address: hp.GetAddr(), Username: "aaa"}, // not exist username
|
|
{Address: hp.GetAddr(), Username: "root", Password: "aaa"}, // wrong password
|
|
{Address: hp.GetAddr(), DBName: "aaa"}, // not exist db
|
|
}
|
|
return invalidClientConfigs
|
|
}
|
|
|
|
// test connect with timeout and invalid addr
|
|
func TestConnectInvalidAddr(t *testing.T) {
|
|
// connect
|
|
ctx := hp.CreateContext(t, time.Second*5)
|
|
for _, invalidCfg := range genInvalidClientConfig() {
|
|
cfg := invalidCfg
|
|
_, errConnect := base.NewMilvusClient(ctx, &cfg)
|
|
common.CheckErr(t, errConnect, false, "context deadline exceeded")
|
|
}
|
|
}
|
|
|
|
// test connect repeatedly
|
|
func TestConnectRepeat(t *testing.T) {
|
|
// connect
|
|
ctx := hp.CreateContext(t, time.Second*10)
|
|
|
|
_, errConnect := base.NewMilvusClient(ctx, hp.GetDefaultClientConfig())
|
|
common.CheckErr(t, errConnect, true)
|
|
|
|
// connect again
|
|
mc, errConnect2 := base.NewMilvusClient(ctx, hp.GetDefaultClientConfig())
|
|
common.CheckErr(t, errConnect2, true)
|
|
|
|
_, err := mc.ListCollections(ctx, client.NewListCollectionOption())
|
|
common.CheckErr(t, err, true)
|
|
}
|
|
|
|
// test close repeatedly
|
|
func TestCloseRepeat(t *testing.T) {
|
|
// connect
|
|
ctx := hp.CreateContext(t, time.Second*10)
|
|
mc, errConnect2 := base.NewMilvusClient(ctx, hp.GetDefaultClientConfig())
|
|
common.CheckErr(t, errConnect2, true)
|
|
|
|
// close and again
|
|
err := mc.Close(ctx)
|
|
common.CheckErr(t, err, true)
|
|
err = mc.Close(ctx)
|
|
common.CheckErr(t, err, true)
|
|
}
|