Skip to content

Commit 06b6a38

Browse files
committed
test(scheduler): fix StatFile test cases for non-directory URL handling
Signed-off-by: Gaius <gaius.qi@gmail.com>
1 parent 744c6fb commit 06b6a38

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

scheduler/service/service_v2_test.go

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4328,7 +4328,7 @@ func TestServiceV2_StatFile(t *testing.T) {
43284328
{
43294329
name: "list task entries failed",
43304330
req: &schedulerv2.StatFileRequest{
4331-
Url: "https://example.com/file.txt",
4331+
Url: "https://example.com/dir/",
43324332
},
43334333
run: func(t *testing.T, svc *V2, req *schedulerv2.StatFileRequest, mj *jobmocks.MockJobMockRecorder) {
43344334
mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New("list task entries failed")).Times(1)
@@ -4350,7 +4350,6 @@ func TestServiceV2_StatFile(t *testing.T) {
43504350
defer wg.Wait()
43514351

43524352
gomock.InOrder(
4353-
mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
43544353
mj.GetTask(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.GetTaskRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(nil, errors.New("get task failed")).Times(1),
43554354
)
43564355

@@ -4371,7 +4370,6 @@ func TestServiceV2_StatFile(t *testing.T) {
43714370
defer wg.Wait()
43724371

43734372
gomock.InOrder(
4374-
mj.ListTaskEntries(gomock.Any(), gomock.Any(), gomock.Any()).Return(&internaljob.ListTaskEntriesResponse{Entries: []*dfdaemonv2.Entry{{Url: "https://example.com/file.txt"}}}, nil).Times(1),
43754373
mj.GetTask(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(context.Context, *internaljob.GetTaskRequest, *logger.SugaredLoggerOnWith) { wg.Done() }).Return(&internaljob.GetTaskResponse{Peers: []*internaljob.Peer{{IP: "127.0.0.1", Hostname: "peer-1"}}}, nil).Times(1),
43764374
)
43774375

0 commit comments

Comments
 (0)