def test_package_name_and_logging_provided( self, mock_logging, mock_run_node, mock_configure_project, is_async, conf_logging, mocker, ): mocker.patch("multiprocessing.get_start_method", return_value="spawn") node_ = mocker.sentinel.node catalog = mocker.sentinel.catalog run_id = "fake_run_id" package_name = mocker.sentinel.package_name _run_node_synchronization( node_, catalog, is_async, run_id, package_name=package_name, conf_logging=conf_logging, ) mock_run_node.assert_called_once_with(node_, catalog, is_async, run_id) mock_logging.assert_called_once_with(conf_logging) mock_configure_project.assert_called_once_with(package_name)
def test_package_name_provided( self, mock_logging, mock_run_node, mock_register_hooks, mock_configure_project, is_async, mocker, ): mocker.patch("multiprocessing.get_start_method", return_value="spawn") node = mocker.sentinel.node catalog = mocker.sentinel.catalog run_id = "fake_run_id" package_name = mocker.sentinel.package_name hook_manager = get_hook_manager() _run_node_synchronization(node, catalog, is_async, run_id, package_name=package_name) mock_run_node.assert_called_once_with(node, catalog, is_async, run_id) mock_register_hooks.assert_called_once_with(hook_manager) mock_logging.assert_called_once_with({}) mock_configure_project.assert_called_once_with(package_name)
def test_package_name_not_provided(self, mock_logging, mock_run_node, is_async, mocker): mocker.patch("multiprocessing.get_start_method", return_value="fork") node_ = mocker.sentinel.node catalog = mocker.sentinel.catalog run_id = "fake_run_id" package_name = mocker.sentinel.package_name _run_node_synchronization(node_, catalog, is_async, run_id, package_name=package_name) mock_run_node.assert_called_once_with(node_, catalog, is_async, run_id) mock_logging.assert_not_called()