Fix test cases

MonkeyCanCode · MonkeyCanCode · commit e86515a6a7be · 2025-11-15T13:03:53.000-06:00
diff --git a/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/SparkConnectTestUtils.scala b/sql/connect/server/src/test/scala/org/apache/spark/sql/connect/SparkConnectTestUtils.scala
@@ -31,6 +31,9 @@ object SparkConnectTestUtils {
         sessionId = UUID.randomUUID().toString,
         session = session)
     SparkConnectService.sessionManager.putSessionForTesting(ret)
+    if (session != null) {
+      ret.initializeSession()
+    }
     ret
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,9 @@ object SparkConnectTestUtils {`
`31`	`31`	`sessionId = UUID.randomUUID().toString,`
`32`	`32`	`session = session)`
`33`	`33`	`SparkConnectService.sessionManager.putSessionForTesting(ret)`
	`34`	`+ if (session != null) {`
	`35`	`+ ret.initializeSession()`
	`36`	`+ }`
`34`	`37`	`ret`
`35`	`38`	`}`
`36`	`39`	`}`