address a test

tchow-zlai · thomaschow · tchow-zlai · commit ea92661e40dd · 2025-04-18T06:23:15.000-07:00
Co-authored-by: Thomas Chow &lt;thomaschow369@gmail.com&gt;

Co-authored-by: Thomas Chow &lt;thomaschow369@gmail.com&gt;
diff --git a/spark/src/test/scala/ai/chronon/spark/test/ResultValidationAbilityTest.scala b/spark/src/test/scala/ai/chronon/spark/test/ResultValidationAbilityTest.scala
@@ -69,7 +69,7 @@ class ResultValidationAbilityTest extends AnyFlatSpec with BeforeAndAfter {
     val rdd = args.sparkSession.sparkContext.parallelize(leftData)
     val df = args.sparkSession.createDataFrame(rdd).toDF(columns: _*)
 
-    when(mockTableUtils.loadTable(any())).thenReturn(df)
+    when(mockTableUtils.loadTable(any(), any())).thenReturn(df)
 
     assertTrue(args.validateResult(df, Seq("keyId", "ds"), mockTableUtils))
   }
@@ -85,7 +85,7 @@ class ResultValidationAbilityTest extends AnyFlatSpec with BeforeAndAfter {
     val rightRdd = args.sparkSession.sparkContext.parallelize(rightData)
     val rightDf = args.sparkSession.createDataFrame(rightRdd).toDF(columns: _*)
 
-    when(mockTableUtils.loadTable(any())).thenReturn(rightDf)
+    when(mockTableUtils.loadTable(any(), any())).thenReturn(rightDf)
 
     assertFalse(args.validateResult(leftDf, Seq("keyId", "ds"), mockTableUtils))
   }

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ class ResultValidationAbilityTest extends AnyFlatSpec with BeforeAndAfter {`
`69`	`69`	`val rdd = args.sparkSession.sparkContext.parallelize(leftData)`
`70`	`70`	`val df = args.sparkSession.createDataFrame(rdd).toDF(columns: _*)`
`71`	`71`
`72`		`- when(mockTableUtils.loadTable(any())).thenReturn(df)`
	`72`	`+ when(mockTableUtils.loadTable(any(), any())).thenReturn(df)`
`73`	`73`
`74`	`74`	`assertTrue(args.validateResult(df, Seq("keyId", "ds"), mockTableUtils))`
`75`	`75`	`}`
`@@ -85,7 +85,7 @@ class ResultValidationAbilityTest extends AnyFlatSpec with BeforeAndAfter {`
`85`	`85`	`val rightRdd = args.sparkSession.sparkContext.parallelize(rightData)`
`86`	`86`	`val rightDf = args.sparkSession.createDataFrame(rightRdd).toDF(columns: _*)`
`87`	`87`
`88`		`- when(mockTableUtils.loadTable(any())).thenReturn(rightDf)`
	`88`	`+ when(mockTableUtils.loadTable(any(), any())).thenReturn(rightDf)`
`89`	`89`
`90`	`90`	`assertFalse(args.validateResult(leftDf, Seq("keyId", "ds"), mockTableUtils))`
`91`	`91`	`}`