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])