mirror of
https://github.com/predis/predis.git
synced 2026-08-22 10:22:07 +00:00
b253bdc41e
We now have a base test case class for Predis (namely PredisTestCase) grouping various commonly used utility methods shared by all of the tests in the suite, greatly improving reusability.
481 lines
16 KiB
PHP
481 lines
16 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Predis package.
|
|
*
|
|
* (c) Daniele Alessandri <suppakilla@gmail.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Predis\Collection\Iterator;
|
|
|
|
use PredisTestCase;
|
|
use Predis\Client;
|
|
use Predis\Profile;
|
|
|
|
/**
|
|
* @group realm-iterators
|
|
*/
|
|
class SortedSetTest extends PredisTestCase
|
|
{
|
|
/**
|
|
* @group disconnected
|
|
* @expectedException Predis\NotSupportedException
|
|
* @expectedExceptionMessage The specified server profile does not support the `ZSCAN` command.
|
|
*/
|
|
public function testThrowsExceptionOnInvalidProfile()
|
|
{
|
|
$client = $this->getMock('Predis\ClientInterface');
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.0')));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithNoResults()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->once())
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(0, array())));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationOnSingleFetch()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->once())
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0), array('member:3rd', 3.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(3.0, $iterator->current());
|
|
$this->assertSame('member:3rd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationOnMultipleFetches()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(2, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 2, array())
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:3rd', 3.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(3.0, $iterator->current());
|
|
$this->assertSame('member:3rd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationOnMultipleFetchesAndHoleInFirstFetch()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(4, array())));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 4, array())
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationOnMultipleFetchesAndHoleInMidFetch()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(2, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 2, array())
|
|
->will($this->returnValue(array(5, array())));
|
|
$client->expects($this->at(3))
|
|
->method('zscan')
|
|
->with('key:zset', 5, array())
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:3rd', 3.0)
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(3.0, $iterator->current());
|
|
$this->assertSame('member:3rd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionMatch()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('MATCH' => 'member:*'))
|
|
->will($this->returnValue(array(2, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', 'member:*');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionMatchOnMultipleFetches()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('MATCH' => 'member:*'))
|
|
->will($this->returnValue(array(1, array(
|
|
array('member:1st', 1.0),
|
|
))));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 1, array('MATCH' => 'member:*'))
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', 'member:*');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionCount()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('COUNT' => 2))
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', null, 2);
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionCountOnMultipleFetches()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('COUNT' => 1))
|
|
->will($this->returnValue(array(1, array(
|
|
array('member:1st', 1.0),
|
|
))));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 1, array('COUNT' => 1))
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', null, 1);
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionsMatchAndCount()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('MATCH' => 'member:*', 'COUNT' => 2))
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', 'member:*', 2);
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationWithOptionsMatchAndCountOnMultipleFetches()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->at(1))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array('MATCH' => 'member:*', 'COUNT' => 1))
|
|
->will($this->returnValue(array(1, array(
|
|
array('member:1st', 1.0),
|
|
))));
|
|
$client->expects($this->at(2))
|
|
->method('zscan')
|
|
->with('key:zset', 1, array('MATCH' => 'member:*', 'COUNT' => 1))
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset', 'member:*', 1);
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIterationRewindable()
|
|
{
|
|
$client = $this->getMock('Predis\Client', array('getProfile', 'zscan'));
|
|
|
|
$client->expects($this->any())
|
|
->method('getProfile')
|
|
->will($this->returnValue(Profile\Factory::get('2.8')));
|
|
$client->expects($this->exactly(2))
|
|
->method('zscan')
|
|
->with('key:zset', 0, array())
|
|
->will($this->returnValue(array(0, array(
|
|
array('member:1st', 1.0), array('member:2nd', 2.0),
|
|
))));
|
|
|
|
$iterator = new SortedSetKey($client, 'key:zset');
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->rewind();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(1.0, $iterator->current());
|
|
$this->assertSame('member:1st', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertTrue($iterator->valid());
|
|
$this->assertSame(2.0, $iterator->current());
|
|
$this->assertSame('member:2nd', $iterator->key());
|
|
|
|
$iterator->next();
|
|
$this->assertFalse($iterator->valid());
|
|
}
|
|
}
|