test: fix flaky case (#50008)

close #49984
pingcap · Jan 3, 2024 · a28d46b · a28d46b
1 parent c082f85
commit a28d46b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/disttask/framework/storage/table_test.go b/pkg/disttask/framework/storage/table_test.go
@@ -208,6 +208,9 @@ func TestSwitchTaskStep(t *testing.T) {
 	require.Equal(t, []byte("changed meta"), task.Meta)
 	checkAfterSwitchStep(t, startTime, task, subtasksStepOne, proto.StepOne)
 	// switch step again, no effect
+	// some fields are changed in prev call, change back.
+	task.State = proto.TaskStatePending
+	task.Step = proto.StepInit
 	require.NoError(t, tm.SwitchTaskStep(ctx, task, proto.TaskStateRunning, proto.StepOne, subtasksStepOne))
 	task, err = tm.GetTaskByID(ctx, taskID)
 	require.NoError(t, err)