<?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);
			}
		}
	}
}