update send_saml_response to use http_client

tokendito/okta.py

@@ -216,26 +216,47 @@ def send_saml_request(saml_request, cookies, user_agent=f"Tokendito/{__version__
 
 
 
-def send_saml_response(saml_response):
-    """Submit SAML response to the SP.
+def send_saml_response(saml_response, user_agent = f"Tokendito/{__version__}"):
+    """
+    Submit SAML response to the SP.
 
-    :param saml_response: dict with with SP post_url, relay_state, and saml_response
+    :param saml_response: dict with SP post_url, relay_state, and saml_response
     :returns: `sid` session cookie
     """
+
+    # Create an HTTP client instance with the version-specific user agent.
+    http_client = HTTPClient(user_agent)
+
+    # Define the payload and headers for the request.
     payload = {
         "SAMLResponse": saml_response["response"],
         "RelayState": saml_response["relay_state"],
     }
     headers = {"accept": "text/html,application/xhtml+xml,application/xml"}
+
+    # Construct the URL from the provided saml_response.
     url = saml_response["post_url"]
 
+    # Log the SAML response details.
     logger.debug(f"Sending SAML response back to {url}")
-    response = user.request_wrapper("POST", url, data=payload, headers=headers)
+
+    # Use the HTTP client to make a POST request.
+    response = http_client.post(url, data=payload, headers=headers)
+
+    # Extract cookies from the response.
     session_cookies = response.cookies
+
+    # Get the 'sid' value from the cookies.
     sid = session_cookies.get("sid")
+
+    # If 'sid' is present, mask its value for logging purposes.
     if sid is not None:
         user.add_sensitive_value_to_be_masked(sid)
+
+    # Log the session cookies.
     logger.debug(f"Have session cookies: {session_cookies}")
+
+    # Return the session cookies.
     return session_cookies