planner: Updated failing test after #33098 (#33299)

close #33297
pingcap · Mar 22, 2022 · b7a2d62 · b7a2d62
1 parent 768ce5f
commit b7a2d62
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/planner/core/prepare_test.go b/planner/core/prepare_test.go
@@ -2630,10 +2630,16 @@ func TestPartitionWithVariedDataSources(t *testing.T) {
 		tk.MustExec(fmt.Sprintf(`set @a0=%v, @a1=%v, @a2=%v`, rand.Intn(40000), rand.Intn(40000), rand.Intn(40000)))
 
 		var rscan, rlookup, rpoint, rbatch [][]interface{}
+		var expectedFromPlanCache string
 		for id, tbl := range []string{"trangeIdx", "thashIdx", "tnormalIdx"} {
 			scan := tk.MustQuery(fmt.Sprintf(`execute stmt%v_indexscan using @mina, @maxa`, tbl)).Sort()
+			if id == 2 {
+				expectedFromPlanCache = "1"
+			} else {
+				expectedFromPlanCache = "0"
+			}
 			if i > 0 {
-				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows("1"))
+				tk.MustQuery(`select @@last_plan_from_cache /* table: ` + tbl + " */").Check(testkit.Rows(expectedFromPlanCache))
 			}
 			if id == 0 {
 				rscan = scan.Rows()
@@ -2643,7 +2649,7 @@ func TestPartitionWithVariedDataSources(t *testing.T) {
 
 			lookup := tk.MustQuery(fmt.Sprintf(`execute stmt%v_indexlookup using @mina, @maxa`, tbl)).Sort()
 			if i > 0 {
-				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows("1"))
+				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows(expectedFromPlanCache))
 			}
 			if id == 0 {
 				rlookup = lookup.Rows()
@@ -2655,7 +2661,7 @@ func TestPartitionWithVariedDataSources(t *testing.T) {
 			if tbl == `tnormalPK` && i > 0 {
 				// PlanCache cannot support PointGet now since we haven't relocated partition after rebuilding range.
 				// Please see Execute.rebuildRange for more details.
-				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows("0"))
+				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows(expectedFromPlanCache))
 			}
 			if id == 0 {
 				rpoint = point.Rows()
@@ -2665,7 +2671,7 @@ func TestPartitionWithVariedDataSources(t *testing.T) {
 
 			batch := tk.MustQuery(fmt.Sprintf(`execute stmt%v_batchget_idx using @a0, @a1, @a2`, tbl)).Sort()
 			if i > 0 {
-				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows("1"))
+				tk.MustQuery(`select @@last_plan_from_cache`).Check(testkit.Rows(expectedFromPlanCache))
 			}
 			if id == 0 {
 				rbatch = batch.Rows()

diff --git a/unstable.txt b/unstable.txt
@@ -27,8 +27,7 @@ planner/core TestPrepareCacheNow
 planner/core TestDAGPlanBuilderSplitAvg
 planner/core TestIndexMerge
 planner/core TestUnmatchedTableInHint
-planner/core TestPartitionWithVariedDataSources
 planner/core TestCopPaging
 planner/core TestPlanCachePointGetAndTableDual
 planner/core TestSimplifyOuterJoinWithCast 
-executor/seqtest TestShow
+executor/seqtest TestShow