From eab15a3dd07f69b89b59dd6a648716f9ea59961b Mon Sep 17 00:00:00 2001
From: Blade Barringer <blade@crookedneighbor.com>
Date: Wed, 30 Dec 2015 07:57:41 -0600
Subject: [PATCH] tests(api): Port over new integration helper from develop

---
 test/helpers/api-integration.helper.js | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/test/helpers/api-integration.helper.js b/test/helpers/api-integration.helper.js
index 973414a3a8..dd0d5baf0b 100644
--- a/test/helpers/api-integration.helper.js
+++ b/test/helpers/api-integration.helper.js
@@ -1,9 +1,10 @@
 /* eslint-disable no-use-before-define */
 
 import {
-  set,
+  assign,
   each,
   isEmpty,
+  set,
   times,
 } from 'lodash';
 import { MongoClient as mongo } from 'mongodb';
@@ -23,6 +24,17 @@ const ROUTES = {
   },
 };
 
+class ApiUser {
+  constructor (options) {
+    assign(this, options);
+
+    this.get = _requestMaker(this, 'get');
+    this.post = _requestMaker(this, 'post');
+    this.put = _requestMaker(this, 'put');
+    this.del = _requestMaker(this, 'del');
+  }
+}
+
 // Sets up an abject that can make all REST requests
 // If a user is passed in, the uuid and api token of
 // the user are used to make the requests
@@ -94,7 +106,9 @@ export function generateUser (update = {}) {
       confirmPassword: password,
     }).then((user) => {
       _updateDocument('users', user, update, () => {
-        resolve(user);
+        let apiUser = new ApiUser(user);
+
+        resolve(apiUser);
       });
     }).catch(reject);
   });
@@ -277,7 +291,6 @@ function _updateDocument (collectionName, doc, update, cb) {
     return cb();
   }
 
-  // TODO use config for db url?
   mongo.connect('mongodb://localhost/habitrpg_test', (connectErr, db) => {
     if (connectErr) throw new Error(`Error connecting to database when updating ${collectionName} collection: ${connectErr}`);