def test_create_authorization_headers(self, mock_presto_config): auth_headers = PrestoClient._create_auth_headers( "Aladdin", "open sesame") expected_auth_headers = { "Authorization": "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ==" } self.assertEqual(auth_headers, expected_auth_headers)
def test_create_authorization_headers_fails_with_colon_in_user( self, mock_error, mock_presto_config): PrestoClient._create_auth_headers("Aladdin:1", "open sesame") error_message = "LDAP user cannot contain ':': Aladdin:1" mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers_fails_with_empty_password( self, mock_error, mock_presto_config): PrestoClient._create_auth_headers("Aladdin", "") error_message = 'LDAP password (taken from internal-communication.authentication.ldap.password in ' \ '/etc/presto/config.properties on the coordinator) cannot be null or empty' mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers_fails_with_null_user( self, mock_error, mock_presto_config): PrestoClient._create_auth_headers(None, "open sesame") error_message = 'LDAP user (taken from internal-communication.authentication.ldap.user in ' \ '/etc/presto/config.properties on the coordinator) cannot be null or empty' mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers_fails_with_colon_in_user(self, mock_error, mock_presto_config): PrestoClient._create_auth_headers("Aladdin:1", "open sesame") error_message = "LDAP user cannot contain ':': Aladdin:1" mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers_fails_with_empty_password(self, mock_error, mock_presto_config): PrestoClient._create_auth_headers("Aladdin", "") error_message = 'LDAP password (taken from internal-communication.authentication.ldap.password in ' \ '/etc/presto/config.properties on the coordinator) cannot be null or empty' mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers_fails_with_null_user(self, mock_error, mock_presto_config): PrestoClient._create_auth_headers(None, "open sesame") error_message = 'LDAP user (taken from internal-communication.authentication.ldap.user in ' \ '/etc/presto/config.properties on the coordinator) cannot be null or empty' mock_error.assert_called_once_with(error_message)
def test_create_authorization_headers(self, mock_presto_config): auth_headers = PrestoClient._create_auth_headers("Aladdin", "open sesame") expected_auth_headers = {"Authorization": "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=="} self.assertEqual(auth_headers, expected_auth_headers)