From 0cd03c88834b106f6ae97173278aaf943e0d9d90 Mon Sep 17 00:00:00 2001
From: Bharat Mediratta <bharat@menalto.com>
Date: Mon, 21 Dec 2009 11:30:26 -0800
Subject: [PATCH] Add Database::set_default_instance() for tests

---
 modules/gallery/libraries/MY_Database.php | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/modules/gallery/libraries/MY_Database.php b/modules/gallery/libraries/MY_Database.php
index fd48603970..a8f4bc0860 100644
--- 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;
+  }
 }
\ No newline at end of file