mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 17:48:29 +08:00
858dc10ef9
issue: #38399 pr: #39522 - broadcast message can carry multi resource key now. - implement event-based notification for broadcast messages - broadcast message use broadcast id as a unique identifier in message - broadcasted message on vchannels keep the broadcasted vchannel now. - broadcasted message and broadcast message have a common broadcast header now. --------- Signed-off-by: chyezh <chyezh@outlook.com>
80 lines
2.7 KiB
Go
80 lines
2.7 KiB
Go
package coordclient
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus/pkg/proto/datapb"
|
|
"github.com/milvus-io/milvus/pkg/proto/querypb"
|
|
"github.com/milvus-io/milvus/pkg/proto/rootcoordpb"
|
|
"github.com/milvus-io/milvus/pkg/util/paramtable"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
func TestRegistry(t *testing.T) {
|
|
paramtable.Init()
|
|
paramtable.Get().Save(paramtable.Get().CommonCfg.LocalRPCEnabled.Key, "true")
|
|
|
|
assert.False(t, enableLocal.EnableQueryCoord)
|
|
assert.False(t, enableLocal.EnableDataCoord)
|
|
assert.False(t, enableLocal.EnableRootCoord)
|
|
|
|
EnableLocalClientRole(&LocalClientRoleConfig{
|
|
ServerType: typeutil.RootCoordRole,
|
|
EnableQueryCoord: true,
|
|
EnableDataCoord: true,
|
|
EnableRootCoord: true,
|
|
})
|
|
assert.False(t, enableLocal.EnableQueryCoord)
|
|
assert.False(t, enableLocal.EnableDataCoord)
|
|
assert.False(t, enableLocal.EnableRootCoord)
|
|
|
|
RegisterRootCoordServer(&rootcoordpb.UnimplementedRootCoordServer{})
|
|
RegisterDataCoordServer(&datapb.UnimplementedDataCoordServer{})
|
|
RegisterQueryCoordServer(&querypb.UnimplementedQueryCoordServer{})
|
|
assert.True(t, glocalClient.dataCoordClient.Ready())
|
|
assert.True(t, glocalClient.queryCoordClient.Ready())
|
|
assert.True(t, glocalClient.rootCoordClient.Ready())
|
|
ResetRegistration()
|
|
|
|
enableLocal = &LocalClientRoleConfig{}
|
|
|
|
EnableLocalClientRole(&LocalClientRoleConfig{
|
|
ServerType: typeutil.StandaloneRole,
|
|
EnableQueryCoord: true,
|
|
EnableDataCoord: true,
|
|
EnableRootCoord: true,
|
|
})
|
|
assert.True(t, enableLocal.EnableDataCoord)
|
|
assert.True(t, enableLocal.EnableQueryCoord)
|
|
assert.True(t, enableLocal.EnableRootCoord)
|
|
|
|
RegisterRootCoordServer(&rootcoordpb.UnimplementedRootCoordServer{})
|
|
RegisterDataCoordServer(&datapb.UnimplementedDataCoordServer{})
|
|
RegisterQueryCoordServer(&querypb.UnimplementedQueryCoordServer{})
|
|
assert.True(t, glocalClient.dataCoordClient.Ready())
|
|
assert.True(t, glocalClient.queryCoordClient.Ready())
|
|
assert.True(t, glocalClient.rootCoordClient.Ready())
|
|
|
|
enableLocal = &LocalClientRoleConfig{}
|
|
|
|
EnableLocalClientRole(&LocalClientRoleConfig{
|
|
ServerType: typeutil.MixtureRole,
|
|
EnableQueryCoord: true,
|
|
EnableDataCoord: true,
|
|
EnableRootCoord: true,
|
|
})
|
|
assert.True(t, enableLocal.EnableDataCoord)
|
|
assert.True(t, enableLocal.EnableQueryCoord)
|
|
assert.True(t, enableLocal.EnableRootCoord)
|
|
|
|
assert.NotNil(t, GetQueryCoordClient(context.Background()))
|
|
assert.NotNil(t, GetDataCoordClient(context.Background()))
|
|
assert.NotNil(t, GetRootCoordClient(context.Background()))
|
|
GetQueryCoordClient(context.Background()).Close()
|
|
GetDataCoordClient(context.Background()).Close()
|
|
GetRootCoordClient(context.Background()).Close()
|
|
}
|