open-nomad/ui/mirage/factories/task-state.js

import { Factory } from 'ember-cli-mirage';
import faker from 'nomad-ui/mirage/faker';

const TASK_STATUSES = ['pending', 'running', 'finished', 'failed'];
const REF_TIME = new Date();

export default Factory.extend({
  name: () => '!!!this should be set by the allocation that owns this task state!!!',
  state: () => faker.helpers.randomize(TASK_STATUSES),
  kind: null,
  startedAt: () => faker.date.past(2 / 365, REF_TIME),
  finishedAt() {
    if (['pending', 'running'].includes(this.state)) {
      return '0001-01-01T00:00:00Z';
    }
    return new Date(this.startedAt + Math.random(1000 * 60 * 3) + 50);
  },

  afterCreate(state, server) {
    const props = [
      'task-event',
      faker.random.number({ min: 1, max: 3 }),
      {
        taskStateId: state.id,
      },
    ].compact();

    const events = server.createList(...props);

    state.update({
      eventIds: events.mapBy('id'),
    });
  },
});