fix 340

binmahone · binmahone · commit 48a20116ad5f · 2025-02-25T12:12:18.000+08:00
Signed-off-by: Hongbin Ma (Mahone) &lt;mahongbin@apache.org&gt;
diff --git a/tests/src/test/spark340/scala/com/nvidia/spark/rapids/shuffle/RapidsShuffleTestHelper.scala b/tests/src/test/spark340/scala/com/nvidia/spark/rapids/shuffle/RapidsShuffleTestHelper.scala
@@ -175,7 +175,7 @@ abstract class RapidsShuffleTestHelper
   }
 }
 
-object RapidsShuffleTestHelper extends MockitoSugar {
+object RapidsShuffleTestHelper extends MockitoSugar with MoreMockitoSugar {
   def buildMockTableMeta(tableId: Int, contigTable: ContiguousTable): TableMeta = {
     MetaUtils.buildTableMeta(tableId, contigTable)
   }

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ abstract class RapidsShuffleTestHelper`
`175`	`175`	`}`
`176`	`176`	`}`
`177`	`177`
`178`		`-object RapidsShuffleTestHelper extends MockitoSugar {`
	`178`	`+object RapidsShuffleTestHelper extends MockitoSugar with MoreMockitoSugar {`
`179`	`179`	`def buildMockTableMeta(tableId: Int, contigTable: ContiguousTable): TableMeta = {`
`180`	`180`	`MetaUtils.buildTableMeta(tableId, contigTable)`
`181`	`181`	`}`