Skip to content

Commit

Permalink
Migrate to class
Browse files Browse the repository at this point in the history
  • Loading branch information
@marcbachmann
marcbachmann committed Oct 9, 2024
1 parent cf3d881 commit c958a54
Showing 1 changed file with 57 additions and 65 deletions.
122 changes: 57 additions & 65 deletions src/index.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,17 +15,11 @@ const VERBS = [
"unlink",
];

function adapter() {
return (config) => {
return handleRequest(this, config);
};
}

function getVerbArray() {
const arr = [];
VERBS.forEach(function (verb) {
Object.defineProperty(arr, verb, {
get: function () {
get () {
return arr.filter(function (h) {
return !h.method || h.method === verb;
});
Expand All @@ -35,49 +29,52 @@ function getVerbArray() {
return arr;
}

function MockAdapter(axiosInstance, options = {}) {
this.reset();

if (axiosInstance) {
this.axiosInstance = axiosInstance;
// Clone the axios instance to remove interceptors
// this is used for the passThrough mode with axios > 1.2
this.axiosInstanceWithoutInterceptors = axiosInstance.create
? axiosInstance.create()
: undefined;

this.originalAdapter = axiosInstance.defaults.adapter;
this.delayResponse = options.delayResponse > 0 ? options.delayResponse : null;
this.onNoMatch = options.onNoMatch || null;
axiosInstance.defaults.adapter = this.adapter.call(this);
} else {
throw new Error("Please provide an instance of axios to mock");
class AxiosMockAdapter {
constructor (axiosInstance, options = {}) {
this.reset();

if (axiosInstance) {
this.axiosInstance = axiosInstance;
// Clone the axios instance to remove interceptors
// this is used for the passThrough mode with axios > 1.2
this.axiosInstanceWithoutInterceptors = axiosInstance.create
? axiosInstance.create()
: undefined;

this.originalAdapter = axiosInstance.defaults.adapter;
this.delayResponse = options.delayResponse > 0 ? options.delayResponse : null;
this.onNoMatch = options.onNoMatch || null;
axiosInstance.defaults.adapter = this.adapter();
} else {
throw new Error("Please provide an instance of axios to mock");
}
}
}

MockAdapter.prototype.adapter = adapter;
adapter () {
return (config) => handleRequest(this, config);
}

MockAdapter.prototype.restore = function restore() {
if (this.axiosInstance) {
restore () {
if (!this.axiosInstance) return;
this.axiosInstance.defaults.adapter = this.originalAdapter;
this.axiosInstance = undefined;
}
};

MockAdapter.prototype.reset = function reset() {
this.resetHandlers();
this.resetHistory();
};
reset () {
this.resetHandlers();
this.resetHistory();
}

MockAdapter.prototype.resetHandlers = function resetHandlers() {
if (this.handlers) this.handlers.length = 0;
else this.handlers = getVerbArray();
};
resetHandlers () {
if (this.handlers) this.handlers.length = 0;
else this.handlers = getVerbArray();
}

MockAdapter.prototype.resetHistory = function resetHistory() {
if (this.history) this.history.length = 0;
else this.history = getVerbArray();
};
resetHistory () {
if (this.history) this.history.length = 0;
else this.history = getVerbArray();
}
}

const methodsWithConfigsAsSecondArg = ["any", "get", "delete", "head", "options"];
function convertDataAndConfigToConfig (method, data, config) {
Expand Down Expand Up @@ -111,14 +108,14 @@ function toMethodName (method) {
}

VERBS.concat("any").forEach(function (method) {
MockAdapter.prototype[toMethodName(method)] = function (matcher, data, config) {
AxiosMockAdapter.prototype[toMethodName(method)] = function (matcher, data, config) {
const self = this;
let delay;
matcher = matcher === undefined ? /.*/ : matcher;

const paramsAndBody = convertDataAndConfigToConfig(method, data, config);

function reply(code, response, headers) {
function reply (code, response, headers) {
const handler = {
url: matcher,
method: method === "any" ? undefined : method,
Expand All @@ -137,14 +134,14 @@ VERBS.concat("any").forEach(function (method) {
return self;
}

function withDelayInMs(_delay) {
function withDelayInMs (_delay) {
delay = _delay;
const respond = requestApi.reply.bind(requestApi);
Object.assign(respond, requestApi);
return respond;
}

function replyOnce(code, response, headers) {
function replyOnce (code, response, headers) {
const handler = {
url: matcher,
method: method === "any" ? undefined : method,
Expand All @@ -164,13 +161,10 @@ VERBS.concat("any").forEach(function (method) {
}

const requestApi = {
reply: reply,

replyOnce: replyOnce,

withDelayInMs: withDelayInMs,

passThrough: function passThrough() {
reply,
replyOnce,
withDelayInMs,
passThrough () {
const handler = {
passThrough: true,
method: method === "any" ? undefined : method,
Expand All @@ -182,8 +176,7 @@ VERBS.concat("any").forEach(function (method) {
addHandler(method, self.handlers, handler);
return self;
},

abortRequest: function () {
abortRequest () {
return reply(async function (config) {
throw utils.createAxiosError(
"Request aborted",
Expand All @@ -193,8 +186,7 @@ VERBS.concat("any").forEach(function (method) {
);
});
},

abortRequestOnce: function () {
abortRequestOnce () {
return replyOnce(async function (config) {
throw utils.createAxiosError(
"Request aborted",
Expand All @@ -205,19 +197,19 @@ VERBS.concat("any").forEach(function (method) {
});
},

networkError: function () {
networkError () {
return reply(async function (config) {
throw utils.createAxiosError("Network Error", config);
});
},

networkErrorOnce: function () {
networkErrorOnce () {
return replyOnce(async function (config) {
throw utils.createAxiosError("Network Error", config);
});
},

timeout: function () {
timeout () {
return reply(async function (config) {
throw utils.createAxiosError(
config.timeoutErrorMessage ||
Expand All @@ -231,7 +223,7 @@ VERBS.concat("any").forEach(function (method) {
});
},

timeoutOnce: function () {
timeoutOnce () {
return replyOnce(async function (config) {
throw utils.createAxiosError(
config.timeoutErrorMessage ||
Expand All @@ -250,7 +242,7 @@ VERBS.concat("any").forEach(function (method) {
};
});

function findInHandlers(method, handlers, handler) {
function findInHandlers (handlers, handler) {
let index = -1;
for (let i = 0; i < handlers.length; i += 1) {
const item = handlers[i];
Expand All @@ -273,11 +265,11 @@ function findInHandlers(method, handlers, handler) {
return index;
}

function addHandler(method, handlers, handler) {
function addHandler (method, handlers, handler) {
if (method === "any") {
handlers.push(handler);
} else {
const indexOfExistingHandler = findInHandlers(method, handlers, handler);
const indexOfExistingHandler = findInHandlers(handlers, handler);
// handler.replyOnce indicates that a handler only runs once.
// It's supported to register muliple ones like that without
// overwriting the previous one.
Expand All @@ -289,5 +281,5 @@ function addHandler(method, handlers, handler) {
}
}

module.exports = MockAdapter;
module.exports.default = MockAdapter;
module.exports = AxiosMockAdapter;
module.exports.default = AxiosMockAdapter;

0 comments on commit c958a54

Please sign in to comment.