fix table tests

pingcap · Sep 14, 2023 · 32fb861 · 32fb861
1 parent cb2dbc6
commit 32fb861
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/disttask/framework/storage/table_test.go b/disttask/framework/storage/table_test.go
@@ -80,6 +80,7 @@ func TestGlobalTaskTable(t *testing.T) {
 	require.NoError(t, err)
 	require.Len(t, task4, 1)
 	require.Equal(t, task, task4[0])
+	require.GreaterOrEqual(t, task4[0].StateUpdateTime, task.StateUpdateTime)
 
 	prevState := task.State
 	task.State = proto.TaskStateRunning
@@ -90,11 +91,12 @@ func TestGlobalTaskTable(t *testing.T) {
 	task5, err := gm.GetGlobalTasksInStates(proto.TaskStateRunning)
 	require.NoError(t, err)
 	require.Len(t, task5, 1)
-	require.Equal(t, task, task5[0])
+	require.Equal(t, task.State, task5[0].State)
 
 	task6, err := gm.GetGlobalTaskByKey("key1")
 	require.NoError(t, err)
-	require.Equal(t, task, task6)
+	require.Len(t, task5, 1)
+	require.Equal(t, task.State, task6.State)
 
 	// test cannot insert task with dup key
 	_, err = gm.AddNewGlobalTask("key1", "test2", 4, []byte("test2"))