LicenseCsvExportTest.php

<?php
/*
Copyright (C) 2015, Siemens AG

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
version 2 as published by the Free Software Foundation.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

namespace Fossology\Lib\Application;

use Fossology\Lib\BusinessRules\LicenseMap;
use Fossology\Lib\Db\DbManager;
use Fossology\Lib\Test\TestPgDb;
use Mockery as M;

class LicenseCsvExportTest extends \PHPUnit\Framework\TestCase
{
  protected function setUp()
  {
    $this->assertCountBefore = \Hamcrest\MatcherAssert::getCount();
  }

  protected function tearDown()
  {
    $this->addToAssertionCount(\Hamcrest\MatcherAssert::getCount()-$this->assertCountBefore);
    M::close();
  }

  public function testCreateCsv()
  {
    $testDb = new TestPgDb("licenseCsvExport");
    $testDb->createPlainTables(array('license_ref','license_map','groups'));
    $testDb->createInheritedTables(array('license_candidate'));
    $dbManager = $testDb->getDbManager();
    $licenses = array();
    $candLicenses = array();
    $dbManager->insertTableRow('groups', array(
      'group_pk' => 2, 'group_name' => 'test'
    ));
    for ($i = 1; $i < 4; $i ++) {
      $licenses[$i] = array(
        'rf_pk' => $i,
        'rf_shortname' => 'lic' . $i,
        'rf_fullname' => 'lice' . $i,
        'rf_text' => 'text' . $i,
        'rf_url' => $i . $i,
        'rf_notes' => 'note' . $i,
        'rf_source' => 's' . $i,
        'rf_detector_type' => 1,
        'rf_risk' => ($i - 1)
      );
      $dbManager->insertTableRow('license_ref', $licenses[$i]);
    }
    for ($i = 1; $i <= 4; $i ++) {
      $candLicenses[$i] = array(
        'rf_pk' => $i + 4,
        'rf_shortname' => 'candlic' . $i,
        'rf_fullname' => 'candlice' . $i,
        'rf_text' => 'text' . $i,
        'rf_url' => $i . $i,
        'rf_notes' => 'note' . $i,
        'rf_source' => 's' . $i,
        'rf_detector_type' => 1,
        'rf_risk' => ($i - 1),
        'marydone' => false,
        'group_fk' => 2
      );
      if ($i % 2 == 0) {
        $candLicenses[$i]['marydone'] = true;
      }
      $dbManager->insertTableRow('license_candidate', $candLicenses[$i]);
    }

    $dbManager->insertTableRow('license_map', array('rf_fk'=>3,'rf_parent'=>1,'usage'=>LicenseMap::CONCLUSION));
    $dbManager->insertTableRow('license_map', array('rf_fk'=>3,'rf_parent'=>2,'usage'=>LicenseMap::REPORT));

    $licenseCsvExport = new LicenseCsvExport($dbManager);
    $head = array('shortname','fullname','text','parent_shortname','report_shortname','url','notes','source','risk','group');
    $out = fopen('php://output', 'w');

    $csv = $licenseCsvExport->createCsv();
    ob_start();
    fputcsv($out, $head);
    fputcsv($out, array($licenses[1]['rf_shortname'],
        $licenses[1]['rf_fullname'],
        $licenses[1]['rf_text'],
        null,
        null,
        $licenses[1]['rf_url'],
        $licenses[1]['rf_notes'],
        $licenses[1]['rf_source'],
        $licenses[1]['rf_risk'],
        null));

    fputcsv($out, array($licenses[2]['rf_shortname'],
        $licenses[2]['rf_fullname'],
        $licenses[2]['rf_text'],
        null,
        null,
        $licenses[2]['rf_url'],
        $licenses[2]['rf_notes'],
        $licenses[2]['rf_source'],
        $licenses[2]['rf_risk'],
        null));

    fputcsv($out, array($licenses[3]['rf_shortname'],
        $licenses[3]['rf_fullname'],
        $licenses[3]['rf_text'],
        $licenses[1]['rf_shortname'],
        $licenses[2]['rf_shortname'],
        $licenses[3]['rf_url'],
        $licenses[3]['rf_notes'],
        $licenses[3]['rf_source'],
        $licenses[3]['rf_risk'],
        null));

    fputcsv($out, array($candLicenses[2]['rf_shortname'],
      $candLicenses[2]['rf_fullname'],
      $candLicenses[2]['rf_text'],
      null,
      null,
      $candLicenses[2]['rf_url'],
      $candLicenses[2]['rf_notes'],
      $candLicenses[2]['rf_source'],
      $candLicenses[2]['rf_risk'],
      "test"));

    fputcsv($out, array($candLicenses[4]['rf_shortname'],
      $candLicenses[4]['rf_fullname'],
      $candLicenses[4]['rf_text'],
      null,
      null,
      $candLicenses[4]['rf_url'],
      $candLicenses[4]['rf_notes'],
      $candLicenses[4]['rf_source'],
      $candLicenses[4]['rf_risk'],
      "test"));
    $expected = ob_get_contents();
    ob_end_clean();

    assertThat($csv, is(equalTo($expected)));

    $delimiter = '|';
    $licenseCsvExport->setDelimiter($delimiter);
    $csv3 = $licenseCsvExport->createCsv(3);
    ob_start();
    fputcsv($out, $head, $delimiter);
    fputcsv($out, array($licenses[3]['rf_shortname'],
          $licenses[3]['rf_fullname'],
          $licenses[3]['rf_text'],
          $licenses[1]['rf_shortname'],
          $licenses[2]['rf_shortname'],
          $licenses[3]['rf_url'],
          $licenses[3]['rf_notes'],
          $licenses[3]['rf_source'],
          $licenses[3]['rf_risk'],
          null
        ),
        $delimiter);
    $expected3 = ob_get_contents();
    ob_end_clean();
    assertThat($csv3, is(equalTo($expected3)));
  }

  public function testSetDelimiter()
  {
    $dbManager = M::mock(DbManager::class);
    $licenseCsvExport = new LicenseCsvExport($dbManager);
    $reflection = new \ReflectionClass($licenseCsvExport);
    $delimiter = $reflection->getProperty('delimiter');
    $delimiter->setAccessible(true);

    $licenseCsvExport->setDelimiter('|');
    assertThat($delimiter->getValue($licenseCsvExport),is('|'));

    $licenseCsvExport->setDelimiter('<>');
    assertThat($delimiter->getValue($licenseCsvExport),is('<'));
  }

  public function testSetEnclosure()
  {
    $dbManager = M::mock(DbManager::class);
    $licenseCsvExport = new LicenseCsvExport($dbManager);
    $reflection = new \ReflectionClass($licenseCsvExport);
    $enclosure = $reflection->getProperty('enclosure');
    $enclosure->setAccessible(true);

    $licenseCsvExport->setEnclosure('|');
    assertThat($enclosure->getValue($licenseCsvExport),is('|'));

    $licenseCsvExport->setEnclosure('<>');
    assertThat($enclosure->getValue($licenseCsvExport),is('<'));
  }
}