added base url to DescopeClient

Author: gaokevin1

descope/descope_client.py

@@ -27,6 +27,7 @@ class DescopeClient:
     def __init__(
         self,
         project_id: str,
+        base_url: Optional[str] = None,
         public_key: Optional[dict] = None,
         skip_verify: bool = False,
         management_key: Optional[str] = None,
@@ -50,6 +51,7 @@ def __init__(
         # Auth Initialization
         auth_http_client = HTTPClient(
             project_id=project_id,
+            base_url=base_url,
             timeout_seconds=timeout_seconds,
             secure=not skip_verify,
             management_key=auth_management_key

tests/test_descope_client.py

@@ -1052,6 +1052,21 @@ def test_auth_management_key_with_refresh_token(self):
                 timeout=DEFAULT_TIMEOUT_SECONDS,
             )
 
+    def test_base_url_setting(self):
+        """Test that base_url parameter is correctly set in DescopeClient"""
+        custom_base_url = "https://api.use1.descope.com"
+        client = DescopeClient(
+            project_id=self.dummy_project_id,
+            base_url=custom_base_url,
+            public_key=self.public_key_dict,
+        )
+
+        # Verify that the base_url is set in the auth HTTP client
+        self.assertEqual(client._auth.http_client.base_url, custom_base_url)
+
+        # Verify that the base_url is set in the mgmt HTTP client
+        self.assertEqual(client._mgmt._http.base_url, custom_base_url)
+
 
 if __name__ == "__main__":
     unittest.main()