Fix multiple challenges handling in OAuth2

Author: lukasz-walkiewicz

tests/unit/test_client.py

@@ -421,6 +421,8 @@ def test_oauth2_authentication_missing_headers(header, error):
     'Bearer x_redirect_server="{redirect_server}", x_token_server="{token_server}", additional_challenge',
     'Bearer x_redirect_server="{redirect_server}", x_token_server="{token_server}", additional_challenge="value"',
     'Bearer x_token_server="{token_server}", x_redirect_server="{redirect_server}"',
+    'Basic realm="Trino", Bearer x_redirect_server="{redirect_server}", x_token_server="{token_server}"',
+    'Bearer x_redirect_server="{redirect_server}", x_token_server="{token_server}", Basic realm="Trino"',
 ])
 @httprettified
 def test_oauth2_header_parsing(header, sample_post_response_data):

trino/auth.py

@@ -190,7 +190,7 @@ def _authenticate(self, response, **kwargs):
         if not auth_info:
             raise exceptions.TrinoAuthError("Error: header WWW-Authenticate not available in the response.")
 
-        if not _OAuth2TokenBearer._BEARER_PREFIX.match(auth_info):
+        if not _OAuth2TokenBearer._BEARER_PREFIX.search(auth_info):
             raise exceptions.TrinoAuthError(f"Error: header info didn't match {auth_info}")
 
         auth_info_headers = parse_dict_header(_OAuth2TokenBearer._BEARER_PREFIX.sub("", auth_info, count=1))