diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestHiveUdfsJar.scala b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestHiveUdfsJar.scala index 73d88c392f109..3885d0df56d69 100644 --- a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestHiveUdfsJar.scala +++ b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestHiveUdfsJar.scala @@ -20,7 +20,6 @@ package org.apache.spark.sql.hive.test import java.io.File import java.lang.management.ManagementFactory import java.nio.charset.StandardCharsets -import java.nio.file.{Files, Paths} import org.apache.spark.util.{SparkTestUtils, Utils} @@ -64,6 +63,8 @@ object TestHiveUdfsJar { private def readResource(name: String): String = { val url = Thread.currentThread().getContextClassLoader.getResource(name) assert(url != null, s"Resource not found: $name") - new String(Files.readAllBytes(Paths.get(url.toURI)), StandardCharsets.UTF_8) + Utils.tryWithResource(url.openStream()) { is => + new String(is.readAllBytes(), StandardCharsets.UTF_8) + } } } diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestSpark21101Jar.scala b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestSpark21101Jar.scala index 0263afd8e488a..bd4c4eeddf8ac 100644 --- a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestSpark21101Jar.scala +++ b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestSpark21101Jar.scala @@ -20,7 +20,6 @@ package org.apache.spark.sql.hive.test import java.io.File import java.lang.management.ManagementFactory import java.nio.charset.StandardCharsets -import java.nio.file.{Files, Paths} import org.apache.spark.util.{SparkTestUtils, Utils} @@ -52,6 +51,8 @@ object TestSpark21101Jar { private def readResource(name: String): String = { val url = Thread.currentThread().getContextClassLoader.getResource(name) assert(url != null, s"Resource not found: $name") - new String(Files.readAllBytes(Paths.get(url.toURI)), StandardCharsets.UTF_8) + Utils.tryWithResource(url.openStream()) { is => + new String(is.readAllBytes(), StandardCharsets.UTF_8) + } } } diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestUDTFJar.scala b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestUDTFJar.scala index b3fb0aca19777..a2eee6097ba89 100644 --- a/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestUDTFJar.scala +++ b/sql/hive/src/test/scala/org/apache/spark/sql/hive/test/TestUDTFJar.scala @@ -20,7 +20,6 @@ package org.apache.spark.sql.hive.test import java.io.File import java.lang.management.ManagementFactory import java.nio.charset.StandardCharsets -import java.nio.file.{Files, Paths} import org.apache.spark.TestUtils import org.apache.spark.util.Utils @@ -34,7 +33,9 @@ object TestUDTFJar { private def loadResource(name: String): String = { val url = Thread.currentThread().getContextClassLoader.getResource(name) assert(url != null, s"Resource not found: $name") - new String(Files.readAllBytes(Paths.get(url.toURI)), StandardCharsets.UTF_8) + Utils.tryWithResource(url.openStream()) { is => + new String(is.readAllBytes(), StandardCharsets.UTF_8) + } } private val source = Map(