mirror of
https://github.com/predis/predis.git
synced 2026-08-22 02:30:56 +00:00
Peter
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(): void
|
|
{
|
|
$cluster = new PredisCluster();
|
|
$this->assertInstanceOf('Predis\Cluster\PredisStrategy', $cluster->getClusterStrategy());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testAddingConnectionsToCluster(): void
|
|
{
|
|
$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->assertCount(2, $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(): void
|
|
{
|
|
$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->assertCount(2, $cluster);
|
|
$this->assertSame($connection1, $cluster->getConnectionByAlias('node01'));
|
|
$this->assertSame($connection2, $cluster->getConnectionByAlias('node02'));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testRemovingConnectionsFromCluster(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->willReturn(false);
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->willReturn(true);
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertTrue($cluster->isConnected());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsConnectedReturnsFalseIfAllConnectionsAreClosed(): void
|
|
{
|
|
$connection1 = $this->getMockConnection('tcp://127.0.0.1:7001');
|
|
$connection1
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->willReturn(false);
|
|
|
|
$connection2 = $this->getMockConnection('tcp://127.0.0.1:7002');
|
|
$connection2
|
|
->expects($this->once())
|
|
->method('isConnected')
|
|
->willReturn(false);
|
|
|
|
$cluster = new PredisCluster();
|
|
|
|
$cluster->add($connection1);
|
|
$cluster->add($connection2);
|
|
|
|
$this->assertFalse($cluster->isConnected());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testCanReturnAnIteratorForConnections(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$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);
|
|
}
|
|
}
|