habitica/test/helpers/api-unit.helper.js

import '../../website/server/libs/i18n';
import mongoose from 'mongoose';
import defaultsDeep from 'lodash/defaultsDeep';
import moment from 'moment';
import { model as User } from '../../website/server/models/user';
import { model as Group } from '../../website/server/models/group';
import { model as Challenge } from '../../website/server/models/challenge';
import mongo from './mongo'; // eslint-disable-line
import i18n from '../../website/common/script/i18n';
import * as Tasks from '../../website/server/models/task';

export { translationCheck } from './translate';

afterEach(() => {
  sandbox.restore();
  return mongoose.connection.dropDatabase();
});

export { sleep } from './sleep';

export function generateUser (options = {}) {
  return new User(options);
}

export function generateGroup (options = {}) {
  return new Group(options);
}

export function generateChallenge (options = {}) {
  return new Challenge(options);
}

export function generateRes (options = {}) {
  const defaultRes = {
    json: sandbox.stub(),
    locals: {
      user: generateUser(options.localsUser),
      group: generateGroup(options.localsGroup),
    },
    redirect: sandbox.stub(),
    render: sandbox.stub(),
    send: sandbox.stub(),
    sendFile: sandbox.stub(),
    sendStatus: sandbox.stub().returnsThis(),
    set: sandbox.stub(),
    status: sandbox.stub().returnsThis(),
    t (string) {
      return i18n.t(string);
    },
  };

  return defaultsDeep(options, defaultRes);
}

export function generateReq (options = {}) {
  const defaultReq = {
    body: {},
    query: {},
    headers: {},
    header (header) {
      return this.headers[header];
    },
    listeners: {},
    session: {},
    on (key, func) {
      if (!this.listeners[key]) {
        this.listeners[key] = [];
      }
      this.listeners[key].push(func);
    },
    end () {
      this.listeners.close.forEach(func => func());
    },
  };

  const req = defaultsDeep(options, defaultReq);

  return req;
}

export function generateNext (func) {
  return func || sandbox.stub();
}

export function generateHistory (days) {
  const history = [];
  const now = Number(moment().toDate());

  while (days > 0) {
    history.push({
      value: days,
      date: Number(moment(now).subtract(days, 'days').toDate()),
    });
    days -= 1; // eslint-disable-line no-param-reassign
  }

  return history;
}

export function generateTodo (user) {
  const todo = {
    text: 'test todo',
    type: 'todo',
    value: 0,
    completed: false,
  };

  const task = new Tasks.todo(Tasks.Task.sanitize(todo)); // eslint-disable-line new-cap
  task.userId = user._id;

  return task;
}

export function generateDaily (user) {
  const daily = {
    text: 'test daily',
    type: 'daily',
    value: 0,
    completed: false,
  };

  const task = new Tasks.daily(Tasks.Task.sanitize(daily)); // eslint-disable-line new-cap
  task.userId = user._id;

  return task;
}

export function defer () {
  let resolve;
  let reject;

  const promise = new Promise((resolveParam, rejectParam) => {
    resolve = resolveParam;
    reject = rejectParam;
  });

  return {
    resolve,
    reject,
    promise,
  };
}