59 lines
1.2 KiB
PHP
59 lines
1.2 KiB
PHP
|
<?php
|
||
|
|
/**
|
||
|
|
* DatabaseTest class.
|
||
|
|
*/
|
||
|
|
|
||
|
|
namespace Friendica\Test;
|
||
|
|
|
||
|
|
use Friendica\Database\Database;
|
||
|
|
use Friendica\Test\Util\Database\StaticDatabase;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Abstract class used by tests that need a database.
|
||
|
|
*/
|
||
|
|
trait DatabaseTestTrait
|
||
|
|
{
|
||
|
|
protected function setUp()
|
||
|
|
{
|
||
|
|
StaticDatabase::statConnect($_SERVER);
|
||
|
|
// Rollbacks every DB usage (in case the test couldn't call tearDown)
|
||
|
|
StaticDatabase::statRollback();
|
||
|
|
// Start the first, outer transaction
|
||
|
|
StaticDatabase::getGlobConnection()->beginTransaction();
|
||
|
|
|
||
|
|
parent::setUp();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function tearDown()
|
||
|
|
{
|
||
|
|
// Rollbacks every DB usage so we don't commit anything into the DB
|
||
|
|
StaticDatabase::statRollback();
|
||
|
|
|
||
|
|
parent::tearDown();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Loads a given DB fixture for this DB test
|
||
|
|
*
|
||
|
|
* @param string $fixture The path to the fixture
|
||
|
|
* @param Database $dba The DB connection
|
||
|
|
*
|
||
|
|
* @throws \Exception
|
||
|
|
*/
|
||
|
|
protected function loadFixture(string $fixture, Database $dba)
|
||
|
|
{
|
||
|
|
$data = include $fixture;
|
||
|
|
|
||
|
|
foreach ($data as $tableName => $rows) {
|
||
|
|
if (!is_array($rows)) {
|
||
|
|
$dba->p('TRUNCATE TABLE `' . $tableName . '``');
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
foreach ($rows as $row) {
|
||
|
|
$dba->insert($tableName, $row);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|