test: avoid local env impacting unit test (#143)

fangyh20 · web-flow · commit 2a4d911d775d · 2025-09-26T18:18:29.000-07:00
diff --git a/tests/unit/test_session.py b/tests/unit/test_session.py
@@ -2229,6 +2229,7 @@ def test_builder_pattern_system_label_protection(
             )
             self.stopSession(mock_session_controller_client_instance, session)
 
+    @mock.patch.dict(os.environ, {}, clear=True)
     @mock.patch("google.auth.default")
     @mock.patch("google.cloud.dataproc_v1.SessionControllerClient")
     @mock.patch("pyspark.sql.connect.client.SparkConnectClient.config")

Original file line number	Diff line number	Diff line change
`@@ -2229,6 +2229,7 @@ def test_builder_pattern_system_label_protection(`
`2229`	`2229`	`)`
`2230`	`2230`	`self.stopSession(mock_session_controller_client_instance, session)`
`2231`	`2231`
	`2232`	`+ @mock.patch.dict(os.environ, {}, clear=True)`
`2232`	`2233`	`@mock.patch("google.auth.default")`
`2233`	`2234`	`@mock.patch("google.cloud.dataproc_v1.SessionControllerClient")`
`2234`	`2235`	`@mock.patch("pyspark.sql.connect.client.SparkConnectClient.config")`