tests: Add tests for recoverCron

This commit is contained in:
Blade Barringer
2016-05-26 22:08:31 -05:00
parent 8292903444
commit 91aba965b0
4 changed files with 124 additions and 25 deletions

View File

@@ -1,4 +1,6 @@
import moment from 'moment';
import Bluebird from 'bluebird';
import { model as User } from '../../models/user';
import common from '../../../../common/';
import { preenUserHistory } from '../../libs/api-v3/preening';
import _ from 'lodash';
@@ -79,6 +81,29 @@ function performSleepTasks (user, tasksByType, now) {
});
}
export async function recoverCron (status, locals) {
let {user} = locals;
await Bluebird.delay(300);
let reloadedUser = await User.findOne({_id: user._id}).exec();
if (!reloadedUser) {
throw new Error(`User ${user._id} not found while recovering.`);
} else if (reloadedUser._cronSignature !== 'NOT_RUNNING') {
status.times++;
if (status.times < 4) {
await recoverCron(status, locals);
} else {
throw new Error(`Impossible to recover from cron for user ${user._id}.`);
}
} else {
locals.user = reloadedUser;
return null;
}
}
// Perform various beginning-of-day reset actions.
export function cron (options = {}) {
let {user, tasksByType, analytics, now = new Date(), daysMissed, timezoneOffsetFromUserPrefs} = options;