def open_output_file_if_needed(self): if self.config.output_mode == Config.OUTPUT_MODE_FILE: item_writer_factory = ItemWriterFactory() self.item_writer = item_writer_factory.create_item_writer_based_on_file_extension( self.config.output_file_name) self.item_writer.open_output_file(self.config.output_file_name)
def setUp(self): filepath_splitter = mock.create_autospec(FilePathSplitter).return_value object_writer_factory = mock.create_autospec( ObjectWriterFactory).return_value self.item_writer_factory = ItemWriterFactory(filepath_splitter, object_writer_factory)