Merge pull request #397 from dozed/response-time-axios

Use axios interceptor to measure response time
This commit is contained in:
Anoop M D
2023-10-06 04:13:52 +05:30
committed by GitHub
2 changed files with 54 additions and 8 deletions

View File

@@ -11,6 +11,7 @@ const interpolateVars = require('./interpolate-vars');
const { ScriptRuntime, TestRuntime, VarsRuntime, AssertRuntime } = require('@usebruno/js');
const { stripExtension } = require('../utils/filesystem');
const { getOptions } = require('../utils/bru');
const { makeAxiosInstance } = require('../utils/axios-instance');
const runSingleRequest = async function (
filename,
@@ -124,15 +125,23 @@ const runSingleRequest = async function (
}
let response, responseTime;
const start = Date.now();
try {
// run request
response = await axios(request);
responseTime = Date.now() - start;
const axiosInstance = makeAxiosInstance();
/** @type {import('axios').AxiosResponse} */
response = await axiosInstance(request);
// Prevents the duration on leaking to the actual result
responseTime = response.headers.get('request-duration');
response.headers.delete('request-duration');
} catch (err) {
if (err && err.response) {
response = err.response;
responseTime = Date.now() - start;
// Prevents the duration on leaking to the actual result
responseTime = response.headers.get('request-duration');
response.headers.delete('request-duration');
} else {
console.log(chalk.red(stripExtension(filename)) + chalk.dim(` (${err.message})`));
return {
@@ -156,10 +165,7 @@ const runSingleRequest = async function (
}
}
console.log(
chalk.green(stripExtension(filename)) +
chalk.dim(` (${response.status} ${response.statusText}) - ${responseTime} ms`)
);
console.log(chalk.green(stripExtension(filename)) + chalk.dim(` (${response.status} ${response.statusText}) - ${responseTime} ms`));
// run post-response vars
const postResponseVars = get(bruJson, 'request.vars.res');