diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/lasso.py | 2 | ||||
| -rw-r--r-- | python/tests/login_tests.py | 7 |
2 files changed, 5 insertions, 4 deletions
diff --git a/python/lasso.py b/python/lasso.py index 135e3a0a..df9e57ce 100644 --- a/python/lasso.py +++ b/python/lasso.py @@ -874,7 +874,7 @@ class User: def get_assertion(self, remote_providerID): return Node(lassomod.user_get_assertion(self, remote_providerID)) - def get_authentication_method(self, remote_providerID): + def get_authentication_method(self, remote_providerID = None): return lassomod.user_get_authentication_method(self, remote_providerID) def get_next_assertion_remote_providerID(self): diff --git a/python/tests/login_tests.py b/python/tests/login_tests.py index b97652f8..061fede5 100644 --- a/python/tests/login_tests.py +++ b/python/tests/login_tests.py @@ -139,10 +139,11 @@ class LoginTestCase(unittest.TestCase): self.failUnlessEqual(spLoginContext.nameIdentifier, nameIdentifier) # The user doesn't have any federation yet. self.failUnlessEqual(spLoginContext.create_user(None), 0) - spUserContextDump = spLoginContext.user.dump() + spUserContext = spLoginContext.user + self.failUnless(spUserContext) + spUserContextDump = spUserContext.dump() self.failUnless(spUserContextDump) - authenticationMethod = spLoginContext.response.get_child( - "AuthenticationStatement").get_attr_value("AuthenticationMethod") + authenticationMethod = spUserContext.get_authentication_method() self.failUnlessEqual(authenticationMethod, lasso.samlAuthenticationMethodPassword) # Service provider logout. |