mirror of
https://github.com/predis/predis.git
synced 2026-10-09 21:38:07 +00:00
Daniele Alessandri
cc2a7657db
We have renamed most methods to drop the "command" suffix as it is quite redundant. Due to this change and thanks to variadic methods introduced with PHP 5.6 we took the opportunity to replace both "supportsCommand()" and "supportsCommands()" with a single new method "supports()". Added more stringent typehints for method arguments and typehints for return values now that we do not need to support anything below PHP 7.2. Also moved from using array() to [] in source code of class involved.
419 lines
13 KiB
PHP
419 lines
13 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\Connection\Cluster;
|
|
|
|
use PredisTestCase;
|
|
|
|
/**
|
|
*
|
|
*/
|
|
class PredisClusterTest extends PredisTestCase
|
|
{
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testExposesCommandHashStrategy()
|
|
{
|
|
$cluster = new PredisCluster();
|
|
$this->assertInstanceOf('Predis\Cluster\PredisStrategy', $cluster->getClusterStrategy());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testAddingConnectionsToCluster()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertSame(2, count($cluster));
|
|
$this->assertSame($connection1, $cluster->getConnectionById('127.0.0.1:7001'));
|
|
$this->assertSame($connection2, $cluster->getConnectionById('127.0.0.1:7002'));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testAddingConnectionsWithAliasParameterToCluster()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001?alias=node01');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002?alias=node02');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertSame(2, count($cluster));
|
|
$this->assertSame($connection1, $cluster->getConnectionByAlias('node01'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByAlias('node02'));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testRemovingConnectionsFromCluster()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001?alias=node01');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection3 = $this->getMockConnection('tcp://127.0.0.1:7003');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertTrue($cluster->remove($connection1));
|
|
$this->assertNull($cluster->getConnectionByAlias('node02'));
|
|
|
|
$this->assertFalse($cluster->remove($connection3));
|
|
|
|
$this->assertCount(1, $cluster);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testConnectForcesAllConnectionsToConnect()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('connect');
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('connect');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$cluster->connect();
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testDisconnectForcesAllConnectionsToDisconnect()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('disconnect');
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('disconnect');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$cluster->disconnect();
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsConnectedReturnsTrueIfAtLeastOneConnectionIsOpen()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->will($this->returnValue(false));
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->will($this->returnValue(true));
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertTrue($cluster->isConnected());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsConnectedReturnsFalseIfAllConnectionsAreClosed()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->will($this->returnValue(false));
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->will($this->returnValue(false));
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertFalse($cluster->isConnected());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testCanReturnAnIteratorForConnections()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertInstanceOf('Iterator', $iterator = $cluster->getIterator());
|
|
$connections = iterator_to_array($iterator);
|
|
|
|
$this->assertSame(array(
|
|
'127.0.0.1:7001' => $connection1,
|
|
'127.0.0.1:7002' => $connection2,
|
|
), iterator_to_array($iterator));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testReturnsCorrectConnectionUsingSlot()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertSame($connection1, $cluster->getConnectionBySlot(1839357934));
|
|
$this->assertSame($connection2, $cluster->getConnectionBySlot(2146453549));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testReturnsCorrectConnectionUsingKey()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertSame($connection1, $cluster->getConnectionByKey('node01:5431'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByKey('node02:3212'));
|
|
$this->assertSame($connection1, $cluster->getConnectionByKey('prefix:{node01:5431}'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByKey('prefix:{node02:3212}'));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testReturnsCorrectConnectionUsingCommandInstance()
|
|
{
|
|
$commands = $this->getCommandFactory();
|
|
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$set = $commands->create('set', array('node01:5431', 'foobar'));
|
|
$get = $commands->create('get', array('node01:5431'));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($get));
|
|
|
|
$set = $commands->create('set', array('prefix:{node01:5431}', 'foobar'));
|
|
$get = $commands->create('get', array('prefix:{node01:5431}'));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($get));
|
|
|
|
$set = $commands->create('set', array('node02:3212', 'foobar'));
|
|
$get = $commands->create('get', array('node02:3212'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection2, $cluster->getConnectionByCommand($get));
|
|
|
|
$set = $commands->create('set', array('prefix:{node02:3212}', 'foobar'));
|
|
$get = $commands->create('get', array('prefix:{node02:3212}'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection2, $cluster->getConnectionByCommand($get));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testThrowsExceptionOnNonShardableCommand()
|
|
{
|
|
$this->expectException('Predis\NotSupportedException');
|
|
$this->expectExceptionMessage("Cannot use 'PING' over clusters of connections.");
|
|
|
|
$ping = $this->getCommandFactory()->create('ping');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($this->getMockConnection('tcp://127.0.0.1:6379'));
|
|
|
|
$cluster->getConnectionByCommand($ping);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSupportsKeyHashTags()
|
|
{
|
|
$commands = $this->getCommandFactory();
|
|
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:6379');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:6380');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$set = $commands->create('set', array('{node:1001}:foo', 'foobar'));
|
|
$get = $commands->create('get', array('{node:1001}:foo'));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($get));
|
|
|
|
$set = $commands->create('set', array('{node:1001}:bar', 'foobar'));
|
|
$get = $commands->create('get', array('{node:1001}:bar'));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($set));
|
|
$this->assertSame($connection1, $cluster->getConnectionByCommand($get));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testWritesCommandToCorrectConnection()
|
|
{
|
|
$command = $this->getCommandFactory()->create('get', array('node01:5431'));
|
|
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('writeRequest')
|
|
->with($command);
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->never())
|
|
->method('writeRequest');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$cluster->writeRequest($command);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testReadsCommandFromCorrectConnection()
|
|
{
|
|
$command = $this->getCommandFactory()->create('get', array('node02:3212'));
|
|
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->never())
|
|
->method('readResponse');
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('readResponse')
|
|
->with($command);
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$cluster->readResponse($command);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testExecutesCommandOnCorrectConnection()
|
|
{
|
|
$command = $this->getCommandFactory()->create('get', array('node01:5431'));
|
|
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('executeCommand')
|
|
->with($command);
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->never())
|
|
->method('executeCommand');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$cluster->executeCommand($command);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testCanBeSerialized()
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001?alias=first');
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002?alias=second');
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
// We use the following line to initialize the underlying hashring.
|
|
$cluster->getConnectionByKey('foo');
|
|
$unserialized = unserialize(serialize($cluster));
|
|
|
|
$this->assertEquals($cluster, $unserialized);
|
|
}
|
|
}
|