/** * @dataProvider provider_readerSetFetchColumns */ public function testReaderSetFetchColumns($message, array $columns, array $primaryKeys, array $fetchColumns) { $db = $this->mockDb(); $db->expects($this->once())->method('select')->with('some_table', $columns)->will($this->returnValue(new ArrayIterator(array()))); $reader = new BatchRowIterator($db, 'some_table', $primaryKeys, 22); $reader->setFetchColumns($fetchColumns); // triggers first database select $reader->rewind(); }