milvus/internal/streamingcoord/server/service_test.go

44 lines
1.0 KiB
Go

package server
import (
"context"
"strings"
"testing"
"github.com/stretchr/testify/assert"
etcdkv "github.com/milvus-io/milvus/internal/kv/etcd"
"github.com/milvus-io/milvus/internal/types"
"github.com/milvus-io/milvus/internal/util/sessionutil"
"github.com/milvus-io/milvus/pkg/util/etcd"
"github.com/milvus-io/milvus/pkg/util/paramtable"
"github.com/milvus-io/milvus/pkg/util/syncutil"
)
func TestServer(t *testing.T) {
paramtable.Init()
params := paramtable.Get()
endpoints := params.EtcdCfg.Endpoints.GetValue()
etcdEndpoints := strings.Split(endpoints, ",")
c, err := etcd.GetRemoteEtcdClient(etcdEndpoints)
assert.NoError(t, err)
assert.NotNil(t, c)
b := NewServerBuilder()
metaKV := etcdkv.NewEtcdKV(c, "test")
s := sessionutil.NewMockSession(t)
f := syncutil.NewFuture[types.RootCoordClient]()
newServer := b.WithETCD(c).
WithMetaKV(metaKV).
WithSession(s).
WithRootCoordClient(f).
Build()
ctx := context.Background()
err = newServer.Start(ctx)
assert.NoError(t, err)
newServer.Stop()
}