From 9364d0ea492f3be3e5dce6876e751d81738afb85 Mon Sep 17 00:00:00 2001 From: congqixia Date: Mon, 28 Aug 2023 11:20:25 +0800 Subject: [PATCH] Remove etcd dependency for querycoord unit test (#26550) Signed-off-by: Congqi Xia --- internal/querycoordv2/server_test.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/internal/querycoordv2/server_test.go b/internal/querycoordv2/server_test.go index 32f8d9c034..d1047c0f37 100644 --- a/internal/querycoordv2/server_test.go +++ b/internal/querycoordv2/server_test.go @@ -18,6 +18,9 @@ package querycoordv2 import ( "context" + "math/rand" + "os" + "strings" "testing" "time" @@ -46,6 +49,24 @@ import ( "github.com/milvus-io/milvus/pkg/util/paramtable" ) +func TestMain(m *testing.M) { + // init embed etcd + embedetcdServer, tempDir, err := etcd.StartTestEmbedEtcdServer() + if err != nil { + log.Fatal(err.Error()) + } + defer os.RemoveAll(tempDir) + defer embedetcdServer.Close() + + addrs := etcd.GetEmbedEtcdEndpoints(embedetcdServer) + + paramtable.Init() + paramtable.Get().Save(Params.EtcdCfg.Endpoints.Key, strings.Join(addrs, ",")) + + rand.Seed(time.Now().UnixNano()) + os.Exit(m.Run()) +} + type ServerSuite struct { suite.Suite