diff --git a/src/Parser/Csv.php b/src/Parser/Csv.php index 451f698..89065b9 100644 --- a/src/Parser/Csv.php +++ b/src/Parser/Csv.php @@ -81,6 +81,15 @@ public function getRecord() return $record; } + public function getRecords() : array + { + $records = []; + while ($record = $this->getRecord()) { + $records[] = $record; + } + return $records; + } + public function reset() { $this->field = ""; diff --git a/test/FileTest.php b/test/FileTest.php index deec857..81c39fa 100644 --- a/test/FileTest.php +++ b/test/FileTest.php @@ -10,10 +10,7 @@ public function test() $parser->feed(file_get_contents(__DIR__ . "/data/countries.csv")); $parser->finish(); - $records = []; - while ($record = $parser->getRecord()) { - $records[] = $record; - } + $records = $parser->getRecords(); $this->assertEquals(5, count($records)); }