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_get_state_token_from_html_format_c(self):
html = b"\n var stateToken = \'win\\x2Dning\';\n"
ret = OktaSaml.get_state_token_from_html(html)
self.assertEqual(ret, 'win-ning')
def test_get_state_token_from_html_format_a(self):
html = b"\n var stateToken = 'win';\n"
ret = OktaSaml.get_state_token_from_html(html)
self.assertEqual(ret, "win")
