mirror of https://github.com/friendica/friendica
84 lines
2.2 KiB
PHP
84 lines
2.2 KiB
PHP
<?php
|
|
/**
|
|
* DatabaseTest class.
|
|
*/
|
|
|
|
namespace Friendica\Test;
|
|
|
|
use PDO;
|
|
use PHPUnit\DbUnit\DataSet\YamlDataSet;
|
|
use PHPUnit\DbUnit\TestCaseTrait;
|
|
use PHPUnit_Extensions_Database_DB_IDatabaseConnection;
|
|
|
|
/**
|
|
* Abstract class used by tests that need a database.
|
|
*/
|
|
abstract class DatabaseTest extends MockedTest
|
|
{
|
|
use TestCaseTrait;
|
|
|
|
// only instantiate pdo once for test clean-up/fixture load
|
|
static private $pdo = null;
|
|
|
|
// only instantiate PHPUnit_Extensions_Database_DB_IDatabaseConnection once per test
|
|
private $conn = null;
|
|
|
|
/**
|
|
* 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.
|
|
*
|
|
* @return PHPUnit_Extensions_Database_DB_IDatabaseConnection
|
|
* @see https://phpunit.de/manual/5.7/en/database.html
|
|
*/
|
|
protected function getConnection()
|
|
{
|
|
$server = $_SERVER;
|
|
|
|
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);
|
|
}
|
|
}
|
|
$this->conn = $this->createDefaultDBConnection(self::$pdo, getenv('MYSQL_DATABASE'));
|
|
}
|
|
|
|
return $this->conn;
|
|
}
|
|
|
|
/**
|
|
* Get dataset to populate the database with.
|
|
*
|
|
* @return YamlDataSet
|
|
* @see https://phtablepunit.de/manual/5.7/en/database.html
|
|
*/
|
|
protected function getDataSet()
|
|
{
|
|
return new YamlDataSet(__DIR__ . '/datasets/api.yml');
|
|
}
|
|
}
|