another compatibility issue with tests

Author: pitbulk

tests/src/OneLogin/saml2_tests/auth_test.py

@@ -778,7 +778,6 @@ def testLogoutNoSLO(self):
             exception = context.exception
             self.assertIn("The IdP does not support Single Log Out", str(exception))
 
-
     def testLogoutNameIDandSessionIndex(self):
         """
         Tests the logout method of the OneLogin_Saml2_Auth class

tests/src/OneLogin/saml2_tests/logout_request_test.py

@@ -126,7 +126,6 @@ def testGetNameIdData(self):
             exception = context.exception
             self.assertIn("Key is required in order to decrypt the NameID", str(exception))
 
-
         settings = OneLogin_Saml2_Settings(self.loadSettingsJSON())
         key = settings.get_sp_key()
         name_id_data_4 = OneLogin_Saml2_Logout_Request.get_nameid_data(request_2, key)

tests/src/OneLogin/saml2_tests/logout_response_test.py

@@ -54,8 +54,7 @@ def testConstructor(self):
         settings = OneLogin_Saml2_Settings(self.loadSettingsJSON())
         message = self.file_contents(join(self.data_path, 'logout_responses', 'logout_response_deflated.xml.base64'))
         response = OneLogin_Saml2_Logout_Response(settings, message)
-        self.assertRegex(compat.to_string(OneLogin_Saml2_XML.to_string(response.document)),
-                                 '<samlp:LogoutResponse')
+        self.assertRegex(compat.to_string(OneLogin_Saml2_XML.to_string(response.document)), '<samlp:LogoutResponse')
 
     def testCreateDeflatedSAMLLogoutResponseURLParameter(self):
         """

tests/src/OneLogin/saml2_tests/response_test.py

@@ -93,7 +93,7 @@ def testReturnNameId(self):
             response_5.get_nameid()
             self.assertTrue(False)
         except Exception as e:
-            self.assertIn('Not NameID found in the assertion of the Response', e.message)
+            self.assertIn('Not NameID found in the assertion of the Response', str(e))
 
         json_settings['security']['wantNameId'] = False
         settings = OneLogin_Saml2_Settings(json_settings)
@@ -110,7 +110,7 @@ def testReturnNameId(self):
             response_7.get_nameid()
             self.assertTrue(False)
         except Exception as e:
-            self.assertIn('Not NameID found in the assertion of the Response', e.message)
+            self.assertIn('Not NameID found in the assertion of the Response', str(e))
 
     def testGetNameIdData(self):
         """
@@ -163,7 +163,7 @@ def testGetNameIdData(self):
             response_5.get_nameid_data()
             self.assertTrue(False)
         except Exception as e:
-            self.assertIn('Not NameID found in the assertion of the Response', e.message)
+            self.assertIn('Not NameID found in the assertion of the Response', str(e))
 
         json_settings['security']['wantNameId'] = False
         settings = OneLogin_Saml2_Settings(json_settings)
@@ -180,7 +180,7 @@ def testGetNameIdData(self):
             response_7.get_nameid_data()
             self.assertTrue(False)
         except Exception as e:
-            self.assertIn('Not NameID found in the assertion of the Response', e.message)
+            self.assertIn('Not NameID found in the assertion of the Response', str(e))
 
     def testCheckStatus(self):
         """

tests/src/OneLogin/saml2_tests/xml_utils_test.py

@@ -73,7 +73,6 @@ def testToString(self):
             exception = context.exception
             self.assertIn("unsupported type", str(exception))
 
-
     def testToElement(self):
         """
         Tests the to_etree method of the OneLogin_Saml2_XML