Files
piratepoet/tests/integration/Doctrine/ConnectionFactoryTest.php
Jan Jakeš aa13de72cc Fix setup of socket and IPv6 connection
[MAILPOET-2014]
2019-07-24 15:03:16 -04:00

126 lines
4.8 KiB
PHP

<?php
namespace MailPoet\Test\Config;
use MailPoet\Config\Env;
use MailPoet\Doctrine\ConnectionFactory;
use MailPoet\Doctrine\SerializableConnection;
use MailPoetVendor\Doctrine\DBAL\Driver\PDOMySql;
use MailPoetVendor\Doctrine\DBAL\Platforms\MySqlPlatform;
use PDO;
class ConnectionFactoryTest extends \MailPoetTest {
private $env_backup = [];
function _before() {
parent::_before();
$this->env_backup['db_host'] = Env::$db_host;
$this->env_backup['db_is_ipv6'] = Env::$db_is_ipv6;
$this->env_backup['db_port'] = Env::$db_port;
$this->env_backup['db_socket'] = Env::$db_socket;
}
function testItSetsUpConnection() {
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
expect($connection)->isInstanceOf(SerializableConnection::class);
expect($connection->getWrappedConnection())->isInstanceOf(PDO::class);
expect($connection->getDriver())->isInstanceOf(PDOMySql\Driver::class);
expect($connection->getDatabasePlatform())->isInstanceOf(MySqlPlatform::class);
expect($connection->getHost())->equals(Env::$db_host);
expect($connection->getPort())->equals(Env::$db_port);
expect($connection->getParams())->notContains('unix_socket');
expect($connection->getUsername())->equals(Env::$db_username);
expect($connection->getPassword())->equals(Env::$db_password);
expect($connection->getParams()['charset'])->equals(Env::$db_charset);
expect($connection->getDatabase())->equals(Env::$db_name);
}
function testItSetsUpSocket() {
Env::$db_host = null;
Env::$db_port = null;
Env::$db_socket = 'socket';
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
expect($connection->getHost())->null();
expect($connection->getPort())->null();
expect($connection->getParams()['unix_socket'])->equals('socket');
}
function testItSetsUpIpV6() {
Env::$db_is_ipv6 = true;
Env::$db_host = '::1';
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
expect($connection->getHost())->equals('[::1]');
Env::$db_host = 'b57e:9b70:ab96:6a0b:5ba2:49e3:ebba:a036';
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
expect($connection->getHost())->equals('[b57e:9b70:ab96:6a0b:5ba2:49e3:ebba:a036]');
// try to actually connect to the DB over IPv6
Env::$db_host = '::ffff:' . gethostbyname($this->env_backup['db_host']);
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
expect($connection->getWrappedConnection())->isInstanceOf(PDO::class);
expect($connection->executeQuery('SELECT 1')->fetchColumn())->same('1');
}
function testItSetsDriverOptions() {
$connection_factory = new ConnectionFactory();
$connection = $connection_factory->createConnection();
$result = $connection->executeQuery('
SELECT
@@session.sql_mode,
@@session.time_zone,
@@session.wait_timeout,
@@session.character_set_client,
@@session.character_set_connection,
@@session.character_set_results,
@@session.collation_connection
')->fetch();
// check timezone, SQL mode, wait timeout
expect($result['@@session.sql_mode'])->notContains('ONLY_FULL_GROUP_BY');
expect($result['@@session.time_zone'])->equals(Env::$db_timezone_offset);
expect($result['@@session.wait_timeout'])->greaterOrEquals(60);
// check "SET NAMES ... COLLATE ..."
expect($result['@@session.character_set_client'])->equals(Env::$db_charset);
expect($result['@@session.character_set_connection'])->equals(Env::$db_charset);
expect($result['@@session.character_set_results'])->equals(Env::$db_charset);
expect($result['@@session.collation_connection'])->equals(Env::$db_collation);
}
function testItSelectivelyUpdatesWaitTimeoutOption() {
// timeout will be kept from DB defaults
$connection_factory = $this->make(ConnectionFactory::class, [
'min_wait_timeout' => 1,
]);
$connection = $connection_factory->createConnection();
$current = $connection->executeQuery('SELECT @@session.wait_timeout')->fetchColumn();
expect($current)->greaterThan(1);
// timeout will be set to higher value
$connection_factory = $this->make(ConnectionFactory::class, [
'min_wait_timeout' => 999999,
]);
$connection = $connection_factory->createConnection();
$current = $connection->executeQuery('SELECT @@session.wait_timeout')->fetchColumn();
expect($current)->equals(999999);
}
function _after() {
parent::_after();
Env::$db_host = $this->env_backup['db_host'];
Env::$db_port = $this->env_backup['db_port'];
Env::$db_is_ipv6 = $this->env_backup['db_is_ipv6'];
Env::$db_socket = $this->env_backup['db_socket'];
}
}