Esempio n. 1
0
    def test_get_assertion_missing(self):
        okta_saml = OktaSaml('org', 'user', 'password')
        okta_saml.assertion = mock.MagicMock()
        okta_saml.assertion.return_value = b''
        okta_saml.session = mock.MagicMock()
        okta_saml.session.get.return_value = MockResponse(None, None, 'assert')
        okta_saml.get_okta_error_from_response = mock.MagicMock()

        with self.assertRaises(UnknownError):
            okta_saml.get_assertion('foo')

        okta_saml.get_okta_error_from_response.assert_has_calls([mock.ANY])
Esempio n. 2
0
    def test_get_assertion_missing(self):
        okta_saml = OktaSaml("org", "user", "password")
        okta_saml.assertion = mock.MagicMock()
        okta_saml.assertion.return_value = b""
        okta_saml.session = mock.MagicMock()
        okta_saml.session.get.return_value = MockResponse(None, None, "assert")
        okta_saml.get_okta_error_from_response = mock.MagicMock()

        with self.assertRaises(UnknownError):
            okta_saml.get_assertion("foo")

        okta_saml.get_okta_error_from_response.assert_has_calls([mock.ANY])
Esempio n. 3
0
    def test_get_okta_error_from_response_specific(self):
        okta_saml = OktaSaml('org', 'user', 'password')
        response = MockResponse('', 200, AWS_HTML_ERROR)
        ret = okta_saml.get_okta_error_from_response(response)

        self.assertEqual(ret, 'BAD STUFF')
Esempio n. 4
0
    def test_get_okta_error_from_response(self):
        okta_saml = OktaSaml('org', 'user', 'password')
        response = MockResponse('', 200, 'html')
        ret = okta_saml.get_okta_error_from_response(response)

        self.assertEqual(ret, 'Unknown error')
Esempio n. 5
0
    def test_get_okta_error_from_response_specific(self):
        okta_saml = OktaSaml("org", "user", "password")
        response = MockResponse("", 200, AWS_HTML_ERROR)
        ret = okta_saml.get_okta_error_from_response(response)

        self.assertEqual(ret, "BAD STUFF")
Esempio n. 6
0
    def test_get_okta_error_from_response(self):
        okta_saml = OktaSaml("org", "user", "password")
        response = MockResponse("", 200, "html")
        ret = okta_saml.get_okta_error_from_response(response)

        self.assertEqual(ret, "Unknown error")