Allow unsetting the db (ORM::set_db(null)) to make the test work again

1 files changed, 4 insertions, 2 deletions

diff --git a/idiorm.php b/idiorm.php
index 7ac1652..143ecb7 100644
--- a/idiorm.php
+++ b/idiorm.php
@@ -279,8 +279,10 @@
         public static function set_db($db, $connection_name = self::DEFAULT_CONNECTION) {
             self::_setup_db_config($connection_name);
             self::$_db[$connection_name] = $db;
-            self::_setup_identifier_quote_character($connection_name);
-            self::_setup_limit_clause_style($connection_name);
+            if(!is_null(self::$_db[$connection_name])) {
+                self::_setup_identifier_quote_character($connection_name);
+                self::_setup_limit_clause_style($connection_name);
+            }
         }
 
         /**