From 532d62a3d78a571af0f0f7ee485e224a5ea65182 Mon Sep 17 00:00:00 2001 From: congqixia Date: Mon, 16 Dec 2024 14:24:37 +0800 Subject: [PATCH] fix: Set show partition in-memory & fix case (#857) Related to #374 - Set show partition request in-memory req type - Fix test case logic --------- Signed-off-by: Congqi Xia --- test/testcases/load_release_test.go | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/test/testcases/load_release_test.go b/test/testcases/load_release_test.go index 61c4a2d2..887927da 100644 --- a/test/testcases/load_release_test.go +++ b/test/testcases/load_release_test.go @@ -297,7 +297,6 @@ func TestLoadPartitionsRepeatedly(t *testing.T) { // test load partitions async func TestLoadPartitionsAsync(t *testing.T) { - t.Skipf("Issue: https://github.com/milvus-io/milvus-sdk-go/issues/374") ctx := createContext(t, time.Second*common.DefaultTimeout) // connect mc := createMilvusClient(ctx, t) @@ -319,13 +318,11 @@ func TestLoadPartitionsAsync(t *testing.T) { time.Sleep(time.Second * 5) // check partition loaded - partitions, errShow := mc.ShowPartitions(ctx, collName) - if errShow == nil { - for _, p := range partitions { - log.Printf("id: %d, name: %s, loaded %t", p.ID, p.Name, p.Loaded) - if p.Name == partitionName && p.Loaded { - break - } + state, err := mc.GetLoadState(ctx, collName, []string{partitionName}) + // partitions, errShow := mc.ShowPartitions(ctx, collName) + if err == nil { + if state == entity.LoadStateLoaded { + return } } else { t.FailNow()