From b5a2fad42e8584156b4d2c72b1ea06615305d38c Mon Sep 17 00:00:00 2001 From: ywqzzy <592838129@qq.com> Date: Fri, 23 Feb 2024 15:35:51 +0800 Subject: [PATCH] fix --- tests/realtikvtest/importintotest4/global_sort_test.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/realtikvtest/importintotest4/global_sort_test.go b/tests/realtikvtest/importintotest4/global_sort_test.go index 32f4da62c02ae..6427e7a563e0d 100644 --- a/tests/realtikvtest/importintotest4/global_sort_test.go +++ b/tests/realtikvtest/importintotest4/global_sort_test.go @@ -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" @@ -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