From e82085ac1bd49754150036eb2f48bdd1a9ee1c0c Mon Sep 17 00:00:00 2001
From: Blade Barringer <blade@crookedneighbor.com>
Date: Sun, 13 Dec 2015 16:22:58 -0600
Subject: [PATCH] tests(helpers): Add common helpers for generating tasks

---
 test/helpers/common.helper.js | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/test/helpers/common.helper.js b/test/helpers/common.helper.js
index cb52d7e767..96064b8142 100644
--- a/test/helpers/common.helper.js
+++ b/test/helpers/common.helper.js
@@ -2,7 +2,12 @@ import mongoose from 'mongoose';
 
 import { wrap as wrapUser } from '../../common/script/index';
 import { model as User } from '../../website/src/models/user';
-import { TodoSchema } from '../../website/src/models/task';
+import {
+  DailySchema,
+  HabitSchema,
+  RewardSchema,
+  TodoSchema,
+} from '../../website/src/models/task';
 
 export function generateUser (options = {}) {
   let user = new User(options).toObject();
@@ -12,6 +17,24 @@ export function generateUser (options = {}) {
   return user;
 }
 
+export function generateDaily (options = {}) {
+  let Daily = mongoose.model('Daily', DailySchema);
+
+  return new Daily(options).toObject();
+}
+
+export function generateHabit (options = {}) {
+  let Habit = mongoose.model('Habit', HabitSchema);
+
+  return new Habit(options).toObject();
+}
+
+export function generateReward (options = {}) {
+  let Reward = mongoose.model('Reward', RewardSchema);
+
+  return new Reward(options).toObject();
+}
+
 export function generateTodo (options = {}) {
   let Todo = mongoose.model('Todo', TodoSchema);