<?php
/**
- * @copyright Copyright (C) 2020, Friendica
+ * @copyright Copyright (C) 2010-2023, the Friendica project
*
* @license GNU AGPL version 3 or any later version
*
*/
trait DatabaseTestTrait
{
- protected function setUp()
+ protected function setUpDb()
{
StaticDatabase::statConnect($_SERVER);
// Rollbacks every DB usage (in case the test couldn't call tearDown)
StaticDatabase::statRollback();
+ // Rollback the first, outer transaction just 2 be sure
+ StaticDatabase::getGlobConnection()->rollBack();
// Start the first, outer transaction
StaticDatabase::getGlobConnection()->beginTransaction();
-
- parent::setUp();
}
- protected function tearDown()
+ protected function tearDownDb()
{
- // Rollbacks every DB usage so we don't commit anything into the DB
- StaticDatabase::statRollback();
-
- parent::tearDown();
+ try {
+ // Rollbacks every DB usage so we don't commit anything into the DB
+ StaticDatabase::statRollback();
+ } catch (\PDOException $exception) {
+ print_r("Found already rolled back transaction");
+ }
}
/**
* Loads a given DB fixture for this DB test
*
- * @param string $fixture The path to the fixture
+ * @param string[][] $fixture The fixture array
* @param Database $dba The DB connection
*
* @throws \Exception
*/
- protected function loadFixture(string $fixture, Database $dba)
+ protected function loadDirectFixture(array $fixture, Database $dba)
{
- $data = include $fixture;
+ foreach ($fixture as $tableName => $rows) {
+ if (is_numeric($tableName)) {
+ continue;
+ }
- foreach ($data as $tableName => $rows) {
if (!is_array($rows)) {
- $dba->p('TRUNCATE TABLE `' . $tableName . '``');
+ $dba->e('TRUNCATE TABLE `' . $tableName . '``');
continue;
}
foreach ($rows as $row) {
- $dba->insert($tableName, $row);
+ if (is_array($row)) {
+ $dba->insert($tableName, $row, true);
+ } else {
+ throw new \Exception('row isn\'t an array');
+ }
}
}
}
+
+ /**
+ * Loads a given DB fixture-file 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;
+
+ $this->loadDirectFixture($data, $dba);
+ }
}