feat: udpated stress tests

project-n-oss · May 30, 2024 · f6aad1d · f6aad1d
1 parent 1c95b43
commit f6aad1d
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 51 deletions.
diff --git a/load_tests/src/get-original-stress-test.ts b/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 { getCSVStressMetrics } from './stress-metrics';
+import { getStressTestCsv } from './stress-test-csv';
 
 // this test calls s3 directly
 const signedRequest = SignedRequest({
@@ -24,7 +24,7 @@ export default async function () {
 }
 
 export function handleSummary(data: any) {
-    const csv = getCSVStressMetrics(data['metrics']);
+    const csv = getStressTestCsv(data);
     return {
         stdout: textSummary(data, { indent: ' ', enableColors: true }),
         'get-original-load-test-summary.json': JSON.stringify(data, null, 2),

diff --git a/load_tests/src/get-sidekick-stress-test.ts b/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 { getCSVStressMetrics } from './stress-metrics';
+import { getStressTestCsv } from './stress-test-csv';
 
 // 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 = getCSVStressMetrics(data['metrics']);
+    const csv = getStressTestCsv(data);
 
     return {
         stdout: textSummary(data, { indent: ' ', enableColors: true }),

diff --git a/load_tests/src/stress-metrics.ts b/load_tests/src/stress-metrics.ts
diff --git a/load_tests/src/stress-test-csv.ts b/load_tests/src/stress-test-csv.ts
@@ -0,0 +1,25 @@
+export function getStressTestCsv(data: any) {
+    let csv: string[] = ['scenario,avg,min,med,max,p(90),p(95)'];
+
+    for (const metric in data['metrics']) {
+        // Check if the metrics is a http_req_duration scenario
+        if (metric.startsWith('http_req_duration') && metric.includes('scenario')) {
+            const scenario = metric.split('{')[1].split('}')[0];
+            const values = data['metrics'][metric]['values'];
+
+            const csvRow: string[] = [
+                scenario,
+                values['avg'].toString(),
+                values['min'].toString(),
+                values['med'].toString(),
+                values['max'].toString(),
+                values['p(90)'].toString(),
+                values['p(95)'].toString(),
+            ];
+
+            csv.push(csvRow.join(','));
+        }
+    }
+
+    return csv.join('\n');
+}