From 9e1855b5b1b1688233d65c642827a7c48c3c717b Mon Sep 17 00:00:00 2001 From: Eric Barry Date: Sat, 27 Jul 2019 16:58:38 -0400 Subject: [PATCH] #35 Adding in tests to cover bugs --- tests/unit/test_saml.py | 67 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 67 insertions(+) diff --git a/tests/unit/test_saml.py b/tests/unit/test_saml.py index db2e218..89a63b6 100644 --- a/tests/unit/test_saml.py +++ b/tests/unit/test_saml.py @@ -258,6 +258,73 @@ def tests_uses_default_form_values(self, generic_auth, generic_config, } ) + def test_input_missing_name_attribute(self, generic_auth, + generic_config, + mock_requests_session): + saml_form = ( + '' + '
' + '' + '' + '' + '' + '
' + '' + ) + mock_requests_session.get.return_value = mock.Mock( + spec=requests.Response, status_code=200, text=saml_form + ) + mock_requests_session.post.return_value = mock.Mock( + spec=requests.Response, status_code=200, text=( + '
' + ) + ) + saml_assertion = generic_auth.retrieve_saml_assertion(generic_config) + assert saml_assertion == 'fakeassertion' + + mock_requests_session.post.assert_called_with( + "https://example.com/path/login/", verify=True, + data={ + 'username': 'monty', + 'password': 'mypassword', + 'spam': 'eggs' + } + ) + + def test_boolean_presence_attribute(self, generic_auth, + generic_config, + mock_requests_session): + saml_form = ( + '' + '
' + '' + '' + '' + '
' + '' + ) + mock_requests_session.get.return_value = mock.Mock( + spec=requests.Response, status_code=200, text=saml_form + ) + mock_requests_session.post.return_value = mock.Mock( + spec=requests.Response, status_code=200, text=( + '
' + ) + ) + saml_assertion = generic_auth.retrieve_saml_assertion(generic_config) + assert saml_assertion == 'fakeassertion' + + mock_requests_session.post.assert_called_with( + "https://example.com/path/login/", verify=True, + data={ + 'username': 'monty', + 'password': 'mypassword', + 'spam': 'eggs' + } + ) + def test_error_getting_form(self, generic_auth, mock_requests_session, generic_config): mock_requests_session.get.return_value = mock.Mock(