feat: refactor stress-metrics

load_tests/src/get-original-stress-test.ts

@@ -4,7 +4,7 @@ import { check } from 'k6';
 import { textSummary } from 'https://jslib.k6.io/k6-summary/0.1.0/index.js';
 import { getStressTestOptions } from './options';
 import { SignedRequest } from './signed-request';
-import { getStressTestCsv } from './stress-test-csv';
+import { getCSVStressMetrics } from './stress-metrics';
 
 // this test calls s3 directly
 const signedRequest = SignedRequest({
@@ -24,7 +24,7 @@ export default async function () {
 }
 
 export function handleSummary(data: any) {
-    const csv = getStressTestCsv(data);
+    const csv = getCSVStressMetrics(data['metrics']);
     return {
         stdout: textSummary(data, { indent: ' ', enableColors: true }),
         'get-original-load-test-summary.json': JSON.stringify(data, null, 2),

load_tests/src/get-sidekick-stress-test.ts

@@ -6,7 +6,7 @@ import { tagWithCurrentStageIndex } from 'https://jslib.k6.io/k6-utils/1.3.0/ind
 import { textSummary } from 'https://jslib.k6.io/k6-summary/0.1.0/index.js';
 import { getStressTestOptions } from './options';
 import { SignedRequest } from './signed-request';
-import { getStressTestCsv } from './stress-test-csv';
+import { getCSVStressMetrics } from './stress-metrics';
 
 // This test will get the original file through sidekick
 const signedRequest = SignedRequest({
@@ -26,7 +26,7 @@ export default async function () {
 }
 
 export function handleSummary(data: any) {
-    const csv = getStressTestCsv(data);
+    const csv = getCSVStressMetrics(data['metrics']);
 
     return {
         stdout: textSummary(data, { indent: ' ', enableColors: true }),

load_tests/src/stress-metrics.ts

@@ -0,0 +1,47 @@
+type MetricData = {
+    type: string;
+    contains: string;
+    values: {
+        avg: number;
+        min: number;
+        med: number;
+        max: number;
+        'p(90)': number;
+        'p(95)': number;
+    };
+};
+
+type Data = {
+    [key: string]: MetricData;
+};
+
+export function getCSVStressMetrics(data: Data) {
+    let rows: string[][] = [];
+
+    for (const metric in data) {
+        if (metric.startsWith('http_req_duration') && metric.includes('scenario')) {
+            const scenarioTag = metric.split('{')[1].split('}')[0];
+            const name = scenarioTag.split(':')[1];
+            const values = data[metric].values;
+
+            rows.push([
+                name,
+                values['avg'].toString(),
+                values['min'].toString(),
+                values['med'].toString(),
+                values['max'].toString(),
+                values['p(90)'].toString(),
+                values['p(95)'].toString(),
+            ]);
+        }
+    }
+
+    rows.sort((a, b) => {
+        const numA = Number(a[0].substring(2));
+        const numB = Number(b[0].substring(2));
+        return numA - numB;
+    });
+    rows.unshift(['scenario', 'avg', 'min', 'med', 'max', 'p(90)', 'p(95)']);
+
+    return rows.map((row) => row.join(',')).join('\n');
+}