Update tests/test_api_endpoint_config.py

Co-authored-by: Copilot <[email protected]>

Author: sylwia-budzynska

tests/test_api_endpoint_config.py

@@ -53,17 +53,18 @@ def test_to_url_githubcopilot(self):
 
     def test_unsupported_endpoint(self):
         original_env = os.environ.pop('AI_API_ENDPOINT', None)
-        api_endpoint = 'https://unsupported.example.com'
-        os.environ['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
-        if original_env:
-            os.environ['AI_API_ENDPOINT'] = original_env
+        try:
+            api_endpoint = 'https://unsupported.example.com'
+            os.environ['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
+        finally:
+            if original_env:
+                os.environ['AI_API_ENDPOINT'] = original_env
 if __name__ == '__main__':
     pytest.main([__file__, '-v'])