wpstudio/majestic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added MongoStatement method to count results before skip/limit

Browse Source
This commit is contained in:
Anton Grebnev
2012-02-06 18:12:23 +04:00
parent 0140c3f4ad
commit e5e7fdb7e6
3 changed files with 61 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
model/MongoStatement.php
View File

@ -16,6 +16,15 @@ class MongoStatement extends DbStatement
protected $insertId = false;
public function count()
{
if ($this->result instanceof MongoCursor) {
return $this->result->count();
} else {
throw new GeneralException('MongoStatement error. Impossible count result of opened cursor.');
}
}
public function order($sort = array())
{
if ($this->result instanceof MongoCursor) {

15
tests/model/MongoDriverTest.php
View File

@ -197,6 +197,21 @@ class MongoDriverTest extends PHPUnit_Framework_TestCase
* @runInSeparateProcess
* @group Mongo
*/
public function testCursorCount()
{
if (!defined('DEBUG')) {
define('DEBUG', false);
}
$mongo = new MongoDriver($this->conf);
$this->assertEquals(5, $mongo->find('items')->count(5));
$this->assertCount(3, $mongo->find('items')->limit(3)->fetchAll());
$this->assertEquals(5, $mongo->count('items'));
}
/**
* @runInSeparateProcess
* @group Mongo
*/
public function testGet()
{
if (!defined('DEBUG')) {

37
tests/model/MongoStatementTest.php
View File

@ -338,6 +338,43 @@ class MongoStatementTest extends PHPUnit_Framework_TestCase
* @runInSeparateProcess
* @group Mongo
*/
public function testCount()
{
if (!defined('DEBUG')) {
define('DEBUG', true);
}
$this->setDriverGetConnectionMethod()->setRequestForFetch();
$this->stmt->execute();
$this->assertSame(10, $this->stmt->count());
$this->stmt->fetch();
}
/**
* @runInSeparateProcess
* @group Mongo
*/
public function testCountException()
{
if (!defined('DEBUG')) {
define('DEBUG', true);
}
$this->setDriverGetConnectionMethod();
$this->request
->expects($this->once())
->method('execute')
->will($this->returnValue(array('some' => 'val')));
$this->stmt->execute();
$this->setExpectedException('GeneralException', 'MongoStatement error. Impossible count result of opened cursor');
$this->stmt->count();
}
/**
* @runInSeparateProcess
* @group Mongo
*/
public function testOrderException()
{
if (!defined('DEBUG')) {