Fix previous commited tests

SAML-Toolkits · Sep 14, 2017 · 70d70fc · 70d70fc
1 parent 30cbe7c
commit 70d70fc
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tests/src/OneLogin/saml2_tests/response_test.py b/tests/src/OneLogin/saml2_tests/response_test.py
@@ -139,8 +139,9 @@ def testReturnNameId(self):
         json_settings['security']['wantNameId'] = False
         settings = OneLogin_Saml2_Settings(json_settings)
 
-        nameid_9 = response_9.get_nameid()
-        self.assertEqual(None, nameid_9)
+        response_10 = OneLogin_Saml2_Response(settings, xml_6)
+        nameid_10 = response_10.get_nameid()
+        self.assertEqual(None, nameid_10)
 
     def testReturnNameIdFormat(self):
         """
@@ -204,8 +205,9 @@ def testReturnNameIdFormat(self):
         json_settings['security']['wantNameId'] = False
         settings = OneLogin_Saml2_Settings(json_settings)
 
-        nameid_9 = response_9.get_nameid_format()
-        self.assertEqual('urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress', nameid_9)
+        response_10 = OneLogin_Saml2_Response(settings, xml_6)
+        nameid_10 = response_10.get_nameid_format()
+        self.assertEqual('urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress', nameid_10)
 
     def testGetNameIdData(self):
         """
@@ -287,13 +289,13 @@ def testGetNameIdData(self):
         json_settings['security']['wantNameId'] = False
         settings = OneLogin_Saml2_Settings(json_settings)
 
-        nameid_data_9 = response_9.get_nameid_data()
-
+        response_10 = OneLogin_Saml2_Response(settings, xml_6)
+        nameid_data_10 = response_10.get_nameid_data()
         expected_nameid_data_4 = {
             'Value': None,
             'Format': 'urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress',
         }
-        self.assertEqual(expected_nameid_data_4, nameid_data_9)
+        self.assertEqual(expected_nameid_data_4, nameid_data_10)
 
     def testCheckStatus(self):
         """