mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-10-22 14:05:37 +02:00
80 lines
3.2 KiB
PHP
80 lines
3.2 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
class CSVParserTest extends SapphireTest {
|
||
|
|
function testParsingWithHeaders() {
|
||
|
|
/* By default, a CSV file will be interpreted as having headers */
|
||
|
|
$csv = new CSVParser('sapphire/tests/dev/CSVBulkLoaderTest_PlayersWithHeader.csv');
|
||
|
|
|
||
|
|
$firstNames = $birthdays = $biographies = array();
|
||
|
|
foreach($csv as $record) {
|
||
|
|
/* Each row in the CSV file will be keyed with the header row */
|
||
|
|
$this->assertEquals(array('FirstName','Biography','Birthday'), array_keys($record));
|
||
|
|
$firstNames[] = $record['FirstName'];
|
||
|
|
$biographies[] = $record['Biography'];
|
||
|
|
$birthdays[] = $record['Birthday'];
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->assertEquals(array('John','Jane','Jamie','Järg'), $firstNames);
|
||
|
|
$this->assertEquals(array(
|
||
|
|
"He's a good guy",
|
||
|
|
"She is awesome.\nSo awesome that she gets multiple rows and \"escaped\" strings in her biography",
|
||
|
|
"Pretty old, with an escaped comma",
|
||
|
|
"Unicode FTW"), $biographies);
|
||
|
|
$this->assertEquals(array("31/01/1988","31/01/1982","31/01/1882","31/06/1982"), $birthdays);
|
||
|
|
}
|
||
|
|
|
||
|
|
function testParsingWithHeadersAndColumnMap() {
|
||
|
|
/* By default, a CSV file will be interpreted as having headers */
|
||
|
|
$csv = new CSVParser('sapphire/tests/dev/CSVBulkLoaderTest_PlayersWithHeader.csv');
|
||
|
|
|
||
|
|
/* We can set up column remapping. The keys are case-insensitive. */
|
||
|
|
$csv->mapColumns(array(
|
||
|
|
'FirstName' => '__fn',
|
||
|
|
'bIoGrApHy' => '__BG',
|
||
|
|
));
|
||
|
|
|
||
|
|
$firstNames = $birthdays = $biographies = array();
|
||
|
|
foreach($csv as $record) {
|
||
|
|
/* Each row in the CSV file will be keyed with the renamed columns. Any unmapped column names will be left as-is. */
|
||
|
|
$this->assertEquals(array('__fn','__BG','Birthday'), array_keys($record));
|
||
|
|
$firstNames[] = $record['__fn'];
|
||
|
|
$biographies[] = $record['__BG'];
|
||
|
|
$birthdays[] = $record['Birthday'];
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->assertEquals(array('John','Jane','Jamie','Järg'), $firstNames);
|
||
|
|
$this->assertEquals(array(
|
||
|
|
"He's a good guy",
|
||
|
|
"She is awesome.\nSo awesome that she gets multiple rows and \"escaped\" strings in her biography",
|
||
|
|
"Pretty old, with an escaped comma",
|
||
|
|
"Unicode FTW"), $biographies);
|
||
|
|
$this->assertEquals(array("31/01/1988","31/01/1982","31/01/1882","31/06/1982"), $birthdays);
|
||
|
|
}
|
||
|
|
|
||
|
|
function testParsingWithExplicitHeaderRow() {
|
||
|
|
/* If your CSV file doesn't have a header row */
|
||
|
|
$csv = new CSVParser('sapphire/tests/dev/CSVBulkLoaderTest_PlayersWithHeader.csv');
|
||
|
|
|
||
|
|
$csv->provideHeaderRow(array('__fn','__bio','__bd'));
|
||
|
|
|
||
|
|
$firstNames = $birthdays = $biographies = array();
|
||
|
|
foreach($csv as $record) {
|
||
|
|
/* Each row in the CSV file will be keyed with the header row that you gave */
|
||
|
|
$this->assertEquals(array('__fn','__bio','__bd'), array_keys($record));
|
||
|
|
$firstNames[] = $record['__fn'];
|
||
|
|
$biographies[] = $record['__bio'];
|
||
|
|
$birthdays[] = $record['__bd'];
|
||
|
|
}
|
||
|
|
|
||
|
|
/* And the first row will be returned in the data */
|
||
|
|
$this->assertEquals(array('FirstName','John','Jane','Jamie','Järg'), $firstNames);
|
||
|
|
$this->assertEquals(array(
|
||
|
|
'Biography',
|
||
|
|
"He's a good guy",
|
||
|
|
"She is awesome.\nSo awesome that she gets multiple rows and \"escaped\" strings in her biography",
|
||
|
|
"Pretty old, with an escaped comma",
|
||
|
|
"Unicode FTW"), $biographies);
|
||
|
|
$this->assertEquals(array("Birthday","31/01/1988","31/01/1982","31/01/1882","31/06/1982"), $birthdays);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|