def test_user_agent_set(self): config = self.config okta = OktaAPIAuth(**config) okta.pool = MagicMock() class Urlopen_Mock: data = '{}' okta.pool.urlopen.return_value = Urlopen_Mock() user_agent = 'user-agent' # http://www.ietf.org/rfc/rfc2616.txt # OktaOpenVPN/1.0.0 (Darwin 13.4.0) CPython/2.7.1 okta.preauth() args = okta.pool.urlopen.call_args_list headers = args[0][1]['headers'] self.assertIn(user_agent, headers) actual = headers[user_agent] import platform system = platform.uname()[0] system_version = platform.uname()[2] python_version = "{}/{}".format( platform.python_implementation(), platform.python_version(), ) for expected in [ 'OktaOpenVPN/', system, system_version, python_version ]: self.assertIn(expected, actual)
def test_user_agent_set(self): config = self.config okta = OktaAPIAuth(**config) okta.pool = MagicMock() class Urlopen_Mock: data = '{}' okta.pool.urlopen.return_value = Urlopen_Mock() user_agent = 'user-agent' # http://www.ietf.org/rfc/rfc2616.txt # OktaOpenVPN/1.0.0 (Darwin 13.4.0) CPython/2.7.1 okta.preauth() args = okta.pool.urlopen.call_args_list headers = args[0][1]['headers'] self.assertIn(user_agent, headers) actual = headers[user_agent] import platform system = platform.uname()[0] system_version = platform.uname()[2] python_version = "{}/{}".format( platform.python_implementation(), platform.python_version(), ) for expected in ['OktaOpenVPN/', system, system_version, python_version]: self.assertIn(expected, actual)
def test_connect_to_example_with_good_pin(self): config = self.config config['assert_pinset'] = [self.herokuapp_dot_com_pin] okta = OktaAPIAuth(**config) result = okta.preauth() self.assertIn('status', result) self.assertEquals(result['status'], 'MFA_REQUIRED')
def test_connect_to_okta_with_good_pins(self): config = self.config config['okta_url'] = 'https://example.okta.com' okta = OktaAPIAuth(**config) result = okta.preauth() # This is what we'll get since we're sending an invalid token: self.assertIn('errorSummary', result) self.assertEquals(result['errorSummary'], 'Invalid token provided')