public function testGetNoOfRecords() { $csvSplitter = new CSVSplitter(); try { $r = $csvSplitter->getNoOfRecords(); $this->fail('An exception should be thrown when no. of rows is not calculated'); } catch (Exception $e) { $this->assertEquals(CSVSplitterException::ROWS_NOT_CALCULATED_YET, $e->getCode()); } $this->assertTrue($csvSplitter->split($this->testFilePath)); $noOfRows = $csvSplitter->getNoOfRecords(); $this->assertEquals(2024, $noOfRows); }
/** * Do the data import * * @return array Array containing results of import */ public function handleUpload() { CSVSplitter::setRecordLimit(50); $csvSplitter = new CSVSplitter(); $csvSplitter->setTempDir(ROOT_PATH . '/lib/confs/temp'); $success = $csvSplitter->split($this->fileName); if ($success) { $noOfRecords = $csvSplitter->getNoOfRecords(); $tempFileList = $csvSplitter->getTempFileList(); $result = new CSVSplitStatus('success', $this->importType, $noOfRecords, $tempFileList); } else { $result = new CSVSplitStatus('failure', $this->importType); } return $result; }