tests/phpunit/CRM/Import/DataSource/CsvTest.php

   1 <?php
   2 /*
   3 +--------------------------------------------------------------------+
   4 | CiviCRM version 4.7                                                |
   5 +--------------------------------------------------------------------+
   6 | Copyright CiviCRM LLC (c) 2004-2017                                |
   7 +--------------------------------------------------------------------+
   8 | This file is a part of CiviCRM.                                    |
   9 |                                                                    |
  10 | CiviCRM is free software; you can copy, modify, and distribute it  |
  11 | under the terms of the GNU Affero General Public License           |
  12 | Version 3, 19 November 2007 and the CiviCRM Licensing Exception.   |
  13 |                                                                    |
  14 | CiviCRM is distributed in the hope that it will be useful, but     |
  15 | WITHOUT ANY WARRANTY; without even the implied warranty of         |
  16 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.               |
  17 | See the GNU Affero General Public License for more details.        |
  18 |                                                                    |
  19 | You should have received a copy of the GNU Affero General Public   |
  20 | License and the CiviCRM Licensing Exception along                  |
  21 | with this program; if not, contact CiviCRM LLC                     |
  22 | at info[AT]civicrm[DOT]org. If you have questions about the        |
  23 | GNU Affero General Public License or the licensing of CiviCRM,     |
  24 | see the CiviCRM license FAQ at http://civicrm.org/licensing        |
  25 +--------------------------------------------------------------------+
  26  */
  27
  28 /**
  29  * Tests for the CRM_Import_Datasource_Csv class.
  30  */
  31 class CRM_Import_Datasource_CsvTest extends CiviUnitTestCase {
  32
  33   /**
  34    * Test the to csv function.
  35    *
  36    * @param string $fileName
  37    *
  38    * @dataProvider getCsvFiles
  39    */
  40   public function testToCsv($fileName) {
  41     $dataSource = new CRM_Import_DataSource_Csv();
  42     $params = array(
  43       'uploadFile' => array(
  44         'name' => __DIR__ . '/' . $fileName,
  45       ),
  46       'skipColumnHeader' => TRUE,
  47     );
  48
  49     // Get the PEAR::DB object
  50     $dao = new CRM_Core_DAO();
  51     $db = $dao->getDatabaseConnection();
  52
  53     $form = new CRM_Contact_Import_Form_DataSource();
  54     $form->controller = new CRM_Contact_Import_Controller();
  55
  56     $dataSource->postProcess($params, $db, $form);
  57     $tableName = $form->get('importTableName');
  58     $this->assertEquals(4,
  59       CRM_Core_DAO::singleValueQuery("SELECT LENGTH(last_name) FROM $tableName"),
  60       $fileName . ' failed on last_name'
  61     );
  62     $this->assertEquals(21,
  63       CRM_Core_DAO::singleValueQuery("SELECT LENGTH(email) FROM $tableName"),
  64       $fileName . ' failed on email'
  65     );
  66     CRM_Core_DAO::executeQuery("DROP TABLE $tableName");
  67   }
  68
  69   /**
  70    * Get csv files to test.
  71    *
  72    * @return array
  73    */
  74   public function getCsvFiles() {
  75     return array(array('import.csv'), array('yogi.csv'));
  76   }
  77
  78 }