From 5308607eb89a7b71575728f29bd4025671be5989 Mon Sep 17 00:00:00 2001
From: Dongjoon Hyun <dongjoon@apache.org>
Date: Mon, 12 May 2025 21:29:04 -0700
Subject: [PATCH] [SPARK-52094][FOLLOWUP] Fix the test case to respect
 SPARK_REMOTE env variable

---
 Tests/SparkConnectTests/SparkSessionTests.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Tests/SparkConnectTests/SparkSessionTests.swift b/Tests/SparkConnectTests/SparkSessionTests.swift
index 5124c72..e8213cc 100644
--- a/Tests/SparkConnectTests/SparkSessionTests.swift
+++ b/Tests/SparkConnectTests/SparkSessionTests.swift
@@ -54,8 +54,8 @@ struct SparkSessionTests {
   func sessionID() async throws {
     let spark1 = try await SparkSession.builder.getOrCreate()
     await spark1.stop()
-    let remote = "sc://localhost/;session_id=\(spark1.sessionID)"
-    let spark2 = try await SparkSession.builder.remote(remote).getOrCreate()
+    let remote = ProcessInfo.processInfo.environment["SPARK_REMOTE"] ?? "sc://localhost"
+    let spark2 = try await SparkSession.builder.remote("\(remote)/;session_id=\(spark1.sessionID)").getOrCreate()
     await spark2.stop()
     #expect(spark1.sessionID == spark2.sessionID)
     #expect(spark1 == spark2)