open-consul/ui/packages/consul-ui/tests/helpers/repo.js

109 lines
4.1 KiB
JavaScript

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
);
});
}