fix indexnode ut

Signed-off-by: Cai Zhang <[email protected]>
milvus-io · Jan 8, 2025 · 5cc499d · 5cc499d
1 parent 8e637ee
commit 5cc499d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/internal/indexnode/indexnode_test.go b/internal/indexnode/indexnode_test.go
@@ -675,7 +675,7 @@ func (s *IndexNodeSuite) Test_CreateStatsTask() {
 		err = merr.Error(slotResp.GetStatus())
 		s.NoError(err)
 
-		s.Equal(int64(1), slotResp.GetTaskSlots())
+		s.Equal(int64(16), slotResp.GetTaskSlots())
 
 		status, err = s.in.DropJobsV2(ctx, &workerpb.DropJobsV2Request{
 			ClusterID: "cluster2",

diff --git a/internal/indexnode/task_scheduler_test.go b/internal/indexnode/task_scheduler_test.go
@@ -87,6 +87,10 @@ func (t *fakeTask) Ctx() context.Context {
 	return t.ctx
 }
 
+func (t *fakeTask) GetSlot() int64 {
+	return 1
+}
+
 func (t *fakeTask) OnEnqueue(ctx context.Context) error {
 	_taskwg.Add(1)
 	t.state = fakeTaskEnqueued