diff --git a/sql/core/src/test/scala/org/apache/spark/sql/StreamTest.scala b/sql/core/src/test/scala/org/apache/spark/sql/StreamTest.scala index f356cde9cf87ab3ed828ccaaf00b4940e08be781..26c597bf349b3ccc1f6a373e09a1dc9ef1a19954 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/StreamTest.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/StreamTest.scala @@ -28,6 +28,7 @@ import scala.util.control.NonFatal import org.scalatest.Assertions import org.scalatest.concurrent.{Eventually, Timeouts} +import org.scalatest.concurrent.Eventually.timeout import org.scalatest.concurrent.PatienceConfiguration.Timeout import org.scalatest.exceptions.TestFailedDueToTimeoutException import org.scalatest.time.Span @@ -67,7 +68,14 @@ trait StreamTest extends QueryTest with Timeouts { implicit class RichContinuousQuery(cq: ContinuousQuery) { def stopQuietly(): Unit = quietly { - cq.stop() + try { + failAfter(10.seconds) { + cq.stop() + } + } catch { + case e: TestFailedDueToTimeoutException => + logError(e.getMessage(), e) + } } }