|
@@ -0,0 +1,481 @@
|
|
|
+<?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());
|
|
|
+ }
|
|
|
+}
|