predis/tests/PHPUnit/DistributionStrategyTestCase.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\Distribution;

use \PHPUnit_Framework_TestCase as StandardTestCase;

/**
 *
 */
abstract class DistributionStrategyTestCase extends StandardTestCase
{
    /**
     * Returns a new instance of the tested distributor.
     *
     * @return Predis\Distribution\IDistributionStrategy
     */
    protected abstract function getDistributorInstance();

    /**
     * Returns a list of nodes from the hashring.
     *
     * @param IDistributionStrategy $ring Hashring instance.
     * @param int $iterations Number of nodes to fetch.
     * @return array Nodes from the hashring.
     */
    protected function getNodes(IDistributionStrategy $ring, $iterations = 10)
    {
        $nodes = array();

        for ($i = 0; $i < $iterations; $i++) {
            $key = $ring->generateKey($i * $i);
            $nodes[] = $ring->get($key);
        }

        return $nodes;
    }

    /**
     * @group disconnected
     */
    public function testEmptyRingThrowsException()
    {
        $this->setExpectedException('Predis\Distribution\EmptyRingException');

        $ring = $this->getDistributorInstance();
        $ring->get('nodekey');
    }

    /**
     * @group disconnected
     */
    public function testRemoveOnEmptyRingDoesNotThrowException()
    {
        $ring = $this->getDistributorInstance();

        $this->assertNull($ring->remove('node'));
    }
}