Skip to content

Commit 1da8307

Browse files
1 parent 697a0e0 commit 1da8307

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

‎google/auth/_helpers.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,15 @@
2828
# expiry.
2929
REFRESH_THRESHOLD = datetime.timedelta(minutes=3, seconds=45)
3030

31-
_SENSITIVE_FIELDS = {"accessToken", "access_token", "id_token", "client_id", "refresh_token", "client_secret"}
31+
_SENSITIVE_FIELDS = {
32+
"accessToken",
33+
"access_token",
34+
"id_token",
35+
"client_id",
36+
"refresh_token",
37+
"client_secret",
38+
}
39+
3240

3341
def copy_docstring(source_class):
3442
"""Decorator that copies a method's docstring from another class.
@@ -302,4 +310,4 @@ def _hash_value(value, field_name: str) -> str:
302310
hash_object = hashlib.sha512()
303311
hash_object.update(encoded_value)
304312
hex_digest = hash_object.hexdigest()
305-
return f"hashed_{field_name}-{hex_digest}"
313+
return f"hashed_{field_name}-{hex_digest}"

‎tests/test__helpers.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,7 @@ def test_unpadded_urlsafe_b64encode():
195195
for case, expected in cases:
196196
assert _helpers.unpadded_urlsafe_b64encode(case) == expected
197197

198+
198199
def test_hash_sensitive_info_basic():
199200
test_data = {
200201
"expires_in": 3599,
@@ -208,6 +209,7 @@ def test_hash_sensitive_info_basic():
208209
assert hashed_data["access_token"].startswith("hashed_access_token-")
209210
assert hashed_data["token_type"] == "Bearer"
210211

212+
211213
def test_hash_sensitive_info_multiple_sensitive():
212214
test_data = {
213215
"access_token": "some_long_token",
@@ -235,18 +237,21 @@ def test_hash_sensitive_info_non_string_value():
235237
assert hashed_data["access_token"].startswith("hashed_access_token-")
236238
assert hashed_data["normal_data"] == "def"
237239

240+
238241
def test_hash_sensitive_info_empty_dict():
239242
test_data = {}
240243
hashed_data = _helpers.hash_sensitive_info(test_data)
241244
assert hashed_data == {}
242245

246+
243247
def test_hash_value_consistent_hashing():
244248
value = "test_value"
245249
field_name = "test_field"
246250
hash1 = _helpers._hash_value(value, field_name)
247251
hash2 = _helpers._hash_value(value, field_name)
248252
assert hash1 == hash2
249253

254+
250255
def test_hash_value_different_hashing():
251256
value1 = "test_value1"
252257
value2 = "test_value2"
@@ -255,5 +260,6 @@ def test_hash_value_different_hashing():
255260
hash2 = _helpers._hash_value(value2, field_name)
256261
assert hash1 != hash2
257262

263+
258264
def test_hash_value_none():
259265
assert _helpers._hash_value(None, "test") is None

0 commit comments

Comments
 (0)