From dcdf85977cda9ebe966f54754b8976cee939313a Mon Sep 17 00:00:00 2001 From: Bingyi Sun Date: Fri, 2 Feb 2024 15:25:04 +0800 Subject: [PATCH] enhance: [skip e2e] add alter collection task ut (#30447) Signed-off-by: sunby --- internal/proxy/task_test.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/internal/proxy/task_test.go b/internal/proxy/task_test.go index 59772a761d..ef4d03cce0 100644 --- a/internal/proxy/task_test.go +++ b/internal/proxy/task_test.go @@ -3597,3 +3597,22 @@ func TestClusteringKey(t *testing.T) { assert.Error(t, err) }) } + +func TestAlterCollectionCheckLoaded(t *testing.T) { + qc := &mocks.MockQueryCoordClient{} + qc.EXPECT().ShowCollections(mock.Anything, mock.Anything).Return(&querypb.ShowCollectionsResponse{ + Status: &commonpb.Status{ErrorCode: commonpb.ErrorCode_Success}, + CollectionIDs: []int64{1}, + InMemoryPercentages: []int64{100}, + }, nil) + task := &alterCollectionTask{ + AlterCollectionRequest: &milvuspb.AlterCollectionRequest{ + Base: &commonpb.MsgBase{}, + CollectionID: 1, + Properties: []*commonpb.KeyValuePair{{Key: common.MmapEnabledKey, Value: "true"}}, + }, + queryCoord: qc, + } + err := task.PreExecute(context.Background()) + assert.Equal(t, merr.Code(merr.ErrCollectionLoaded), merr.Code(err)) +}