def test_handle_code_unknown(self):
# Setup
response_doc = auth_utils.generate_failure_response('foo')
e = exceptions.PermissionsException(response_doc)
# Test
code = self.handler.handle_permission(e)
# Verify
self.assertEqual(code, exceptions.CODE_PERMISSIONS_EXCEPTION)
self.assertEqual(self.handler._handle_unknown.call_count, 1)
def test_handle_code_username(self):
# Setup
response_doc = auth_utils.generate_failure_response(
auth_utils.CODE_USER_PASS)
e = exceptions.PermissionsException(response_doc)
# Test
code = self.handler.handle_permission(e)
# Verify
self.assertEqual(code, exceptions.CODE_PERMISSIONS_EXCEPTION)
self.assertEqual(self.handler._handle_invalid_username.call_count, 1)
def test_handle_code_invalid_ssl_cert(self):
# Setup
response_doc = auth_utils.generate_failure_response(
auth_utils.CODE_INVALID_SSL_CERT)
e = exceptions.PermissionsException(response_doc)
# Test
code = self.handler.handle_permission(e)
# Verify
self.assertEqual(code, exceptions.CODE_PERMISSIONS_EXCEPTION)
self.assertEqual(self.handler._handle_authentication_failed.call_count,
1)
def test_permission(self):
"""
Tests a client-side error when the connection is rejected due to auth reasons.
"""
# Test
response_body = {'auth_error_code': 'authentication_failed'}
e = exceptions.PermissionsException(response_body)
e.error_message = "I've made a huge mistake."
code = self.handler.handle_permission(e)
# Verify
self.assertEqual(code, exceptions.CODE_PERMISSIONS_EXCEPTION)
self.assertTrue(
"I've made a huge mistake.\n" == self.recorder.lines[0])
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
def test_permission(self):
"""
Tests a client-side error when the connection is rejected due to auth reasons.
"""
# Test
response_body = auth_utils.generate_failure_response(
auth_utils.CODE_FAILED)
e = exceptions.PermissionsException(response_body)
code = self.handler.handle_permission(e)
# Verify
self.assertEqual(code, exceptions.CODE_PERMISSIONS_EXCEPTION)
self.assertTrue('Authentication' in self.recorder.lines[0])
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.assertTrue('A valid' in self.recorder.lines[2]) # skip blank line
self.assertEqual(TAG_PARAGRAPH, self.prompt.get_write_tags()[1])
