Files
predis/tests/Predis/Collection/Iterator/SortedSetKeyTest.php
T
Daniele Alessandri 5ad00774e1 Rename for the last time all the iterator classes.
We are experimenting with a new approach at naming classes using less
redundant names by leveraging the containing namespace. The PHP "use"
directive is not limited to class names but can be used to import the
whole namespace, which means you can do something like this:

  use Predis\Collection\Iterator;
  // ...
  foreach (new Iterator\Keyspace($client) as $key) {
  	// ...
  }

Alternatively you can always rely on "use ... as ..." to import one of
the classes by giving it a more meaningful name in the context of the
root namespace:

  use Predis\Collection\Iterator\Keyspace as KeyspaceIterator;
  // ...
  foreach (new KeyspaceIterator($client) as $key) {
  	// ...
  }

In this specific case we chose to apply the -Key postfix to classes
iterating Redis keys to be more explicit about the fact that those
iterators does not work on local in-memory collections, but fetch
items from a key stored on a remote Redis server.
2013-11-07 12:09:59 +01:00

482 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 \PHPUnit_Framework_TestCase as StandardTestCase;
use Predis\Client;
use Predis\Profile\ServerProfile;
/**
* @group realm-iterators
*/
class SortedSetTest extends StandardTestCase
{
/**
* @group disconnected
* @expectedException Predis\NotSupportedException
* @expectedExceptionMessage The specified server profile does not support the `ZSCAN` command.
*/
public function testThrowsExceptionOnInvalidServerProfile()
{
$client = $this->getMock('Predis\ClientInterface');
$client->expects($this->any())
->method('getProfile')
->will($this->returnValue(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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(ServerProfile::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());
}
}