mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 17:48:29 +08:00
2b4f211d84
issue: #33285 - Add switch for local rpc --------- Signed-off-by: chyezh <chyezh@outlook.com>
79 lines
2.7 KiB
Go
79 lines
2.7 KiB
Go
package coordclient
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus/internal/proto/datapb"
|
|
"github.com/milvus-io/milvus/internal/proto/querypb"
|
|
"github.com/milvus-io/milvus/internal/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.False(t, glocalClient.dataCoordClient.Ready())
|
|
assert.False(t, glocalClient.queryCoordClient.Ready())
|
|
assert.False(t, glocalClient.rootCoordClient.Ready())
|
|
|
|
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()
|
|
}
|