[NSE-436] Fix for Arrow Data Source test suite (#437)

Closes #436
oap-project · Jul 28, 2021 · 0b9626e · 0b9626e
1 parent 5bcd805
commit 0b9626e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.../test/scala/com/intel/oap/spark/sql/execution/datasources/arrow/ArrowDataSourceTest.scala b/.../test/scala/com/intel/oap/spark/sql/execution/datasources/arrow/ArrowDataSourceTest.scala
@@ -277,7 +277,10 @@ class ArrowDataSourceTest extends QueryTest with SharedSparkSession {
 
   test("file descriptor leak - v1") {
     val path = ArrowDataSourceTest.locateResourcePath(parquetFile1)
-    spark.catalog.createTable("ptab2", path, "arrow")
+    val frame = spark.read
+        .option(ArrowOptions.KEY_ORIGINAL_FORMAT, "parquet")
+        .arrow(path)
+    frame.createOrReplaceTempView("ptab2")
 
     def getFdCount: Long = {
       ManagementFactory.getOperatingSystemMXBean