krikkert/habitica
1
0
Fork 0
mirror of https://github.com/HabitRPG/habitica.git synced 2025-12-17 22:57:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update generated document with update object

Browse Source
This commit is contained in:
Blade Barringer
2015-10-18 07:00:48 -05:00
parent 9b2da51991
commit 4fa82ba196
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/helpers/api.helper.js
View File

@@ -1,4 +1,7 @@
import {isEmpty} from 'lodash'; import {
assign,
isEmpty,
} from 'lodash';
import {MongoClient as mongo} from 'mongodb'; import {MongoClient as mongo} from 'mongodb';
import {v4 as generateUUID} from 'uuid'; import {v4 as generateUUID} from 'uuid';
import superagent from 'superagent'; import superagent from 'superagent';
@@ -28,7 +31,7 @@ export function generateUser(update={}) {
password: password, password: password,
confirmPassword: password, confirmPassword: password,
}).then((user) => { }).then((user) => {
_updateDocument('users', user._id, update, () => { _updateDocument('users', user, update, () => {
resolve(user); resolve(user);
}); });
}); });
@@ -40,7 +43,7 @@ export function generateGroup(leader, update={}) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
request('/groups').then((group) => { request('/groups').then((group) => {
_updateDocument('groups', group._id, update, () => { _updateDocument('groups', group, update, () => {
resolve(group); resolve(group);
}); });
}); });
@@ -77,7 +80,7 @@ function _requestMaker(user, method) {
} }
} }
function _updateDocument(collectionName, uuid, update, cb) { function _updateDocument(collectionName, doc, update, cb) {
if (isEmpty(update)) { return cb(); } if (isEmpty(update)) { return cb(); }
mongo.connect('mongodb://localhost/habitrpg_test', (err, db) => { mongo.connect('mongodb://localhost/habitrpg_test', (err, db) => {
@@ -85,8 +88,9 @@ function _updateDocument(collectionName, uuid, update, cb) {
let collection = db.collection(collectionName); let collection = db.collection(collectionName);
collection.update({ _id: uuid }, { $set: update }, (err, result) => { collection.update({ _id: doc._id }, { $set: update }, (err, result) => {
if (err) throw `Error updating ${collectionName}: ${err}`; if (err) throw `Error updating ${collectionName}: ${err}`;
assign(doc, update);
cb(); cb();
}); });
}); });