From d4cb0b0960b1cd69c5af443a378b28fc2d5241e0 Mon Sep 17 00:00:00 2001 From: Kousuke Saruta Date: Tue, 19 May 2026 13:21:47 +0900 Subject: [PATCH] Initialize base session before each test --- .../sql/connect/service/AddArtifactsHandlerSuite.scala | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/service/AddArtifactsHandlerSuite.scala b/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/service/AddArtifactsHandlerSuite.scala index 75d7fdd4d8848..8094a084d3b65 100644 --- a/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/service/AddArtifactsHandlerSuite.scala +++ b/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/service/AddArtifactsHandlerSuite.scala @@ -48,6 +48,12 @@ class AddArtifactsHandlerSuite extends SharedSparkSession with ResourceHelper { private val sessionId = UUID.randomUUID.toString() private val sessionKey = SessionKey("c1", sessionId) + override def beforeEach(): Unit = { + super.beforeEach() + SparkConnectService.sessionManager.invalidateAllSessions() + SparkConnectService.sessionManager.initializeBaseSession(() => spark.newSession()) + } + class DummyStreamObserver(p: Promise[AddArtifactsResponse]) extends StreamObserver[AddArtifactsResponse] { override def onNext(v: AddArtifactsResponse): Unit = p.success(v)