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'));
     }
 }