Friendica Communications Platform (please note that this is a clone of the repository at github, issues are handled there) https://friendi.ca
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

79 lines
2.0KB

  1. <?php
  2. /**
  3. * @copyright Copyright (C) 2020, Friendica
  4. *
  5. * @license GNU AGPL version 3 or any later version
  6. *
  7. * This program is free software: you can redistribute it and/or modify
  8. * it under the terms of the GNU Affero General Public License as
  9. * published by the Free Software Foundation, either version 3 of the
  10. * License, or (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU Affero General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Affero General Public License
  18. * along with this program. If not, see <https://www.gnu.org/licenses/>.
  19. *
  20. */
  21. namespace Friendica\Test;
  22. use Friendica\Database\Database;
  23. use Friendica\Test\Util\Database\StaticDatabase;
  24. /**
  25. * Abstract class used by tests that need a database.
  26. */
  27. trait DatabaseTestTrait
  28. {
  29. protected function setUp()
  30. {
  31. StaticDatabase::statConnect($_SERVER);
  32. // Rollbacks every DB usage (in case the test couldn't call tearDown)
  33. StaticDatabase::statRollback();
  34. // Start the first, outer transaction
  35. StaticDatabase::getGlobConnection()->beginTransaction();
  36. parent::setUp();
  37. }
  38. protected function tearDown()
  39. {
  40. // Rollbacks every DB usage so we don't commit anything into the DB
  41. StaticDatabase::statRollback();
  42. parent::tearDown();
  43. }
  44. /**
  45. * Loads a given DB fixture for this DB test
  46. *
  47. * @param string $fixture The path to the fixture
  48. * @param Database $dba The DB connection
  49. *
  50. * @throws \Exception
  51. */
  52. protected function loadFixture(string $fixture, Database $dba)
  53. {
  54. $data = include $fixture;
  55. foreach ($data as $tableName => $rows) {
  56. if (is_numeric($tableName)) {
  57. continue;
  58. }
  59. if (!is_array($rows)) {
  60. $dba->p('TRUNCATE TABLE `' . $tableName . '``');
  61. continue;
  62. }
  63. foreach ($rows as $row) {
  64. $dba->insert($tableName, $row, true);
  65. }
  66. }
  67. }
  68. }