fix UT

pingcap · Apr 11, 2019 · 55351a7 · 55351a7
1 parent ccd9b51
commit 55351a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/executor/executor_required_rows_test.go b/executor/executor_required_rows_test.go
@@ -457,7 +457,7 @@ func (s *testExecSuite) TestSelectionRequiredRows(c *C) {
 		chk := exec.newFirstChunk()
 		for i := range testCase.requiredRows {
 			chk.SetRequiredRows(testCase.requiredRows[i], maxChunkSize)
-			c.Assert(exec.Next(ctx, chunk.NewRecordBatch(chk)), IsNil)
+			c.Assert(exec.Next(ctx, chk), IsNil)
 			c.Assert(chk.NumRows(), Equals, testCase.expectedRows[i])
 		}
 		c.Assert(exec.Close(), IsNil)
@@ -515,7 +515,7 @@ func (s *testExecSuite) TestProjectionUnparallelRequiredRows(c *C) {
 		chk := exec.newFirstChunk()
 		for i := range testCase.requiredRows {
 			chk.SetRequiredRows(testCase.requiredRows[i], maxChunkSize)
-			c.Assert(exec.Next(ctx, chunk.NewRecordBatch(chk)), IsNil)
+			c.Assert(exec.Next(ctx, chk), IsNil)
 			c.Assert(chk.NumRows(), Equals, testCase.expectedRows[i])
 		}
 		c.Assert(exec.Close(), IsNil)
@@ -570,7 +570,7 @@ func (s *testExecSuite) TestProjectionParallelRequiredRows(c *C) {
 		chk := exec.newFirstChunk()
 		for i := range testCase.requiredRows {
 			chk.SetRequiredRows(testCase.requiredRows[i], maxChunkSize)
-			c.Assert(exec.Next(ctx, chunk.NewRecordBatch(chk)), IsNil)
+			c.Assert(exec.Next(ctx, chk), IsNil)
 			c.Assert(chk.NumRows(), Equals, testCase.expectedRows[i])
 
 			// wait projectionInputFetcher blocked on fetching data