def test_get_assertion_500(self): okta_saml = OktaSaml('org', 'user', 'password') okta_saml.assertion = mock.MagicMock() okta_saml.session = mock.MagicMock() resp = requests.Response() resp.status_code = 500 resp.url = '' okta_saml.session.get.return_value = resp with self.assertRaises(UnknownError): okta_saml.get_assertion('foo')
def test_get_assertion_successful(self): okta_saml = OktaSaml('org', 'user', 'password') okta_saml.assertion = mock.MagicMock() okta_saml.assertion.return_value = 'assertion' okta_saml.session = mock.MagicMock() okta_saml.session.get.return_value = MockResponse(None, None, 'assert') ret = okta_saml.get_assertion('foo') okta_saml.assertion.assert_has_calls([mock.call('assert')]) self.assertEqual(ret, 'assertion')
def test_get_assertion_successful(self): okta_saml = OktaSaml("org", "user", "password") okta_saml.assertion = mock.MagicMock() okta_saml.assertion.return_value = "assertion" okta_saml.session = mock.MagicMock() okta_saml.session.get.return_value = MockResponse(None, None, "assert") ret = okta_saml.get_assertion("foo") okta_saml.assertion.assert_has_calls([mock.call("assert")]) self.assertEqual(ret, "assertion")
def test_get_assertion_500(self): okta_saml = OktaSaml("org", "user", "password") okta_saml.assertion = mock.MagicMock() okta_saml.session = mock.MagicMock() resp = requests.Response() resp.status_code = 500 resp.url = "" okta_saml.session.get.return_value = resp with self.assertRaises(UnknownError): okta_saml.get_assertion("foo")
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])
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])
def test_get_assertion_2FA_no_token(self): okta_saml = OktaSaml('org', 'user', 'password') okta_saml.assertion = mock.MagicMock() okta_saml.session = mock.MagicMock() resp = MockResponse(None, None, 'assert') resp.url = 'https://foo.bar/second-factor?zort=yep' okta_saml.session.get.return_value = resp okta_saml.get_state_token_from_html = mock.MagicMock() okta_saml.get_state_token_from_html.side_effect = AttributeError with self.assertRaises(ReauthNeeded) as err: okta_saml.get_assertion('foo') exc = err.exception self.assertEqual(exc.state_token, None)
def test_get_assertion_2FA(self): okta_saml = OktaSaml("org", "user", "password") okta_saml.assertion = mock.MagicMock() okta_saml.session = mock.MagicMock() resp = MockResponse(None, None, "assert") resp.url = "https://foo.bar/second-factor?zort=yep" okta_saml.session.get.return_value = resp okta_saml.get_state_token_from_html = mock.MagicMock() okta_saml.get_state_token_from_html.return_value = "token" with self.assertRaises(ReauthNeeded) as err: okta_saml.get_assertion("foo") exc = err.exception self.assertEqual(exc.state_token, "token")
def test_assertion(self): okta_saml = OktaSaml('org', 'user', 'password') ret = okta_saml.assertion(EXAMPLE_ASSERTION) self.assertEqual(ret, b"H\x03\x0bH\x03\x0bH\x03\x0b")
def test_assertion(self): okta_saml = OktaSaml("org", "user", "password") ret = okta_saml.assertion(EXAMPLE_ASSERTION) self.assertEqual(ret, b"H\x03\x0bH\x03\x0bH\x03\x0b")