@@ -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
0 commit comments