open-nomad/ui/mirage/factories/task-group.js
Buck Doyle 5da134d074
UI: Change Mirage data to be stable in development (#6389)
This sets a default-but-query-configurable Faker seed in development,
via faker-seed. It also changes uses of Math.random to use Faker’s
randomness so auto-generated data remains stable in development.
2019-10-03 09:13:08 -05:00

79 lines
2.1 KiB
JavaScript

import { Factory } from 'ember-cli-mirage';
import faker from 'nomad-ui/mirage/faker';
const DISK_RESERVATIONS = [200, 500, 1000, 2000, 5000, 10000, 100000];
export default Factory.extend({
name: id => `${faker.hacker.noun().dasherize()}-g-${id}`,
count: () => faker.random.number({ min: 1, max: 2 }),
ephemeralDisk: () => ({
Sticky: faker.random.boolean(),
SizeMB: faker.helpers.randomize(DISK_RESERVATIONS),
Migrate: faker.random.boolean(),
}),
// Directive used to control whether or not allocations are automatically
// created.
createAllocations: true,
// Directived used to control whether or not the allocation should fail
// and reschedule, creating reschedule events.
withRescheduling: false,
// Directive used to control whether the task group should have services.
withServices: false,
// When true, only creates allocations
shallow: false,
afterCreate(group, server) {
let taskIds = [];
if (!group.shallow) {
const tasks = server.createList('task', group.count, {
taskGroup: group,
});
taskIds = tasks.mapBy('id');
}
group.update({
taskIds: taskIds,
task_ids: taskIds,
});
if (group.createAllocations) {
Array(group.count)
.fill(null)
.forEach((_, i) => {
const props = {
jobId: group.job.id,
namespace: group.job.namespace,
taskGroup: group.name,
name: `${group.name}.[${i}]`,
rescheduleSuccess: group.withRescheduling ? faker.random.boolean() : null,
rescheduleAttempts: group.withRescheduling
? faker.random.number({ min: 1, max: 5 })
: 0,
};
if (group.withRescheduling) {
server.create('allocation', 'rescheduled', props);
} else {
server.create('allocation', props);
}
});
}
if (group.withServices) {
Array(faker.random.number({ min: 1, max: 3 }))
.fill(null)
.forEach(() => {
server.create('service', {
task_group: group,
});
});
}
},
});