52 lines
2.0 KiB
JavaScript
52 lines
2.0 KiB
JavaScript
import Stream from "stream";
|
|
export function mockRequest(requestUrl, requestHeaders, requestMethod, requestConnection) {
|
|
const resBuffers = [];
|
|
const mockRes = new Stream.Writable();
|
|
const isStreamFinished = new Promise(function(resolve, reject) {
|
|
mockRes.on("finish", ()=>resolve(true));
|
|
mockRes.on("end", ()=>resolve(true));
|
|
mockRes.on("error", (err)=>reject(err));
|
|
});
|
|
mockRes.write = (chunk)=>{
|
|
resBuffers.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
|
|
};
|
|
mockRes._write = (chunk, _encoding, callback)=>{
|
|
mockRes.write(chunk);
|
|
// According to Node.js documentation, the callback MUST be invoked to signal that
|
|
// the write completed successfully. If this callback is not invoked, the 'finish' event
|
|
// will not be emitted.
|
|
// https://nodejs.org/docs/latest-v16.x/api/stream.html#writable_writechunk-encoding-callback
|
|
callback();
|
|
};
|
|
const mockHeaders = {};
|
|
mockRes.writeHead = (_status, _headers)=>Object.assign(mockHeaders, _headers);
|
|
mockRes.getHeader = (name)=>mockHeaders[name.toLowerCase()];
|
|
mockRes.getHeaders = ()=>mockHeaders;
|
|
mockRes.getHeaderNames = ()=>Object.keys(mockHeaders);
|
|
mockRes.setHeader = (name, value)=>mockHeaders[name.toLowerCase()] = value;
|
|
mockRes.removeHeader = (name)=>{
|
|
delete mockHeaders[name.toLowerCase()];
|
|
};
|
|
mockRes._implicitHeader = ()=>{};
|
|
mockRes.connection = requestConnection;
|
|
mockRes.finished = false;
|
|
mockRes.statusCode = 200;
|
|
const mockReq = new Stream.Readable();
|
|
mockReq._read = ()=>{
|
|
mockReq.emit("end");
|
|
mockReq.emit("close");
|
|
return Buffer.from("");
|
|
};
|
|
mockReq.headers = requestHeaders;
|
|
mockReq.method = requestMethod;
|
|
mockReq.url = requestUrl;
|
|
mockReq.connection = requestConnection;
|
|
return {
|
|
resBuffers,
|
|
req: mockReq,
|
|
res: mockRes,
|
|
streamPromise: isStreamFinished
|
|
};
|
|
}
|
|
|
|
//# sourceMappingURL=mock-request.js.map
|