Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ywqzzy committed Feb 23, 2024
1 parent 2bf88b2 commit b5a2fad
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/realtikvtest/importintotest4/global_sort_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ import (
"time"

"github.com/fsouza/fake-gcs-server/fakestorage"
"github.com/pingcap/tidb/pkg/disttask/framework/proto"
"github.com/pingcap/tidb/pkg/disttask/framework/scheduler"
"github.com/pingcap/tidb/pkg/disttask/framework/storage"
"github.com/pingcap/tidb/pkg/disttask/importinto"
Expand Down Expand Up @@ -117,7 +118,7 @@ func (s *mockGCSSuite) TestGlobalSortBasic() {
s.Eventually(func() bool {
task, err2 = taskManager.GetTaskByKeyWithHistory(ctx, importinto.TaskKey(int64(jobID)))
s.NoError(err2)
return task.State == "failed"
return task.State == proto.TaskStateReverted
}, 30*time.Second, 300*time.Millisecond)
// check all sorted data cleaned up
<-scheduler.WaitCleanUpFinished
Expand Down

0 comments on commit b5a2fad

Please sign in to comment.