mirror of
https://github.com/danog/postgres.git
synced 2024-12-12 09:29:57 +01:00
75 lines
2.3 KiB
PHP
75 lines
2.3 KiB
PHP
<?php
|
|
|
|
namespace Amp\Postgres\Test;
|
|
|
|
use Amp\CancellationToken;
|
|
use Amp\CancellationTokenSource;
|
|
use Amp\Loop;
|
|
use Amp\Postgres\Connection;
|
|
use Amp\Postgres\ConnectionConfig as PostgresConnectionConfig;
|
|
use Amp\Promise;
|
|
use Amp\Sql\ConnectionConfig;
|
|
use Amp\TimeoutCancellationToken;
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
abstract class AbstractConnectTest extends TestCase
|
|
{
|
|
/**
|
|
* @param ConnectionConfig $connectionConfig
|
|
* @param CancellationToken|null $token
|
|
*
|
|
* @return Promise
|
|
*/
|
|
abstract public function connect(ConnectionConfig $connectionConfig, CancellationToken $token = null): Promise;
|
|
|
|
public function testConnect()
|
|
{
|
|
Loop::run(function () {
|
|
$connection = yield $this->connect(
|
|
PostgresConnectionConfig::fromString('host=localhost user=postgres'),
|
|
new TimeoutCancellationToken(100)
|
|
);
|
|
$this->assertInstanceOf(Connection::class, $connection);
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @depends testConnect
|
|
* @expectedException \Amp\CancelledException
|
|
*/
|
|
public function testConnectCancellationBeforeConnect()
|
|
{
|
|
Loop::run(function () {
|
|
$source = new CancellationTokenSource;
|
|
$token = $source->getToken();
|
|
$source->cancel();
|
|
$connection = yield $this->connect(PostgresConnectionConfig::fromString('host=localhost user=postgres'), $token);
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @depends testConnectCancellationBeforeConnect
|
|
*/
|
|
public function testConnectCancellationAfterConnect()
|
|
{
|
|
Loop::run(function () {
|
|
$source = new CancellationTokenSource;
|
|
$token = $source->getToken();
|
|
$connection = yield $this->connect(PostgresConnectionConfig::fromString('host=localhost user=postgres'), $token);
|
|
$this->assertInstanceOf(Connection::class, $connection);
|
|
$source->cancel();
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @depends testConnectCancellationBeforeConnect
|
|
* @expectedException \Amp\Sql\FailureException
|
|
*/
|
|
public function testConnectInvalidUser()
|
|
{
|
|
Loop::run(function () {
|
|
$connection = yield $this->connect(PostgresConnectionConfig::fromString('host=localhost user=invalid'), new TimeoutCancellationToken(100));
|
|
});
|
|
}
|
|
}
|