Add Database::set_default_instance() for tests

gallery · Dec 21, 2009 · 0cd03c8 · 0cd03c8
1 parent 287f79d
commit 0cd03c8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/modules/gallery/libraries/MY_Database.php b/modules/gallery/libraries/MY_Database.php
@@ -57,4 +57,11 @@ public function add_table_prefixes($sql) {
 
     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;
+  }
 }