SolidDbManagerTest.php

<?php
/*
 SPDX-FileCopyrightText: © 2014 Siemens AG
 
 SPDX-License-Identifier: GPL-2.0-only
*/
 
namespace Fossology\Lib\Db;
 
class SolidDbManagerTest extends DbManagerTestCase
{
  function setUp() : void
  {
    parent::setUp();
    $this->dbManager = new SolidDbManager($this->logger);
    $this->dbManager->setDriver($this->driver);
  }
 
  function tearDown() : void
  {
    parent::tearDown();
  }
 
  function testInsertTableRow()
  {
    $tableName = 'foo';
    $assocParams = array('cola'=>1,'colb'=>2);
    $sqlLog = 'bar';
    $this->driver->shouldReceive('query')->once();
    $this->driver->shouldReceive('freeResult');
    $this->dbManager->insertTableRow($tableName,$assocParams,$sqlLog);
  }
 
  function testCreateMap()
  {
    $keyColumn = 'yek';
    $valueColumn = 'lav';
    $tableName = 'foo';
    $sqlLog = 'bar';
    $preSql = "/ $keyColumn, *$valueColumn /";
    $this->driver->shouldReceive('query')->andReturn('fakeRes');
    $this->driver->shouldReceive('fetchArray')->andReturn(
            array($keyColumn=>'k0',$valueColumn=>'v0'),
            array($keyColumn=>'k1',$valueColumn=>'v1'),
            false
          );
    $this->driver->shouldReceive('freeResult');
    $map = $this->dbManager->createMap($tableName,$keyColumn,$valueColumn,$sqlLog);
    assertThat($map,hasKey('k0'));
    assertThat($map,EqualTo(array('k0'=>'v0','k1'=>'v1')));
  }
 
  function testEvaluateStatement()
  {
    $this->driver->shouldReceive('query');
    $sqlStmt = 'SELECT pet FROM africa WHERE cervical=$1 AND class=$2 AND $3';
    $this->dbManager->prepare($stmt='statement',$sqlStmt);
 
    $reflection = new \ReflectionClass(get_class($this->dbManager));
    $method = $reflection->getMethod('evaluateStatement');
    $method->setAccessible(true);
 
    $params = array(7,'Mammalia',true);
    $sql = $method->invoke($this->dbManager,$stmt,$params);
    assertThat($sql, is('SELECT pet FROM africa WHERE cervical=7 AND class=\'Mammalia\' AND t') );
 
    $params = array(7,'Mammalia\'; SELECT * FROM passwords WHERE user like \'%',true);
    $sql = $method->invoke($this->dbManager,$stmt,$params);
    assertThat($sql, is('SELECT pet FROM africa WHERE cervical=7 AND class=\'Mammalia\'\'; SELECT * FROM passwords WHERE user like \'\'%\' AND t') );
  }
 
  function testEvaluateStatement_exception()
  {
    $sqlStmt = 'SELECT pet FROM africa WHERE cervical=$1 AND class=$2';
    $this->dbManager->prepare($stmt='statement',$sqlStmt);
 
    $reflection = new \ReflectionClass(get_class($this->dbManager));
    $method = $reflection->getMethod('evaluateStatement');
    $method->setAccessible(true);
 
    $exceptionMsg = false;
    $params = array(7,'Mammalia','non-used parameter');
    try {
      $method->invoke($this->dbManager,$stmt,$params);
    }
    catch(\Exception $e) {
      $exceptionMsg = $e->getMessage();
    }
    assertThat($exceptionMsg, is('$3 not found in prepared statement'));
  }
}