132 lines
4.2 KiB
JavaScript
132 lines
4.2 KiB
JavaScript
import { run } from '@ember/runloop';
|
|
import { test } from 'qunit';
|
|
import { startMirage } from 'nomad-ui/initializers/ember-cli-mirage';
|
|
import moduleForAdapter from '../../helpers/module-for-adapter';
|
|
import { settled } from '@ember/test-helpers';
|
|
|
|
moduleForAdapter('node', 'Unit | Adapter | Node', {
|
|
needs: [
|
|
'adapter:node',
|
|
'model:node-attributes',
|
|
'model:allocation',
|
|
'model:node-driver',
|
|
'model:node-event',
|
|
'model:evaluation',
|
|
'model:resources',
|
|
'model:network',
|
|
'model:job',
|
|
'serializer:application',
|
|
'serializer:node',
|
|
'service:system',
|
|
'service:token',
|
|
'service:config',
|
|
'service:watchList',
|
|
'transform:fragment',
|
|
'transform:fragment-array',
|
|
'transform:array',
|
|
],
|
|
beforeEach() {
|
|
this.server = startMirage();
|
|
this.server.create('node', { id: 'node-1' });
|
|
this.server.create('node', { id: 'node-2' });
|
|
this.server.create('job', { id: 'job-1', createAllocations: false });
|
|
|
|
this.server.create('allocation', { id: 'node-1-1', nodeId: 'node-1' });
|
|
this.server.create('allocation', { id: 'node-1-2', nodeId: 'node-1' });
|
|
this.server.create('allocation', { id: 'node-2-1', nodeId: 'node-2' });
|
|
this.server.create('allocation', { id: 'node-2-2', nodeId: 'node-2' });
|
|
},
|
|
afterEach() {
|
|
this.server.shutdown();
|
|
},
|
|
});
|
|
|
|
test('findHasMany removes old related models from the store', function(assert) {
|
|
let node;
|
|
run(() => {
|
|
// Fetch the model
|
|
this.store.findRecord('node', 'node-1').then(model => {
|
|
node = model;
|
|
|
|
// Fetch the related allocations
|
|
return findHasMany(model, 'allocations').then(allocations => {
|
|
assert.equal(
|
|
allocations.get('length'),
|
|
this.server.db.allocations.where({ nodeId: node.get('id') }).length,
|
|
'Allocations returned from the findHasMany matches the db state'
|
|
);
|
|
});
|
|
});
|
|
});
|
|
|
|
return settled().then(() => {
|
|
server.db.allocations.remove('node-1-1');
|
|
|
|
run(() => {
|
|
// Reload the related allocations now that one was removed server-side
|
|
return findHasMany(node, 'allocations').then(allocations => {
|
|
const dbAllocations = this.server.db.allocations.where({ nodeId: node.get('id') });
|
|
assert.equal(
|
|
allocations.get('length'),
|
|
dbAllocations.length,
|
|
'Allocations returned from the findHasMany matches the db state'
|
|
);
|
|
assert.equal(
|
|
this.store.peekAll('allocation').get('length'),
|
|
dbAllocations.length,
|
|
'Server-side deleted allocation was removed from the store'
|
|
);
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
test('findHasMany does not remove old unrelated models from the store', function(assert) {
|
|
let node;
|
|
|
|
run(() => {
|
|
// Fetch the first node and related allocations
|
|
this.store.findRecord('node', 'node-1').then(model => {
|
|
node = model;
|
|
return findHasMany(model, 'allocations');
|
|
});
|
|
|
|
// Also fetch the second node and related allocations;
|
|
this.store.findRecord('node', 'node-2').then(model => findHasMany(model, 'allocations'));
|
|
});
|
|
|
|
return settled().then(() => {
|
|
assert.deepEqual(
|
|
this.store
|
|
.peekAll('allocation')
|
|
.mapBy('id')
|
|
.sort(),
|
|
['node-1-1', 'node-1-2', 'node-2-1', 'node-2-2'],
|
|
'All allocations for the first and second node are in the store'
|
|
);
|
|
|
|
server.db.allocations.remove('node-1-1');
|
|
|
|
run(() => {
|
|
// Reload the related allocations now that one was removed server-side
|
|
return findHasMany(node, 'allocations').then(() => {
|
|
assert.deepEqual(
|
|
this.store
|
|
.peekAll('allocation')
|
|
.mapBy('id')
|
|
.sort(),
|
|
['node-1-2', 'node-2-1', 'node-2-2'],
|
|
'The deleted allocation is removed from the store and the allocations associated with the other node are untouched'
|
|
);
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
// Using fetchLink on a model's hasMany relationship exercises the adapter's
|
|
// findHasMany method as well normalizing the response and pushing it to the store
|
|
function findHasMany(model, relationshipName) {
|
|
const relationship = model.relationshipFor(relationshipName);
|
|
return model.hasMany(relationship.key).hasManyRelationship.fetchLink();
|
|
}
|