78 lines
2.4 KiB
JavaScript
78 lines
2.4 KiB
JavaScript
import { module, test } from 'qunit';
|
|
import { setupTest } from 'ember-qunit';
|
|
import { startMirage } from 'nomad-ui/initializers/ember-cli-mirage';
|
|
|
|
module('Unit | Adapter | Allocation', function(hooks) {
|
|
setupTest(hooks);
|
|
|
|
hooks.beforeEach(async function() {
|
|
this.store = this.owner.lookup('service:store');
|
|
this.subject = () => this.store.adapterFor('allocation');
|
|
|
|
this.server = startMirage();
|
|
|
|
this.server.create('namespace');
|
|
this.server.create('node');
|
|
this.server.create('job', { createAllocations: false });
|
|
this.server.create('allocation', { id: 'alloc-1' });
|
|
});
|
|
|
|
hooks.afterEach(function() {
|
|
this.server.shutdown();
|
|
});
|
|
|
|
test('`stop` makes the correct API call', async function(assert) {
|
|
const { pretender } = this.server;
|
|
const allocationId = 'alloc-1';
|
|
|
|
const allocation = await this.store.findRecord('allocation', allocationId);
|
|
pretender.handledRequests.length = 0;
|
|
|
|
await this.subject().stop(allocation);
|
|
const req = pretender.handledRequests[0];
|
|
assert.equal(
|
|
`${req.method} ${req.url}`,
|
|
`POST /v1/allocation/${allocationId}/stop`,
|
|
`POST /v1/allocation/${allocationId}/stop`
|
|
);
|
|
});
|
|
|
|
test('`restart` makes the correct API call', async function(assert) {
|
|
const { pretender } = this.server;
|
|
const allocationId = 'alloc-1';
|
|
|
|
const allocation = await this.store.findRecord('allocation', allocationId);
|
|
pretender.handledRequests.length = 0;
|
|
|
|
await this.subject().restart(allocation);
|
|
const req = pretender.handledRequests[0];
|
|
assert.equal(
|
|
`${req.method} ${req.url}`,
|
|
`PUT /v1/client/allocation/${allocationId}/restart`,
|
|
`PUT /v1/client/allocation/${allocationId}/restart`
|
|
);
|
|
});
|
|
|
|
test('`restart` takes an optional task name and makes the correct API call', async function(assert) {
|
|
const { pretender } = this.server;
|
|
const allocationId = 'alloc-1';
|
|
const taskName = 'task-name';
|
|
|
|
const allocation = await this.store.findRecord('allocation', allocationId);
|
|
pretender.handledRequests.length = 0;
|
|
|
|
await this.subject().restart(allocation, taskName);
|
|
const req = pretender.handledRequests[0];
|
|
assert.equal(
|
|
`${req.method} ${req.url}`,
|
|
`PUT /v1/client/allocation/${allocationId}/restart`,
|
|
`PUT /v1/client/allocation/${allocationId}/restart`
|
|
);
|
|
assert.deepEqual(
|
|
JSON.parse(req.requestBody),
|
|
{ TaskName: taskName },
|
|
'Request body is correct'
|
|
);
|
|
});
|
|
});
|