Fix unittest exception checking

Author: asloobq

tests/test_encryption.py

@@ -203,11 +203,8 @@ def test_decrypt_token_2_invalid_lifetime_exception(self):
 
                 with self.assertRaises(EncryptionError) as context:
                     decrypt_token(token, key_collection, "", client_type)
-                    # Assert that the exception was raised
-                    self.assertIsNotNone(context.exception)
 
-                    # Assert additional properties of the exception, if needed
-                    self.assertEqual(str(context.exception), "invalid token lifetime")
+                self.assertEqual(str(context.exception), "invalid token lifetime")
 
     def test_decrypt_token_invalid_lifetime_pass(self):
         seconds_since_established = 3600  # from UID2TokenGenerator.generate_uid2_token_v4

tests/test_publisher_client.py

@@ -3,10 +3,11 @@
 
 from uid2_client import Uid2PublisherClient
 from uid2_client import TokenGenerateInput
+from uid2_client import TokenGenerateResponse
 from uid2_client.identity_tokens import IdentityTokens
 from urllib.request import HTTPError
 
-@unittest.skip
+
 class PublisherEuidIntegrationTests(unittest.TestCase):
 
     EUID_SECRET_KEY = None
@@ -60,8 +61,6 @@ def test_integration_optout_generate_token(self):
         self.assertFalse(token_generate_response.is_success())
         self.assertIsNone(token_generate_response.get_identity())
 
-
-@unittest.skip
 class PublisherUid2IntegrationTests(unittest.TestCase):
 
     UID2_SECRET_KEY = None
@@ -87,7 +86,7 @@ def setUpClass(cls):
     def test_integration_generate_and_refresh(self):
 
         token_generate_response = self.publisher_client.generate_token(
-            TokenGenerateInput.from_email("test+xyz@example.com"))
+            TokenGenerateInput.from_email("test@example.com"))
 
         self.assertFalse(token_generate_response.is_optout())