function it_flushes_a_buffer_into_multiple_files_with_extension($columnSorter, FlatItemBuffer $buffer) { $columnSorter->sort(Argument::any())->willReturn(['colA', 'colB']); $buffer->rewind()->willReturn(); $buffer->count()->willReturn(3); $buffer->valid()->willReturn(true, true, true, false); $buffer->next()->willReturn(); $buffer->current()->willReturn(['colA' => 'fooA', 'colB' => 'fooB']); $buffer->key()->willReturn(0); $buffer->getHeaders()->willReturn(['colA', 'colB']); $this->flush($buffer, ['type' => 'csv'], $this->directory . 'output.txt', 2); if (!file_exists($this->directory . 'output_1.txt')) { throw new FailedPredictionException(sprintf('File "%s" should have been flushed', $this->directory . 'output_1.txt')); } if (!file_exists($this->directory . 'output_2.txt')) { throw new FailedPredictionException(sprintf('File "%s" should have been flushed', $this->directory . 'output_2.txt')); } }
function it_flushes_a_buffer_into_multiple_files_with_extension($columnSorter, FlatItemBuffer $buffer, StepExecution $stepExecution, JobParameters $parameters) { $columnSorter->sort(Argument::any(), [])->willReturn(['colA', 'colB']); $stepExecution->getJobParameters()->willReturn($parameters); $stepExecution->incrementSummaryInfo('write')->shouldBeCalled(); $parameters->all()->willReturn([]); $buffer->rewind()->willReturn(); $buffer->count()->willReturn(3); $buffer->valid()->willReturn(true, true, true, false); $buffer->next()->willReturn(); $buffer->current()->willReturn(['colA' => 'fooA', 'colB' => 'fooB']); $buffer->key()->willReturn(0); $buffer->getHeaders()->willReturn(['colA', 'colB']); $this->flush($buffer, ['type' => 'csv'], $this->directory . 'output.txt', 2); if (!file_exists($this->directory . 'output_1.txt')) { throw new FailedPredictionException(sprintf('File "%s" should have been flushed', $this->directory . 'output_1.txt')); } if (!file_exists($this->directory . 'output_2.txt')) { throw new FailedPredictionException(sprintf('File "%s" should have been flushed', $this->directory . 'output_2.txt')); } }