public function testNoCsvHeaders()
 {
     $button = new GridFieldExportButton();
     $button->setExportColumns(array('Name' => 'Name', 'City' => 'City'));
     $button->setCsvHasHeader(false);
     $this->assertEquals("\"Test\",\"City\"\n\"Test2\",\"City2\"\n", $button->generateExportFileData($this->gridField));
 }
 public function testArrayListInput()
 {
     $button = new GridFieldExportButton();
     $this->gridField->getConfig()->addComponent(new GridFieldPaginator());
     //Create an ArrayList 1 greater the Paginator's default 15 rows
     $arrayList = new ArrayList();
     for ($i = 1; $i <= 16; $i++) {
         $dataobject = new DataObject(array('ID' => $i));
         $arrayList->add($dataobject);
     }
     $this->gridField->setList($arrayList);
     $this->assertEquals("\"ID\"\n\"1\"\n\"2\"\n\"3\"\n\"4\"\n\"5\"\n\"6\"\n\"7\"\n\"8\"\n" . "\"9\"\n\"10\"\n\"11\"\n\"12\"\n\"13\"\n\"14\"\n\"15\"\n\"16\"\n", $button->generateExportFileData($this->gridField));
 }
 public function testZeroValue()
 {
     $button = new GridFieldExportButton();
     $button->setExportColumns(array('RugbyTeamNumber' => 'Rugby Team Number'));
     $this->assertEquals("\"Rugby Team Number\"\n\"2\"\n\"0\"\n", $button->generateExportFileData($this->gridField));
 }