def test_warning_if_connector_dir_empty(self, isdir_mock, listdir_mock): isdir_mock.return_value = True listdir_mock.return_value = [] connector.add() self.assertEqual('\nWarning: Directory %s is empty. No connectors will' ' be deployed\n\n' % constants.CONNECTORS_DIR, self.test_stderr.getvalue())
def update_configs(): configure_cmds.deploy() add_tpch_connector() try: connector.add() except ConfigFileNotFoundError: _LOGGER.info('No connector directory found, not adding connectors.')
def test_add_all(self, mock_validate, listdir_mock, isdir_mock, deploy_mock): catalogs = ['tpch.properties', 'another.properties'] listdir_mock.return_value = catalogs connector.add() deploy_mock.assert_called_with(catalogs, constants.CONNECTORS_DIR, constants.REMOTE_CATALOG_DIR)
def test_add_exists(self, isfile_mock, deploy_mock, validate_mock): isfile_mock.return_value = True connector.add('tpch') filenames = ['tpch.properties'] deploy_mock.assert_called_with(filenames, constants.CONNECTORS_DIR, constants.REMOTE_CATALOG_DIR) validate_mock.assert_called_with(filenames)
def test_add_permission_denied(self, isdir_mock, listdir_mock): isdir_mock.return_value = True error_msg = ('Permission denied') listdir_mock.side_effect = OSError(13, error_msg) fabric.api.env.host = 'localhost' connector.add() self.assertEqual('\nWarning: [localhost] %s\n\n' % error_msg, self.test_stderr.getvalue())
def test_add_all(self, mock_validate, listdir_mock, isdir_mock, deploy_mock): catalogs = ['tpch.properties', 'another.properties'] listdir_mock.return_value = catalogs connector.add() deploy_mock.assert_called_with(catalogs, get_connectors_directory(), constants.REMOTE_CATALOG_DIR, PRESTO_STANDALONE_USER_GROUP)
def test_add_exists(self, isfile_mock, deploy_mock, validate_mock): isfile_mock.return_value = True connector.add('tpch') filenames = ['tpch.properties'] deploy_mock.assert_called_with(filenames, get_connectors_directory(), constants.REMOTE_CATALOG_DIR, PRESTO_STANDALONE_USER_GROUP) validate_mock.assert_called_with(filenames)