feat(getDbInstance): Added getDbInstance method

Signed-off-by: BorntraegerMarc <[email protected]>

Author: BorntraegerMarc

index.js

@@ -213,6 +213,16 @@ var MongodbDriver = Base.extend({
     return this._run('insert', this.internals.seedTable, {name: name, run_on: new Date()})
       .nodeify(callback);
   },
+  
+  /**
+   * Returns the DB instance so custom updates can be made
+   *
+   * @param callback
+   */
+  getDbInstance: function (callback) {
+    return this._run('getDbInstance', null, {run_on: new Date()})
+      .nodeify(callback);
+  },
 
   /**
    * Runs a query
@@ -345,6 +355,9 @@ var MongodbDriver = Base.extend({
           case 'updateMany':
             db.collection(collection)[command](options.query, options.update, options.options, callbackFunction);
             break;
+          case 'getDbInstance':
+            callbackFunction(null, db);
+            break;
           default:
             db[command](collection, callbackFunction);
             break;

test/mongodb_test.js

@@ -468,4 +468,18 @@ vows
       }
     }
   })
+  .addBatch({
+    getDbInstance: {
+      topic: function() {
+        db.getDbInstance(this.callback);
+      },
+
+      teardown: function() {
+      },
+
+      "has database reference": function(err, db) {
+        assert.isNotNull(db);
+      }
+    }
+  })
   .export(module);