Support custom attributes (#54)

* Support custom attributes

* remove console

dist/queries.js

@@ -36,7 +36,7 @@ var Queries = function () {
     key: 'track',
     value: function track(query, resolve, _reject) {
       if (this.tracker.tracking) {
-        query.mock = _extends({}, _lodash2.default.pick(query, ['method', 'sql', 'bindings', 'returning', 'transacting']), {
+        query.mock = _extends({}, _lodash2.default.get(query, 'mock', {}), _lodash2.default.pick(query, ['method', 'sql', 'bindings', 'returning', 'transacting']), {
           response: function response(_response) {
             var options = arguments.length <= 1 || arguments[1] === undefined ? {} : arguments[1];
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "mock-knex",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "description": "a knex mock adapter for simulating a db during testing",
   "main": "index.js",
   "scripts": {

src/queries.js

@@ -16,6 +16,7 @@ export default class Queries {
   track(query, resolve, reject) {
     if (this.tracker.tracking) {
       query.mock = {
+        ..._.get(query, 'mock', {}),
         ..._.pick(query, [
           'method',
           'sql',

test/common/tracker.js

@@ -79,6 +79,23 @@ module.exports = (db) => {
       db('users').select().then(_.noop);
     });
 
+    it('should allow passing custom attributes to query',
+    (done) => {
+      tracker.install();
+
+      tracker.once('query', (query) => {
+        expect(query).to.have.property('custom');
+        expect(query.custom).to.be.a('boolean');
+        done();
+      });
+
+      tracker.queries.track({
+        mock : {
+          custom : true,
+        },
+      })
+    });
+
     it('should return a query object with a method property',
     (done) => {
       tracker.once('query', (query) => {