geckon01/predis
1
0
Fork 0
mirror of https://github.com/predis/predis.git synced 2026-08-18 13:23:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
00000fde4e19690c7035cb8fbb9984f4fca4f287
predis/tests/Predis/Cluster/PredisStrategyTest.php
T
Daniele Alessandri 00000fde4e Add new command: GEORADIUS (Redis 3.2.0).
2016-05-24 22:30:06 +02:00

415 lines
14 KiB
PHP
Raw Blame History

<?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\Cluster;
use Predis\Profile;
use PredisTestCase;
/**
*
*/
class PredisStrategyTest extends PredisTestCase
{
/**
* @group disconnected
*/
public function testSupportsKeyTags()
{
// NOTE: 32 and 64 bits PHP runtimes can produce different hash values.
$expected = PHP_INT_SIZE == 4 ? -1954026732 : 2340940564;
$strategy = $this->getClusterStrategy();
$this->assertSame($expected, $strategy->getSlotByKey('{foo}'));
$this->assertSame($expected, $strategy->getSlotByKey('{foo}:bar'));
$this->assertSame($expected, $strategy->getSlotByKey('{foo}:baz'));
$this->assertSame($expected, $strategy->getSlotByKey('bar:{foo}:baz'));
$this->assertSame($expected, $strategy->getSlotByKey('bar:{foo}:{baz}'));
$this->assertSame($expected, $strategy->getSlotByKey('bar:{foo}:baz{}'));
$this->assertSame(PHP_INT_SIZE == 4 ? -1355751440 : 2939215856, $strategy->getSlotByKey('{}bar:{foo}:baz'));
$this->assertSame(PHP_INT_SIZE == 4 ? -18873278 : 4276094018, $strategy->getSlotByKey(''));
$this->assertSame(PHP_INT_SIZE == 4 ? -1574052038 : 2720915258, $strategy->getSlotByKey('{}'));
}
/**
* @group disconnected
*/
public function testSupportedCommands()
{
$strategy = $this->getClusterStrategy();
$this->assertSame($this->getExpectedCommands(), $strategy->getSupportedCommands());
}
/**
* @group disconnected
*/
public function testReturnsNullOnUnsupportedCommand()
{
$strategy = $this->getClusterStrategy();
$command = Profile\Factory::getDevelopment()->createCommand('ping');
$this->assertNull($strategy->getSlot($command));
}
/**
* @group disconnected
*/
public function testFirstKeyCommands()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('key');
foreach ($this->getExpectedCommands('keys-first') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testAllKeysCommands()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('{key}:1', '{key}:2', '{key}:3', '{key}:4');
foreach ($this->getExpectedCommands('keys-all') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testInterleavedKeysCommands()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('{key}:1', 'value1', '{key}:2', 'value2');
foreach ($this->getExpectedCommands('keys-interleaved') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testKeysForBlockingListCommands()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('{key}:1', '{key}:2', 10);
foreach ($this->getExpectedCommands('keys-blockinglist') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testKeysForZsetAggregationCommands()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('{key}:destination', 2, '{key}:1', '{key}:1', array('aggregate' => 'SUM'));
foreach ($this->getExpectedCommands('keys-zaggregated') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testKeysForBitOpCommand()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('AND', '{key}:destination', '{key}:src:1', '{key}:src:2');
foreach ($this->getExpectedCommands('keys-bitop') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testKeysForGeoradiusCommand()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$commandID = 'GEORADIUS';
$command = $profile->createCommand($commandID, array('{key}:1', 10, 10, 1, 'km'));
$this->assertNotNull($strategy->getSlot($command), $commandID);
$command = $profile->createCommand($commandID, array('{key}:1', 10, 10, 1, 'km', 'store', '{key}:2', 'storedist', '{key}:3'));
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
/**
* @group disconnected
*/
public function testKeysForEvalCommand()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$arguments = array('%SCRIPT%', 2, '{key}:1', '{key}:2', 'value1', 'value2');
foreach ($this->getExpectedCommands('keys-script') as $commandID) {
$command = $profile->createCommand($commandID, $arguments);
$this->assertNotNull($strategy->getSlot($command), $commandID);
}
}
/**
* @group disconnected
*/
public function testKeysForScriptCommand()
{
$strategy = $this->getClusterStrategy();
$arguments = array('{key}:1', '{key}:2', 'value1', 'value2');
$command = $this->getMock('Predis\Command\ScriptCommand', array('getScript', 'getKeysCount'));
$command->expects($this->once())
->method('getScript')
->will($this->returnValue('return true'));
$command->expects($this->exactly(2))
->method('getKeysCount')
->will($this->returnValue(2));
$command->setArguments($arguments);
$this->assertNotNull($strategy->getSlot($command), "Script Command [{$command->getId()}]");
}
/**
* @group disconnected
*/
public function testUnsettingCommandHandler()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$strategy->setCommandHandler('set');
$strategy->setCommandHandler('get', null);
$command = $profile->createCommand('set', array('key', 'value'));
$this->assertNull($strategy->getSlot($command));
$command = $profile->createCommand('get', array('key'));
$this->assertNull($strategy->getSlot($command));
}
/**
* @group disconnected
*/
public function testSettingCustomCommandHandler()
{
$strategy = $this->getClusterStrategy();
$profile = Profile\Factory::getDevelopment();
$callable = $this->getMock('stdClass', array('__invoke'));
$callable->expects($this->once())
->method('__invoke')
->with($this->isInstanceOf('Predis\Command\CommandInterface'))
->will($this->returnValue('key'));
$strategy->setCommandHandler('get', $callable);
$command = $profile->createCommand('get', array('key'));
$this->assertNotNull($strategy->getSlot($command));
}
// ******************************************************************** //
// ---- HELPER METHODS ------------------------------------------------ //
// ******************************************************************** //
/**
* Creates the default cluster strategy object.
*
* @return StrategyInterface
*/
protected function getClusterStrategy()
{
$strategy = new PredisStrategy();
$connection = $this->getMock('Predis\Connection\NodeConnectionInterface');
$strategy->getDistributor()->add($connection);
return $strategy;
}
/**
* Returns the list of expected supported commands.
*
* @param string $type Optional type of command (based on its keys)
*
* @return array
*/
protected function getExpectedCommands($type = null)
{
$commands = array(
/* commands operating on the key space */
'EXISTS' => 'keys-all',
'DEL' => 'keys-all',
'TYPE' => 'keys-first',
'EXPIRE' => 'keys-first',
'EXPIREAT' => 'keys-first',
'PERSIST' => 'keys-first',
'PEXPIRE' => 'keys-first',
'PEXPIREAT' => 'keys-first',
'TTL' => 'keys-first',
'PTTL' => 'keys-first',
'SORT' => 'keys-first', // TODO
'DUMP' => 'keys-first',
'RESTORE' => 'keys-first',
/* commands operating on string values */
'APPEND' => 'keys-first',
'DECR' => 'keys-first',
'DECRBY' => 'keys-first',
'GET' => 'keys-first',
'GETBIT' => 'keys-first',
'MGET' => 'keys-all',
'SET' => 'keys-first',
'GETRANGE' => 'keys-first',
'GETSET' => 'keys-first',
'INCR' => 'keys-first',
'INCRBY' => 'keys-first',
'INCRBYFLOAT' => 'keys-first',
'SETBIT' => 'keys-first',
'SETEX' => 'keys-first',
'MSET' => 'keys-interleaved',
'MSETNX' => 'keys-interleaved',
'SETNX' => 'keys-first',
'SETRANGE' => 'keys-first',
'STRLEN' => 'keys-first',
'SUBSTR' => 'keys-first',
'BITOP' => 'keys-bitop',
'BITCOUNT' => 'keys-first',
'BITFIELD' => 'keys-first',
/* commands operating on lists */
'LINSERT' => 'keys-first',
'LINDEX' => 'keys-first',
'LLEN' => 'keys-first',
'LPOP' => 'keys-first',
'RPOP' => 'keys-first',
'RPOPLPUSH' => 'keys-all',
'BLPOP' => 'keys-blockinglist',
'BRPOP' => 'keys-blockinglist',
'BRPOPLPUSH' => 'keys-blockinglist',
'LPUSH' => 'keys-first',
'LPUSHX' => 'keys-first',
'RPUSH' => 'keys-first',
'RPUSHX' => 'keys-first',
'LRANGE' => 'keys-first',
'LREM' => 'keys-first',
'LSET' => 'keys-first',
'LTRIM' => 'keys-first',
/* commands operating on sets */
'SADD' => 'keys-first',
'SCARD' => 'keys-first',
'SDIFF' => 'keys-all',
'SDIFFSTORE' => 'keys-all',
'SINTER' => 'keys-all',
'SINTERSTORE' => 'keys-all',
'SUNION' => 'keys-all',
'SUNIONSTORE' => 'keys-all',
'SISMEMBER' => 'keys-first',
'SMEMBERS' => 'keys-first',
'SSCAN' => 'keys-first',
'SPOP' => 'keys-first',
'SRANDMEMBER' => 'keys-first',
'SREM' => 'keys-first',
/* commands operating on sorted sets */
'ZADD' => 'keys-first',
'ZCARD' => 'keys-first',
'ZCOUNT' => 'keys-first',
'ZINCRBY' => 'keys-first',
'ZINTERSTORE' => 'keys-zaggregated',
'ZRANGE' => 'keys-first',
'ZRANGEBYSCORE' => 'keys-first',
'ZRANK' => 'keys-first',
'ZREM' => 'keys-first',
'ZREMRANGEBYRANK' => 'keys-first',
'ZREMRANGEBYSCORE' => 'keys-first',
'ZREVRANGE' => 'keys-first',
'ZREVRANGEBYSCORE' => 'keys-first',
'ZREVRANK' => 'keys-first',
'ZSCORE' => 'keys-first',
'ZUNIONSTORE' => 'keys-zaggregated',
'ZSCAN' => 'keys-first',
'ZLEXCOUNT' => 'keys-first',
'ZRANGEBYLEX' => 'keys-first',
'ZREMRANGEBYLEX' => 'keys-first',
'ZREVRANGEBYLEX' => 'keys-first',
/* commands operating on hashes */
'HDEL' => 'keys-first',
'HEXISTS' => 'keys-first',
'HGET' => 'keys-first',
'HGETALL' => 'keys-first',
'HMGET' => 'keys-first',
'HMSET' => 'keys-first',
'HINCRBY' => 'keys-first',
'HINCRBYFLOAT' => 'keys-first',
'HKEYS' => 'keys-first',
'HLEN' => 'keys-first',
'HSET' => 'keys-first',
'HSETNX' => 'keys-first',
'HVALS' => 'keys-first',
'HSCAN' => 'keys-first',
'HSTRLEN' => 'keys-first',
/* commands operating on HyperLogLog */
'PFADD' => 'keys-first',
'PFCOUNT' => 'keys-all',
'PFMERGE' => 'keys-all',
/* scripting */
'EVAL' => 'keys-script',
'EVALSHA' => 'keys-script',
/* commands performing geospatial operations */
'GEOADD' => 'keys-first',
'GEOHASH' => 'keys-first',
'GEOPOS' => 'keys-first',
'GEODIST' => 'keys-first',
'GEORADIUS' => 'keys-georadius',
);
if (isset($type)) {
$commands = array_filter($commands, function ($expectedType) use ($type) {
return $expectedType === $type;
});
}
return array_keys($commands);
}
}
Reference in New Issue View Git Blame Copy Permalink