114 lines
3.7 KiB
JavaScript
114 lines
3.7 KiB
JavaScript
/**
|
|
* Copyright (c) HashiCorp, Inc.
|
|
* SPDX-License-Identifier: BUSL-1.1
|
|
*/
|
|
|
|
import sinon from 'sinon';
|
|
import { typeOf } from '@ember/utils';
|
|
import { resourceNames } from 'api/handlers/cache-daemon-handler';
|
|
import { singularize } from 'ember-inflector';
|
|
import { underscore } from '@ember/string';
|
|
|
|
/**
|
|
* This test helper can be used to help setup your sinon stubs in your tests.
|
|
* Must be called after setupMirage.
|
|
*
|
|
* ```js
|
|
* import setupStubs from 'api/test-support/handlers/cache-daemon-search';
|
|
*
|
|
* module('Acceptance | my test', function(hooks) {
|
|
* setupApplicationTest(hooks);
|
|
* setupMirage(hooks);
|
|
* setupStubs(hooks);
|
|
*
|
|
* // add your actual tests here
|
|
* });
|
|
* ```
|
|
*
|
|
* This will setup a stub for the IPC service and include helper methods to
|
|
* help setup what the cache daemon search command returns.
|
|
*
|
|
* @module Test Helpers
|
|
* @public
|
|
*/
|
|
export default function setupStubs(hooks) {
|
|
let stubTypes = [];
|
|
|
|
hooks.beforeEach(function () {
|
|
const ipcService = this.owner.lookup('service:ipc');
|
|
this.ipcStub = sinon.stub(ipcService, 'invoke');
|
|
|
|
/**
|
|
* We aim to still use mirage data when stubbing out the cache daemon
|
|
* search. However, the handler expects the search data to be raw JSON from
|
|
* the API which will get serialized. We add back the scope object for it to
|
|
* get correctly normalized and other fields such as "attributes" from the
|
|
* API will already be hoisted and will not be re-normalized from mirage data.
|
|
*
|
|
* This convenience method can take in multiple types,
|
|
* e.g. this.stubCacheDaemonSearch('targets', 'sessions').
|
|
* This will have sinon stub out each call so the first call to search will
|
|
* provide all the models from the first parameter you specify,
|
|
* the second call will provide all the models from the second argument, etc.
|
|
* @param types
|
|
*/
|
|
this.stubCacheDaemonSearch = (...types) => {
|
|
stubTypes = types;
|
|
|
|
types.forEach((type, i) => {
|
|
let models;
|
|
let resourceName;
|
|
if (typeOf(type) === 'object') {
|
|
models = type.func();
|
|
resourceName = type.resource;
|
|
} else {
|
|
models = this.server.schema[type].all().models;
|
|
resourceName = type;
|
|
}
|
|
|
|
this.ipcStub
|
|
.withArgs('searchCacheDaemon')
|
|
.onCall(i)
|
|
.returns({
|
|
[underscore(resourceNames[singularize(resourceName)])]: models.map(
|
|
(model) => {
|
|
// Use internal serializer to serialize the model correctly
|
|
// according to our mirage serializers
|
|
const modelData =
|
|
this.server.serializerOrRegistry.serialize(model);
|
|
|
|
// Serialize the data properly to standard JSON as that is what
|
|
// we're expecting from the cache daemon response
|
|
return JSON.parse(JSON.stringify(modelData));
|
|
},
|
|
),
|
|
});
|
|
});
|
|
};
|
|
});
|
|
|
|
hooks.afterEach(function () {
|
|
// Verify the number of mocks we set up match with how
|
|
// many times the cache daemon was called with
|
|
sinon.assert.callCount(
|
|
this.ipcStub.withArgs('searchCacheDaemon'),
|
|
stubTypes.length,
|
|
);
|
|
|
|
stubTypes.forEach((type, i) => {
|
|
// Assert that the stub was called with the resource we're trying to mock
|
|
// so we don't mistakenly mock the wrong resource
|
|
const ipcCall = this.ipcStub.withArgs('searchCacheDaemon').getCall(i);
|
|
let resource = type;
|
|
if (typeOf(type) === 'object') {
|
|
resource = type.resource;
|
|
}
|
|
|
|
const mappedResourceName = resourceNames[singularize(resource)];
|
|
sinon.assert.match(ipcCall.args[1], { resource: mappedResourceName });
|
|
});
|
|
|
|
sinon.restore();
|
|
});
|
|
}
|