2019-09-26 18:47:07 +00:00
|
|
|
import { Factory } from 'ember-cli-mirage';
|
2019-10-03 14:13:08 +00:00
|
|
|
import faker from 'nomad-ui/mirage/faker';
|
2017-09-19 14:47:10 +00:00
|
|
|
import { generateResources } from '../common';
|
|
|
|
|
2018-05-12 00:30:15 +00:00
|
|
|
const DRIVERS = ['docker', 'java', 'rkt', 'qemu', 'exec', 'raw_exec'];
|
|
|
|
|
2017-09-19 14:47:10 +00:00
|
|
|
export default Factory.extend({
|
2020-10-29 12:46:42 +00:00
|
|
|
createRecommendations: false,
|
|
|
|
|
2017-09-19 14:47:10 +00:00
|
|
|
// Hidden property used to compute the Summary hash
|
|
|
|
groupNames: [],
|
|
|
|
|
2020-02-12 19:49:28 +00:00
|
|
|
// Set in the TaskGroup factory
|
|
|
|
volumeMounts: [],
|
|
|
|
|
2017-09-19 14:47:10 +00:00
|
|
|
JobID: '',
|
|
|
|
|
2018-06-13 22:04:31 +00:00
|
|
|
name: id => `task-${faker.hacker.noun().dasherize()}-${id}`,
|
2019-09-30 14:44:22 +00:00
|
|
|
driver: () => faker.helpers.randomize(DRIVERS),
|
2017-09-19 14:47:10 +00:00
|
|
|
|
2020-10-12 22:26:54 +00:00
|
|
|
originalResources: generateResources,
|
|
|
|
resources: function() {
|
|
|
|
// Generate resources the usual way, but transform to the old
|
|
|
|
// shape because that's what the job spec uses.
|
|
|
|
const resources = this.originalResources;
|
|
|
|
return {
|
|
|
|
CPU: resources.Cpu.CpuShares,
|
|
|
|
MemoryMB: resources.Memory.MemoryMB,
|
2021-04-28 20:38:14 +00:00
|
|
|
MemoryMaxMB: resources.Memory.MemoryMaxMB,
|
2020-10-12 22:26:54 +00:00
|
|
|
DiskMB: resources.Disk.DiskMB,
|
|
|
|
};
|
|
|
|
},
|
2020-04-30 13:15:19 +00:00
|
|
|
|
|
|
|
Lifecycle: i => {
|
2020-09-01 16:49:36 +00:00
|
|
|
const cycle = i % 6;
|
2020-04-30 13:15:19 +00:00
|
|
|
|
|
|
|
if (cycle === 0) {
|
|
|
|
return null;
|
|
|
|
} else if (cycle === 1) {
|
|
|
|
return { Hook: 'prestart', Sidecar: false };
|
2020-08-25 17:17:18 +00:00
|
|
|
} else if (cycle === 2) {
|
2020-04-30 13:15:19 +00:00
|
|
|
return { Hook: 'prestart', Sidecar: true };
|
2020-08-25 17:17:18 +00:00
|
|
|
} else if (cycle === 3) {
|
|
|
|
return { Hook: 'poststart', Sidecar: false };
|
|
|
|
} else if (cycle === 4) {
|
|
|
|
return { Hook: 'poststart', Sidecar: true };
|
2020-09-01 16:49:36 +00:00
|
|
|
} else if (cycle === 5) {
|
|
|
|
return { Hook: 'poststop' };
|
2020-04-30 13:15:19 +00:00
|
|
|
}
|
|
|
|
},
|
2020-10-29 12:46:42 +00:00
|
|
|
|
|
|
|
afterCreate(task, server) {
|
|
|
|
if (task.createRecommendations) {
|
|
|
|
const recommendations = [];
|
|
|
|
|
|
|
|
if (faker.random.number(10) >= 1) {
|
|
|
|
recommendations.push(server.create('recommendation', { task, resource: 'CPU' }));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (faker.random.number(10) >= 1) {
|
|
|
|
recommendations.push(server.create('recommendation', { task, resource: 'MemoryMB' }));
|
|
|
|
}
|
|
|
|
|
|
|
|
task.save({ recommendationIds: recommendations.mapBy('id') });
|
|
|
|
}
|
|
|
|
},
|
2017-09-19 14:47:10 +00:00
|
|
|
});
|