mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-07-05 19:09:27 +02:00
c8af0fd7d1
Fixes http://open.silverstripe.org/ticket/6473 When using CSVParser::$columnMapping to map columns to a callback action, it previously used the action name as the key value. This prevented users from defining multiple entries to the same callback. This patch retains those key values and simply runs the callback field name filter later on.
97 lines
3.8 KiB
PHP
97 lines
3.8 KiB
PHP
<?php
|
|
|
|
/**
|
|
* @package framework
|
|
* @package tests
|
|
*/
|
|
class CSVParserTest extends SapphireTest {
|
|
|
|
|
|
public function testParsingWithHeaders() {
|
|
/* By default, a CSV file will be interpreted as having headers */
|
|
$csv = new CSVParser($this->getCurrentRelativePath() . '/CsvBulkLoaderTest_PlayersWithHeader.csv');
|
|
|
|
$firstNames = $birthdays = $biographies = $registered = array();
|
|
foreach($csv as $record) {
|
|
/* Each row in the CSV file will be keyed with the header row */
|
|
$this->assertEquals(array('FirstName','Biography','Birthday','IsRegistered'), array_keys($record));
|
|
$firstNames[] = $record['FirstName'];
|
|
$biographies[] = $record['Biography'];
|
|
$birthdays[] = $record['Birthday'];
|
|
$registered[] = $record['IsRegistered'];
|
|
}
|
|
|
|
$this->assertEquals(array('John','Jane','Jamie','Järg'), $firstNames);
|
|
|
|
$this->assertEquals(array(
|
|
"He's a good guy",
|
|
"She is awesome." . PHP_EOL
|
|
. "So 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);
|
|
$this->assertEquals(array('1', '0', '1', '1'), $registered);
|
|
}
|
|
|
|
public function testParsingWithHeadersAndColumnMap() {
|
|
/* By default, a CSV file will be interpreted as having headers */
|
|
$csv = new CSVParser($this->getCurrentRelativePath() . '/CsvBulkLoaderTest_PlayersWithHeader.csv');
|
|
|
|
/* We can set up column remapping. The keys are case-insensitive. */
|
|
$csv->mapColumns(array(
|
|
'FirstName' => '__fn',
|
|
'bIoGrApHy' => '__BG',
|
|
));
|
|
|
|
$firstNames = $birthdays = $biographies = $registered = 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','IsRegistered'), array_keys($record));
|
|
$firstNames[] = $record['__fn'];
|
|
$biographies[] = $record['__BG'];
|
|
$birthdays[] = $record['Birthday'];
|
|
$registered[] = $record['IsRegistered'];
|
|
}
|
|
|
|
$this->assertEquals(array('John','Jane','Jamie','Järg'), $firstNames);
|
|
$this->assertEquals(array(
|
|
"He's a good guy",
|
|
"She is awesome."
|
|
. PHP_EOL . "So 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);
|
|
$this->assertEquals(array('1', '0', '1', '1'), $registered);
|
|
}
|
|
|
|
public function testParsingWithExplicitHeaderRow() {
|
|
/* If your CSV file doesn't have a header row */
|
|
$csv = new CSVParser($this->getCurrentRelativePath() .'/CsvBulkLoaderTest_PlayersWithHeader.csv');
|
|
|
|
$csv->provideHeaderRow(array('__fn','__bio','__bd','__reg'));
|
|
|
|
$firstNames = $birthdays = $biographies = $registered = 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','__reg'), array_keys($record));
|
|
$firstNames[] = $record['__fn'];
|
|
$biographies[] = $record['__bio'];
|
|
$birthdays[] = $record['__bd'];
|
|
$registered[] = $record['__reg'];
|
|
}
|
|
|
|
/* 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." . PHP_EOL
|
|
. "So 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);
|
|
$this->assertEquals(array('IsRegistered', '1', '0', '1', '1'), $registered);
|
|
}
|
|
}
|