tests/modules/Import/ImportFileSplitterTest.php

   1 <?php
   2 /*********************************************************************************
   3  * SugarCRM Community Edition is a customer relationship management program developed by
   4  * SugarCRM, Inc. Copyright (C) 2004-2011 SugarCRM Inc.
   5  *
   6  * This program is free software; you can redistribute it and/or modify it under
   7  * the terms of the GNU Affero General Public License version 3 as published by the
   8  * Free Software Foundation with the addition of the following permission added
   9  * to Section 15 as permitted in Section 7(a): FOR ANY PART OF THE COVERED WORK
  10  * IN WHICH THE COPYRIGHT IS OWNED BY SUGARCRM, SUGARCRM DISCLAIMS THE WARRANTY
  11  * OF NON INFRINGEMENT OF THIRD PARTY RIGHTS.
  12  *
  13  * This program is distributed in the hope that it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  15  * FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
  16  * details.
  17  *
  18  * You should have received a copy of the GNU Affero General Public License along with
  19  * this program; if not, see http://www.gnu.org/licenses or write to the Free
  20  * Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  21  * 02110-1301 USA.
  22  *
  23  * You can contact SugarCRM, Inc. headquarters at 10050 North Wolfe Road,
  24  * SW2-130, Cupertino, CA 95014, USA. or at email address contact@sugarcrm.com.
  25  *
  26  * The interactive user interfaces in modified source and object code versions
  27  * of this program must display Appropriate Legal Notices, as required under
  28  * Section 5 of the GNU Affero General Public License version 3.
  29  *
  30  * In accordance with Section 7(b) of the GNU Affero General Public License version 3,
  31  * these Appropriate Legal Notices must retain the display of the "Powered by
  32  * SugarCRM" logo. If the display of the logo is not reasonably feasible for
  33  * technical reasons, the Appropriate Legal Notices must display the words
  34  * "Powered by SugarCRM".
  35  ********************************************************************************/
  36
  37
  38 require_once 'modules/Import/ImportFile.php';
  39 require_once 'modules/Import/ImportFileSplitter.php';
  40
  41 class ImportFileSplitterTest extends Sugar_PHPUnit_Framework_TestCase
  42 {
  43     protected $_goodFile;
  44     protected $_badFile;
  45
  46     public function setUp()
  47     {
  48         $GLOBALS['current_user'] = SugarTestUserUtilities::createAnonymousUser();
  49         $this->_goodFile = SugarTestImportUtilities::createFile();
  50                 $this->_badFile  = $GLOBALS['sugar_config']['import_dir'].'thisfileisntthere'.date("YmdHis");
  51                 $this->_whiteSpaceFile  = SugarTestImportUtilities::createFileWithWhiteSpace();
  52     }
  53
  54     public function tearDown()
  55     {
  56         SugarTestImportUtilities::removeAllCreatedFiles();
  57         SugarTestUserUtilities::removeAllCreatedAnonymousUsers();
  58         unset($GLOBALS['current_user']);
  59     }
  60
  61     public function testLoadNonExistantFile()
  62     {
  63         $importFileSplitter = new ImportFileSplitter($this->_badFile);
  64         $this->assertFalse($importFileSplitter->fileExists());
  65     }
  66
  67     public function testLoadGoodFile()
  68     {
  69         $importFileSplitter = new ImportFileSplitter($this->_goodFile);
  70         $this->assertTrue($importFileSplitter->fileExists());
  71     }
  72
  73     public function testSplitSourceFile()
  74     {
  75         $importFileSplitter = new ImportFileSplitter($this->_goodFile);
  76         $importFileSplitter->splitSourceFile(',','"');
  77
  78         $this->assertEquals($importFileSplitter->getRecordCount(),2000);
  79         $this->assertEquals($importFileSplitter->getFileCount(),2);
  80     }
  81
  82     public function testSplitSourceFileNoEnclosure()
  83     {
  84         $importFileSplitter = new ImportFileSplitter($this->_goodFile);
  85         $importFileSplitter->splitSourceFile(',','');
  86
  87         $this->assertEquals($importFileSplitter->getRecordCount(),2000);
  88         $this->assertEquals($importFileSplitter->getFileCount(),2);
  89     }
  90
  91     public function testSplitSourceFileWithHeader()
  92     {
  93         $importFileSplitter = new ImportFileSplitter($this->_goodFile);
  94         $importFileSplitter->splitSourceFile(',','"',true);
  95
  96         $this->assertEquals($importFileSplitter->getRecordCount(),1999);
  97         $this->assertEquals($importFileSplitter->getFileCount(),2);
  98     }
  99
 100     public function testSplitSourceFileWithThreshold()
 101     {
 102         $importFileSplitter = new ImportFileSplitter($this->_goodFile,500);
 103         $importFileSplitter->splitSourceFile(',','"');
 104
 105         $this->assertEquals($importFileSplitter->getRecordCount(),2000);
 106         $this->assertEquals($importFileSplitter->getFileCount(),4);
 107     }
 108
 109     public function testGetSplitFileName()
 110     {
 111         $importFileSplitter = new ImportFileSplitter($this->_goodFile);
 112         $importFileSplitter->splitSourceFile(',','"');
 113
 114         $this->assertEquals($importFileSplitter->getSplitFileName(0),"{$this->_goodFile}-0");
 115         $this->assertEquals($importFileSplitter->getSplitFileName(1),"{$this->_goodFile}-1");
 116         $this->assertEquals($importFileSplitter->getSplitFileName(2),false);
 117     }
 118
 119         /**
 120          * @ticket 25119
 121          */
 122     public function testTrimSpaces()
 123     {
 124         $splitter = new ImportFileSplitter($this->_whiteSpaceFile);
 125         $splitter->splitSourceFile(',',' ',false);
 126
 127         $csvString = file_get_contents("{$this->_whiteSpaceFile}-0");
 128
 129         $this->assertEquals(
 130             trim(file_get_contents("{$this->_whiteSpaceFile}-0")),
 131             trim(file_get_contents("{$this->_whiteSpaceFile}"))
 132             );
 133     }
 134 }