78 lines
2.1 KiB
JavaScript
78 lines
2.1 KiB
JavaScript
import { Factory, faker } from 'ember-cli-mirage';
|
|
|
|
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.random.arrayElement(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,
|
|
});
|
|
});
|
|
}
|
|
},
|
|
});
|