Add Database::set_default_instance() for tests

1 files changed, 7 insertions, 0 deletions

diff --git a/modules/gallery/libraries/MY_Database.php b/modules/gallery/libraries/MY_Database.php
index fd486039..a8f4bc08 100644
--- a/modules/gallery/libraries/MY_Database.php
+++ b/modules/gallery/libraries/MY_Database.php
@@ -57,4 +57,11 @@ abstract class Database extends Database_Core {
 
     return strtr($sql, $this->_table_names);
   }
+
+  /**
+   * This is used by the unit test code to switch the active database connection.
+   */
+  static function set_default_instance($db) {
+    self::$instances["default"] = $db;
+  }
 }
\ No newline at end of file