[py] Use mock in test_updates_path_after_setting_env_variable

Author: cgoldberg

py/test/selenium/webdriver/safari/safari_service_tests.py

@@ -17,6 +17,8 @@
 
 import os
 
+from unittest.mock import patch
+
 import pytest
 
 from selenium.webdriver.safari.service import Service
@@ -41,11 +43,9 @@ def test_uses_path_from_env_variable(self, service):
         assert "safaridriver" in service.path
 
     def test_updates_path_after_setting_env_variable(self, service):
-        new_path = "/foo/bar"
-        os.environ["SE_SAFARIDRIVER"] = new_path
         service.executable_path = self.service_path  # Simulating the update
-
-        assert "safaridriver" in service.executable_path
+        with patch.dict("os.environ", {"SE_SAFARIDRIVER": "/foo/bar"}):
+            assert "safaridriver" in service.executable_path
 
 
 def test_enable_logging():