test(scheduler): remove ListTaskEntries mocks from PreheatFile tests

Signed-off-by: Gaius <gaius.qi@gmail.com>

Author: gaius-qi

scheduler/service/service_v2_test.go

@@ -4163,10 +4163,6 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				Scope: "invalid-scope",
 			},
 			run: func(t *testing.T, svc *V2, req *schedulerv2.PreheatFileRequest, mj *jobmocks.MockJobMockRecorder) {
-				mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{
-					Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}},
-				}, nil).Times(1)
-
 				assert := assert.New(t)
 				assert.ErrorIs(svc.PreheatFile(context.Background(), req), status.Errorf(codes.InvalidArgument, "unsupported preheat scope: invalid-scope"))
 			},
@@ -4182,7 +4178,6 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatSingleSeedPeer(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(nil, nil).Times(1),
 				)
 
@@ -4202,9 +4197,9 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatSingleSeedPeer(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(nil, nil).Times(1),
 				)
+
 				assert := assert.New(t)
 				assert.NoError(svc.PreheatFile(context.Background(), req))
 			},
@@ -4221,9 +4216,9 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatSingleSeedPeer(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(nil, nil).Times(1),
 				)
+
 				assert := assert.New(t)
 				assert.NoError(svc.PreheatFile(context.Background(), req))
 			},
@@ -4240,9 +4235,9 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatAllSeedPeers(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(&internaljob.PreheatResponse{SuccessTasks: make([]*internaljob.PreheatSuccessTask, 0), FailureTasks: make([]*internaljob.PreheatFailureTask, 0)}, nil).Times(1),
 				)
+
 				assert := assert.New(t)
 				assert.NoError(svc.PreheatFile(context.Background(), req))
 			},
@@ -4259,9 +4254,9 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatAllSeedPeers(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(&internaljob.PreheatResponse{SuccessTasks: make([]*internaljob.PreheatSuccessTask, 0), FailureTasks: make([]*internaljob.PreheatFailureTask, 0)}, nil).Times(1),
 				)
+
 				assert := assert.New(t)
 				assert.NoError(svc.PreheatFile(context.Background(), req))
 			},
@@ -4278,9 +4273,9 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatAllPeers(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(&internaljob.PreheatResponse{SuccessTasks: make([]*internaljob.PreheatSuccessTask, 0), FailureTasks: make([]*internaljob.PreheatFailureTask, 0)}, nil).Times(1),
 				)
+
 				assert := assert.New(t)
 				assert.NoError(svc.PreheatFile(context.Background(), req))
 			},
@@ -4297,7 +4292,6 @@ func TestServiceV2_PreheatFile(t *testing.T) {
 				defer wg.Wait()
 
 				gomock.InOrder(
-					mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
 					mj.PreheatAllPeers(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.PreheatRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(&internaljob.PreheatResponse{SuccessTasks: make([]*internaljob.PreheatSuccessTask, 0), FailureTasks: make([]*internaljob.PreheatFailureTask, 0)}, nil).Times(1),
 				)