diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py index b33b72f5..a9225cd8 100644 --- a/tests/unit/test_client.py +++ b/tests/unit/test_client.py @@ -135,7 +135,7 @@ def assert_headers(headers): assert headers[constants.HEADER_SESSION] == "" assert headers[constants.HEADER_TRANSACTION] is None assert headers[constants.HEADER_TIMEZONE] == timezone - assert headers[constants.HEADER_CLIENT_CAPABILITIES] == "PARAMETRIC_DATETIME" + assert headers[constants.HEADER_CLIENT_CAPABILITIES] == "PARAMETRIC_DATETIME,SESSION_AUTHORIZATION" assert headers[accept_encoding_header] == accept_encoding_value assert headers[client_info_header] == client_info_value assert headers[constants.HEADER_ROLE] == ( @@ -1173,7 +1173,7 @@ def json(self): sql = 'execute my_stament using 1, 2, 3' additional_headers = { constants.HEADER_PREPARED_STATEMENT: 'my_statement=added_prepare_statement_header', - constants.HEADER_CLIENT_CAPABILITIES: 'PARAMETRIC_DATETIME' + constants.HEADER_CLIENT_CAPABILITIES: 'PARAMETRIC_DATETIME,SESSION_AUTHORIZATION' } # Patch the post function to avoid making the requests, as well as to diff --git a/trino/client.py b/trino/client.py index 730a6f87..47724d2f 100644 --- a/trino/client.py +++ b/trino/client.py @@ -522,7 +522,7 @@ def http_headers(self) -> CaseInsensitiveDict[str]: headers[constants.HEADER_ENCODING] = self._client_session.encoding else: raise ValueError("Invalid type for encoding: expected str or list") - headers[constants.HEADER_CLIENT_CAPABILITIES] = 'PARAMETRIC_DATETIME' + headers[constants.HEADER_CLIENT_CAPABILITIES] = 'PARAMETRIC_DATETIME,SESSION_AUTHORIZATION' headers["user-agent"] = f"{constants.CLIENT_NAME}/{__version__}" if len(self._client_session.roles.values()): headers[constants.HEADER_ROLE] = ",".join(