sql/test/AbstractPoolTest.php
Niklas Keller cf131fa9f0 Fix tests
2018-09-13 20:54:37 +02:00

77 lines
2.1 KiB
PHP

<?php
namespace Amp\Sql\Test;
use Amp\Delayed;
use Amp\Loop;
use Amp\Promise;
use Amp\Sql\AbstractPool;
use Amp\Sql\ConnectionConfig;
use Amp\Sql\Connector;
use Amp\Sql\Link;
use Amp\Success;
use PHPUnit\Framework\TestCase;
class AbstractPoolTest extends TestCase
{
/**
* @expectedException \Error
* @expectedExceptionMessage Pool must contain at least one connection
*/
public function testInvalidMaxConnections()
{
$mock = $this->getMockBuilder(AbstractPool::class)
->setConstructorArgs([$this->createMock(ConnectionConfig::class), 0])
->getMock();
}
public function testIdleConnectionsRemovedAfterTimeout()
{
Loop::run(function () {
$now = \time();
$connector = $this->createMock(Connector::class);
$connector->method('connect')
->willReturnCallback(function () use ($now): Promise {
$link = $this->createMock(Link::class);
$link->method('getLastUsedAt')
->willReturn($now);
$link->method('isAlive')
->willReturn(true);
$link->method('query')
->willReturnCallback(function () {
return new Delayed(100);
});
return new Success($link);
});
/** @var AbstractPool $pool */
$pool = $this->getMockBuilder(AbstractPool::class)
->setConstructorArgs([$this->createMock(ConnectionConfig::class), 100, 2, $connector])
->getMockForAbstractClass();
$count = 3;
$promises = [];
for ($i = 0; $i < $count; ++$i) {
$promises[] = $pool->query("SELECT $i");
}
$results = yield $promises;
$this->assertSame($count, $pool->getConnectionCount());
yield new Delayed(1000);
$this->assertSame($count, $pool->getConnectionCount());
yield new Delayed(1000);
$this->assertSame(0, $pool->getConnectionCount());
});
}
}