Skip to content

Commit 7b6bd59

Browse files
committed
Implement test for x_token_server error bug
1 parent feca584 commit 7b6bd59

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

tests/unit/oauth_test_utils.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,11 +54,14 @@ def __call__(self, request, uri, response_headers):
5454
if authorization and authorization.replace("Bearer ", "") in self.tokens:
5555
return [200, response_headers, json.dumps(self.sample_post_response_data)]
5656
elif self.redirect_server is None and self.token_server is not None:
57-
return [401, {'Www-Authenticate': f'Bearer x_token_server="{self.token_server}"',
58-
'Basic realm': '"Trino"'}, ""]
59-
return [401, {'Www-Authenticate': f'Bearer x_redirect_server="{self.redirect_server}", '
60-
f'x_token_server="{self.token_server}"',
61-
'Basic realm': '"Trino"'}, ""]
57+
return [401,
58+
{'Www-Authenticate': f'Bearer realm="Trino", token_type="JWT", Bearer x_token_server="{self.token_server}"',
59+
'Basic realm': '"Trino"'},
60+
""]
61+
return [401,
62+
{'Www-Authenticate': f'Bearer realm="Trino", token_type="JWT", Bearer x_redirect_server="{self.redirect_server}", x_token_server="{self.token_server}"',
63+
'Basic realm': '"Trino"'},
64+
""]
6265

6366

6467
class GetTokenCallback:

0 commit comments

Comments
 (0)