friendica/tests/DatabaseTest.php

84 lines
2.2 KiB
PHP
Raw Normal View History

2018-04-09 19:23:41 +00:00
<?php
/**
* DatabaseTest class.
*/
namespace Friendica\Test;
2019-07-26 13:54:14 +00:00
use PDO;
2018-04-09 19:23:41 +00:00
use PHPUnit\DbUnit\DataSet\YamlDataSet;
use PHPUnit\DbUnit\TestCaseTrait;
use PHPUnit_Extensions_Database_DB_IDatabaseConnection;
2018-10-22 18:59:51 +00:00
2018-04-09 19:23:41 +00:00
/**
* Abstract class used by tests that need a database.
*/
2018-11-01 12:44:47 +00:00
abstract class DatabaseTest extends MockedTest
2018-04-09 19:23:41 +00:00
{
2018-10-09 22:20:08 +00:00
use TestCaseTrait;
2019-07-26 13:54:14 +00:00
// only instantiate pdo once for test clean-up/fixture load
static private $pdo = null;
2019-07-04 19:39:49 +00:00
2019-07-26 13:54:14 +00:00
// only instantiate PHPUnit_Extensions_Database_DB_IDatabaseConnection once per test
private $conn = null;
2019-07-04 19:39:49 +00:00
2018-04-09 19:23:41 +00:00
/**
* Get database connection.
*
* This function is executed before each test in order to get a database connection that can be used by tests.
* If no prior connection is available, it tries to create one using the USER, PASS and DB environment variables.
*
* If it could not connect to the database, the test is skipped.
*
2018-04-09 19:23:41 +00:00
* @return PHPUnit_Extensions_Database_DB_IDatabaseConnection
* @see https://phpunit.de/manual/5.7/en/database.html
*/
protected function getConnection()
{
2019-07-26 13:54:14 +00:00
$server = $_SERVER;
2019-07-26 13:54:14 +00:00
if ($this->conn === null) {
if (self::$pdo == null) {
if (!empty($server['MYSQL_HOST'])
&& !empty($server['MYSQL_USERNAME'] || !empty($server['MYSQL_USER']))
&& $server['MYSQL_PASSWORD'] !== false
&& !empty($server['MYSQL_DATABASE'])) {
$connect = "mysql:host=" . $server['MYSQL_HOST'] . ";dbname=" . $server['MYSQL_DATABASE'];
if (!empty($server['MYSQL_PORT'])) {
$connect .= ";port=" . $server['MYSQL_PORT'];
}
if (!empty($server['MYSQL_USERNAME'])) {
$db_user = $server['MYSQL_USERNAME'];
} else {
$db_user = $server['MYSQL_USER'];
}
$db_pass = (string)$server['MYSQL_PASSWORD'];
self::$pdo = @new PDO($connect, $db_user, $db_pass);
self::$pdo->setAttribute(PDO::ATTR_EMULATE_PREPARES, false);
}
2019-07-04 19:39:49 +00:00
}
2019-07-26 13:54:14 +00:00
$this->conn = $this->createDefaultDBConnection(self::$pdo, getenv('MYSQL_DATABASE'));
2018-04-09 19:23:41 +00:00
}
2019-07-26 13:54:14 +00:00
return $this->conn;
2018-04-09 19:23:41 +00:00
}
/**
* Get dataset to populate the database with.
2019-07-26 13:54:14 +00:00
*
2018-04-09 19:23:41 +00:00
* @return YamlDataSet
2019-07-26 13:54:14 +00:00
* @see https://phtablepunit.de/manual/5.7/en/database.html
2018-04-09 19:23:41 +00:00
*/
protected function getDataSet()
{
return new YamlDataSet(__DIR__ . '/datasets/api.yml');
2018-04-09 19:23:41 +00:00
}
}