2014-04-19 16:56:16 +04:00
|
|
|
<?php
|
|
|
|
|
|
|
|
/**
|
|
|
|
* ownCloud - App Framework
|
|
|
|
*
|
|
|
|
* @author Bernhard Posselt
|
|
|
|
* @copyright 2012 Bernhard Posselt dev@bernhard-posselt.com
|
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 3 of the License, or any later version.
|
|
|
|
*
|
|
|
|
* This library 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 AFFERO GENERAL PUBLIC LICENSE for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Affero General Public
|
|
|
|
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
2014-07-17 13:21:25 +04:00
|
|
|
namespace Test\AppFramework\Db;
|
2014-04-19 16:56:16 +04:00
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Simple utility class for testing mappers
|
|
|
|
*/
|
2014-11-11 01:30:38 +03:00
|
|
|
abstract class MapperTestUtility extends \Test\TestCase {
|
2014-04-19 16:56:16 +04:00
|
|
|
protected $db;
|
|
|
|
private $query;
|
|
|
|
private $queryAt;
|
|
|
|
private $prepareAt;
|
|
|
|
private $fetchAt;
|
|
|
|
private $iterators;
|
2015-01-29 21:16:28 +03:00
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Run this function before the actual test to either set or initialize the
|
|
|
|
* db. After this the db can be accessed by using $this->db
|
|
|
|
*/
|
|
|
|
protected function setUp(){
|
2014-11-11 01:30:38 +03:00
|
|
|
parent::setUp();
|
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
$this->db = $this->getMockBuilder(
|
2015-02-24 18:47:38 +03:00
|
|
|
'\OCP\IDBConnection')
|
2014-04-19 16:56:16 +04:00
|
|
|
->disableOriginalConstructor()
|
|
|
|
->getMock();
|
|
|
|
|
2015-02-24 18:47:38 +03:00
|
|
|
$this->query = $this->getMock('\PDOStatement');
|
2014-04-19 16:56:16 +04:00
|
|
|
$this->queryAt = 0;
|
|
|
|
$this->prepareAt = 0;
|
2015-02-24 18:47:38 +03:00
|
|
|
$this->iterators = [];
|
2014-04-19 16:56:16 +04:00
|
|
|
$this->fetchAt = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Create mocks and set expected results for database queries
|
|
|
|
* @param string $sql the sql query that you expect to receive
|
|
|
|
* @param array $arguments the expected arguments for the prepare query
|
|
|
|
* method
|
|
|
|
* @param array $returnRows the rows that should be returned for the result
|
2014-05-14 03:07:19 +04:00
|
|
|
* of the database query. If not provided, it wont be assumed that fetch
|
2014-04-19 16:56:16 +04:00
|
|
|
* will be called on the result
|
|
|
|
*/
|
|
|
|
protected function setMapperResult($sql, $arguments=array(), $returnRows=array(),
|
2015-01-29 21:23:57 +03:00
|
|
|
$limit=null, $offset=null, $expectClose=false){
|
2015-02-24 18:47:38 +03:00
|
|
|
if($limit === null && $offset === null) {
|
|
|
|
$this->db->expects($this->at($this->prepareAt))
|
|
|
|
->method('prepare')
|
|
|
|
->with($this->equalTo($sql))
|
|
|
|
->will(($this->returnValue($this->query)));
|
|
|
|
} elseif($limit !== null && $offset === null) {
|
|
|
|
$this->db->expects($this->at($this->prepareAt))
|
|
|
|
->method('prepare')
|
|
|
|
->with($this->equalTo($sql), $this->equalTo($limit))
|
|
|
|
->will(($this->returnValue($this->query)));
|
|
|
|
} elseif($limit === null && $offset !== null) {
|
|
|
|
$this->db->expects($this->at($this->prepareAt))
|
|
|
|
->method('prepare')
|
|
|
|
->with($this->equalTo($sql),
|
|
|
|
$this->equalTo(null),
|
|
|
|
$this->equalTo($offset))
|
|
|
|
->will(($this->returnValue($this->query)));
|
|
|
|
} else {
|
|
|
|
$this->db->expects($this->at($this->prepareAt))
|
|
|
|
->method('prepare')
|
|
|
|
->with($this->equalTo($sql),
|
|
|
|
$this->equalTo($limit),
|
|
|
|
$this->equalTo($offset))
|
|
|
|
->will(($this->returnValue($this->query)));
|
|
|
|
}
|
2014-04-19 16:56:16 +04:00
|
|
|
|
|
|
|
$this->iterators[] = new ArgumentIterator($returnRows);
|
|
|
|
|
|
|
|
$iterators = $this->iterators;
|
|
|
|
$fetchAt = $this->fetchAt;
|
|
|
|
|
2015-02-24 18:47:38 +03:00
|
|
|
$this->query->expects($this->any())
|
2014-05-14 03:07:19 +04:00
|
|
|
->method('fetch')
|
2014-04-19 16:56:16 +04:00
|
|
|
->will($this->returnCallback(
|
|
|
|
function() use ($iterators, $fetchAt){
|
|
|
|
$iterator = $iterators[$fetchAt];
|
|
|
|
$result = $iterator->next();
|
|
|
|
|
|
|
|
if($result === false) {
|
|
|
|
$fetchAt++;
|
|
|
|
}
|
|
|
|
|
2015-02-24 18:47:38 +03:00
|
|
|
$this->queryAt++;
|
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
return $result;
|
2015-01-29 22:10:29 +03:00
|
|
|
}
|
2014-04-19 16:56:16 +04:00
|
|
|
));
|
2015-01-29 21:23:57 +03:00
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
$index = 1;
|
|
|
|
foreach($arguments as $argument) {
|
|
|
|
switch (gettype($argument)) {
|
|
|
|
case 'integer':
|
|
|
|
$pdoConstant = \PDO::PARAM_INT;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 'NULL':
|
|
|
|
$pdoConstant = \PDO::PARAM_NULL;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 'boolean':
|
|
|
|
$pdoConstant = \PDO::PARAM_BOOL;
|
|
|
|
break;
|
2015-01-29 21:16:28 +03:00
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
default:
|
|
|
|
$pdoConstant = \PDO::PARAM_STR;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
$this->query->expects($this->at($this->queryAt))
|
|
|
|
->method('bindValue')
|
|
|
|
->with($this->equalTo($index),
|
|
|
|
$this->equalTo($argument),
|
|
|
|
$this->equalTo($pdoConstant));
|
|
|
|
$index++;
|
|
|
|
$this->queryAt++;
|
|
|
|
}
|
|
|
|
|
|
|
|
$this->query->expects($this->at($this->queryAt))
|
2015-02-24 18:47:38 +03:00
|
|
|
->method('execute');
|
2014-04-19 16:56:16 +04:00
|
|
|
$this->queryAt++;
|
|
|
|
|
2015-02-24 18:47:38 +03:00
|
|
|
|
|
|
|
|
|
|
|
if ($expectClose) {
|
|
|
|
$closing = $this->once();
|
|
|
|
} else {
|
|
|
|
$closing = $this->any();
|
2014-04-19 16:56:16 +04:00
|
|
|
}
|
2015-02-24 18:47:38 +03:00
|
|
|
$this->query->expects($closing)->method('closeCursor');
|
|
|
|
$this->queryAt++;
|
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
$this->prepareAt++;
|
|
|
|
$this->fetchAt++;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
class ArgumentIterator {
|
|
|
|
|
|
|
|
private $arguments;
|
2015-01-29 21:16:28 +03:00
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
public function __construct($arguments){
|
|
|
|
$this->arguments = $arguments;
|
|
|
|
}
|
2015-01-29 21:16:28 +03:00
|
|
|
|
2014-04-19 16:56:16 +04:00
|
|
|
public function next(){
|
|
|
|
$result = array_shift($this->arguments);
|
|
|
|
if($result === null){
|
|
|
|
return false;
|
|
|
|
} else {
|
|
|
|
return $result;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|