mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
69252f812d
Related to #16298 #16291 #16154 Co-authored-by: sunby <bingyi.sun@zilliz.com> Co-authored-by: yangxuan <xuan.yang@zilliz.com> Co-authored-by: yah01 <yang.cen@zilliz.com> Co-authored-by: Letian Jiang <letian.jiang@zilliz.com> Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
35 lines
994 B
Go
35 lines
994 B
Go
package querynode
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/milvus-io/milvus/internal/util/sessionutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"go.etcd.io/etcd/server/v3/etcdserver/api/v3client"
|
|
)
|
|
|
|
func TestShardClusterService(t *testing.T) {
|
|
client := v3client.New(embedetcdServer.Server)
|
|
defer client.Close()
|
|
session := sessionutil.NewSession(context.Background(), "/by-dev/sessions/unittest/querynode/", client)
|
|
clusterService := newShardClusterService(client, session, nil)
|
|
|
|
assert.NotPanics(t, func() {
|
|
clusterService.addShardCluster(defaultCollectionID, defaultReplicaID, defaultDMLChannel)
|
|
})
|
|
|
|
shardCluster, ok := clusterService.getShardCluster(defaultDMLChannel)
|
|
assert.True(t, ok)
|
|
assert.NotNil(t, shardCluster)
|
|
|
|
_, ok = clusterService.getShardCluster("non-exist-channel")
|
|
assert.False(t, ok)
|
|
|
|
err := clusterService.releaseShardCluster(defaultDMLChannel)
|
|
assert.NoError(t, err)
|
|
|
|
err = clusterService.releaseShardCluster("non-exist-channel")
|
|
assert.Error(t, err)
|
|
}
|