import { get as httpGet } from 'consul-ui/tests/helpers/api'; import { get, set } from '@ember/object'; import measure from 'consul-ui/tests/helpers/measure'; /** Stub an ember-data adapter response using the private method * * Allows you to easily specify a HTTP response for the Adapter. The stub only works * during the 'lifetime' of `cb` and is reset to normal unstubbed functionality afterwards. * * Please Note: This overwrites a private ember-data Adapter method, please understand * the consequences of doing this if you are using it * * @param {function} cb - The callback, or test case, to run using the stubbed response * @param {object} payload - The payload to use as the response * @param {DS.Adapter} adapter - An instance of an ember-data Adapter */ const stubAdapterResponse = function(cb, payload, adapter) { const payloadClone = JSON.parse(JSON.stringify(payload)); const client = get(adapter, 'client'); set(adapter, 'client', { request: function(cb) { return cb(function() { const params = client.requestParams(...arguments); payload.headers['X-Consul-Namespace'] = params.data.ns || 'default'; payload.headers['X-Consul-Partition'] = params.data.partition || 'default'; return Promise.resolve(function(cb) { return cb(payload.headers, payloadClone.payload); }); }); }, }); return cb(payload.payload).then(function(result) { set(adapter, 'client', client); return result; }); }; /** `repo` a helper function to faciliate easy integration testing of ember-data Service 'repo' layers * * Test performance is also measured using `consul-ui/tests/helpers/measure` and therefore results * can optionally be sent to a centralized metrics collection stack * * @param {string} name - The name of your repo Service (only used for meta purposes) * @param {string} payload - The method you are testing (only used for meta purposes) * @param {Service} service - An instance of an ember-data based repo Service * @param {function} stub - A function that receives a `stub` function allowing you to stub * an API endpoint with a set of cookies/env vars used by the double * @param {function} test - Your test case. This function receives an instance of the Service provided * above as a first and only argument, it should return the result of your test * @param {function} assert - Your assertion. This receives the result of the previous function as the first * argument and a function to that receives the stubbed payload giving you an * opportunity to mutate it before returning for use in your assertion */ export default function(name, method, service, stub, test, assert) { const adapter = get(service, 'store').adapterFor(name.toLowerCase()); let tags = {}; const requestHeaders = function(url, cookies = {}) { const key = Object.keys(cookies).find(function(item) { return item.indexOf('COUNT') !== -1; }); tags = { count: typeof key !== 'undefined' ? parseInt(cookies[key]) : 1, }; return httpGet(url, { headers: { cookie: cookies, }, }).then(function(payload) { return { headers: {}, payload: payload, }; }); }; const parseResponse = function(response) { let actual; if (typeof response.toArray === 'function') { actual = response.toArray().map(function(item) { return get(item, 'data'); }); } else { if (typeof response.get === 'function') { actual = get(response, 'data'); } else { actual = response; } } return actual; }; return stub(requestHeaders).then(function(payload) { return stubAdapterResponse( function(payload) { return measure( function() { return test(service); }, `${name}Service.${method}`, tags ).then(function(response) { assert(parseResponse(response), function(cb) { return cb(payload); }); }); }, payload, adapter ); }); }