Allow test to remove data from the database

This new removeDBData() is run after the tests result. It is similar
to the addDBData() method.

diff --git a/tests/phpunit/MediaWikiTestCase.php b/tests/phpunit/MediaWikiTestCase.php
index 3363e3f..1051844 100644 (file)
--- a/tests/phpunit/MediaWikiTestCase.php
+++ b/tests/phpunit/MediaWikiTestCase.php
@@ -59,6 +59,9 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
                        $this->addDBData();
                        
                        parent::run( $result );
+
+                       $this->removeDBData();
+
                } else {
                        parent::run( $result );
                }
@@ -84,6 +87,12 @@ abstract class MediaWikiTestCase extends PHPUnit_Framework_TestCase {
         * implement this method and do so
         */
        function addDBData() {}
+
+       /**
+        * Stub. If a test needs to remove data from the database. Called after
+        * test run.
+        */
+       function removeDBData() {}
        
        private function addCoreDBData() {