Skip to content

Commit a871524

Browse files
committed
Deprecate encryptData and decryptData methods
1 parent e7334a3 commit a871524

2 files changed

Lines changed: 24 additions & 1 deletion

File tree

tests/test_bidstream_client.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,28 @@ def test_token_generated_in_the_future_within_allowed_clock_skew(self, mock_refr
7979
self.assertEqual(result.identity_scope, expected_scope)
8080
self.assertEqual(result.advertising_token_version, expected_version)
8181

82+
def test_empty_keys(self, mock_refresh_bidstream_keys): # EmptyKeyContainer
83+
token = generate_uid_token(IdentityScope.UID2, AdvertisingTokenVersion.ADVERTISING_TOKEN_V3)
84+
mock_refresh_bidstream_keys.return_value = None
85+
self._client.refresh_keys()
86+
with self.assertRaises(EncryptionError):
87+
self._client.decrypt_ad_token_into_raw_uid(token, None)
88+
89+
def test_master_key_expired(self, mock_refresh_keys_util): #ExpiredKeyContainer
90+
def get_post_refresh_keys_response_with_key_expired():
91+
master_key_expired = EncryptionKey(master_key_id, site_id, created=now, activates=YESTERDAY, expires=YESTERDAY, secret=master_secret,
92+
keyset_id=99999)
93+
expired_key = EncryptionKey(site_key_id, site_id, created=now, activates=YESTERDAY, expires=YESTERDAY, secret=site_secret,
94+
keyset_id=99999)
95+
return create_default_key_collection([master_key, expired_key])
96+
97+
mock_refresh_keys_util.return_value = get_post_refresh_keys_response_with_key_expired()
98+
self._client.refresh_keys()
99+
100+
with self.assertRaises(EncryptionError) as context:
101+
self._client.decrypt_ad_token_into_raw_uid(example_uid, None)
102+
self.assertTrue('No Keyset Key Found' in context.exception)
103+
82104
def test_refresh_keys(self, mock_refresh_bidstream_keys):
83105
key_collection = create_default_key_collection([master_key])
84106
mock_refresh_bidstream_keys.return_value = key_collection

uid2_client/encryption.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,6 +300,7 @@ def encrypt(uid2, identity_scope, keys, keyset_id=None, **kwargs):
300300
return _encrypt_token(uid2, identity_scope, master_key, key, site_id, now, token_expiry, ad_token_version)
301301

302302

303+
# DEPRECATED, DO NOT CALL
303304
def encrypt_data(data, identity_scope, **kwargs):
304305
"""Encrypt arbitrary binary data.
305306
@@ -386,7 +387,7 @@ def encrypt_data(data, identity_scope, **kwargs):
386387
def _encrypt_data_v1(data, key, iv):
387388
return int.to_bytes(key.key_id, 4, 'big') + iv + _encrypt(data, iv, key)
388389

389-
390+
# DEPRECATED, DO NOT CALL
390391
def decrypt_data(encrypted_data, keys):
391392
"""Decrypt data encrypted with encrypt_data().
392393

0 commit comments

Comments
 (0)