DatabaseTestCase.inc.php

<?php
 
import('lib.pkp.tests.PKPTestCase');
import('lib.pkp.tests.PKPTestHelper');
 
abstract class DatabaseTestCase extends PKPTestCase {
 
   protected function getAffectedTables() {
      return array();
   }
 
   protected function setUp() : void {
      // Switch off xdebug screaming (there are
      // errors in adodb...).
      PKPTestHelper::xdebugScream(false);
 
      // Make sure we have a db connection (some tests
      // might close it and that affects the next ones).
      DBConnection::getInstance()->reconnect();
 
      // Backup affected tables.
      $affectedTables = $this->getAffectedTables();
      if (is_array($affectedTables)) {
         PKPTestHelper::backupTables($affectedTables, $this);
      }
      parent::setUp();
   }
 
   protected function tearDown() : void {
      parent::tearDown();
 
      $affectedTables = $this->getAffectedTables();
      if (is_array($affectedTables)) {
         PKPTestHelper::restoreTables($this->getAffectedTables(), $this);
      } elseif ($affectedTables === PKP_TEST_ENTIRE_DB) {
         PKPTestHelper::restoreDB($this);
      }
 
      // Switch xdebug screaming back on.
      PKPTestHelper::xdebugScream(true);
   }
}