update

Co-authored-by: Thomas Chow <[email protected]>

Author: tchow-zlai

spark/src/test/scala/ai/chronon/spark/test/TableUtilsTest.scala

@@ -26,7 +26,7 @@ import org.junit.Assert.{assertEquals, assertTrue}
 import org.scalatest.flatspec.AnyFlatSpec
 import org.junit.Assert.assertNull
 
-import ai.chronon.spark.format.FormatProvider
+import ai.chronon.spark.format.{FormatProvider, Format}
 import ai.chronon.spark.format.DefaultFormatProvider
 import org.apache.spark.sql.catalyst.parser.ParseException
 
@@ -647,13 +647,13 @@ class TableUtilsTest extends AnyFlatSpec {
 
   it should "test catalog detection" in {
     val fp = FormatProvider.from(spark).asInstanceOf[DefaultFormatProvider]
-    assertEquals("catalogA", fp.getCatalog("catalogA.foo.bar"))
-    assertEquals("catalogA", fp.getCatalog("`catalogA`.foo.bar"))
-    assertEquals("spark_catalog", fp.getCatalog("`catalogA.foo`.bar"))
-    assertEquals("spark_catalog", fp.getCatalog("`catalogA.foo.bar`"))
-    assertEquals("spark_catalog", fp.getCatalog("foo.bar"))
-    assertEquals("spark_catalog", fp.getCatalog("bar"))
-    assertThrows[ParseException](fp.getCatalog(""))
+    assertEquals("catalogA", Format.getCatalog("catalogA.foo.bar")(spark))
+    assertEquals("catalogA", Format.getCatalog("`catalogA`.foo.bar")(spark))
+    assertEquals("spark_catalog", Format.getCatalog("`catalogA.foo`.bar")(spark))
+    assertEquals("spark_catalog", Format.getCatalog("`catalogA.foo.bar`")(spark))
+    assertEquals("spark_catalog", Format.getCatalog("foo.bar")(spark))
+    assertEquals("spark_catalog", Format.getCatalog("bar")(spark))
+    assertThrows[ParseException](Format.getCatalog("")(spark))
   }
 
 }