TestAbstractDb.php

<?php
/*
Copyright (C) 2015,2019 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\Test;

// setup autoloading
require_once(dirname(dirname(dirname(dirname(__FILE__)))) . "/vendor/autoload.php");

use Fossology\Lib\Db\DbManager;

abstract class TestAbstractDb
{
  protected $dbManager;

  protected function dirnameRec($path, $depth = 1)
  {
    for ($i = 0; $i < $depth; $i ++) {
      $path = dirname($path);
    }
    return $path;
  }

  abstract function createPlainTables($tableList, $invert=false);

  public function insertData($tableList, $invert=FALSE, $dataFile=null)
  {
    $testdataFile = $dataFile ?: dirname(__FILE__) . '/testdata.sql';
    $testdata = file_get_contents($testdataFile);
    $delimiter = 'INSERT INTO ';
    $offset = strpos($testdata, $delimiter);
    while (false !== $offset) {
      $nextOffset = strpos($testdata, $delimiter, $offset + 1);
      if (false === $nextOffset) {
        $sql = substr($testdata, $offset);
      } else {
        $sql = substr($testdata, $offset, $nextOffset - $offset);
      }
      $table = array();
      preg_match('/^INSERT INTO (?P<name>\w+) /', $sql, $table);
      if (($invert ^ ! in_array($table['name'], $tableList))) {
        $offset = $nextOffset;
        continue;
      }
      $this->dbManager->queryOnce($this->queryConverter($sql));
      $offset = $nextOffset;
    }
  }

  protected function queryConverter($sql)
  {
    return $sql;
  }

  public function insertData_license_ref($limit=140)
  {
    $keysToBeChanged = array(
      'rf_OSIapproved' => '"rf_OSIapproved"',
      'rf_FSFfree'=> '"rf_FSFfree"',
      'rf_GPLv2compatible' => '"rf_GPLv2compatible"',
      'rf_GPLv3compatible'=> '"rf_GPLv3compatible"',
      'rf_Fedora' => '"rf_Fedora"'
      );

    $LIBEXECDIR = $this->dirnameRec(__FILE__, 5) . '/install/db';
    $jsonData = json_decode(file_get_contents("$LIBEXECDIR/licenseRef.json"), true);
    $statementName = __METHOD__.'.insertInToLicenseRef';
    foreach ($jsonData as $licenseArrayKey => $licenseArray) {
      $keys = strtr(implode(",", array_keys($licenseArray)), $keysToBeChanged);
      $valuePlaceHolders = "$" . join(",$",range(1, count(array_keys($licenseArray))));
      $SQL = "INSERT INTO license_ref ( $keys ) VALUES ($valuePlaceHolders);";
      $this->dbManager->prepare($statementName, $SQL);
      $this->dbManager->execute($statementName, array_values($licenseArray));
      if ($licenseArrayKey >= $limit) {
        break;
      }
    }
  }

  protected function applySchema($type, $elementList, $invert=false)
  {
    $coreSchemaFile = $this->dirnameRec(__FILE__, 4) . '/www/ui/core-schema.dat';
    $Schema = array();
    require($coreSchemaFile);
    foreach ($Schema[$type] as $viewName => $sql) {
      if ($invert ^ ! in_array($viewName, $elementList)) {
        continue;
      }
      $sqlCreate = is_array($sql) ? $sql['CREATE'] : $sql;
      $this->dbManager->queryOnce($sqlCreate);
    }
  }

  public function createViews($viewList, $invert=FALSE)
  {
    $this->applySchema('VIEW', $viewList, $invert);
  }

  public function alterTables($tableList, $invert=FALSE)
  {
    $coreSchemaFile = $this->dirnameRec(__FILE__, 4) . '/www/ui/core-schema.dat';
    $Schema = array();
    require($coreSchemaFile);
    $attributeKey = "ALTER";
    foreach ($Schema['TABLE'] as $tableName => $tableCols) {
      if ($invert ^ ! in_array($tableName, $tableList)) {
        continue;
      }
      foreach ($tableCols as $attributes) {
        if (array_key_exists($attributeKey, $attributes)) {
          $this->dbManager->queryOnce($attributes[$attributeKey]);
        }
      }
    }
  }

  public function &getDbManager()
  {
    return $this->dbManager;
  }
}