Use monkeypatch for env var

Author: sylwia-budzynska

tests/test_api_endpoint_config.py

@@ -53,21 +53,16 @@ def test_to_url_githubcopilot(self):
         endpoint = AI_API_ENDPOINT_ENUM.AI_API_GITHUBCOPILOT
         assert endpoint.to_url() == 'https://api.githubcopilot.com'
 
-    def test_unsupported_endpoint(self):
+    def test_unsupported_endpoint(self, monkeypatch):
         """Test that unsupported API endpoint raises ValueError."""
-        try:
-            original_env = os.environ.pop('AI_API_ENDPOINT', None)
-            api_endpoint = 'https://unsupported.example.com'
-            os.environ['AI_API_ENDPOINT'] = api_endpoint
+        api_endpoint = 'https://unsupported.example.com'
+        monkeypatch.setenv('AI_API_ENDPOINT', api_endpoint)
+        with pytest.raises(ValueError) as excinfo:
+            list_capi_models("abc")
+        msg = str(excinfo.value)
+        assert 'Unsupported Model Endpoint' in msg
+        assert 'https://models.github.ai/inference' in msg
+        assert 'https://api.githubcopilot.com' in msg
 
-            with pytest.raises(ValueError) as excinfo:
-                list_capi_models("abc")
-            msg = str(excinfo.value)
-            assert 'Unsupported Model Endpoint' in msg
-            assert 'https://models.github.ai/inference' in msg
-            assert 'https://api.githubcopilot.com' in msg
-        finally:
-            if original_env:
-                os.environ['AI_API_ENDPOINT'] = original_env
 if __name__ == '__main__':
     pytest.main([__file__, '-v'])