feat: add collection, user credentials, and skip flags to loadtest-public-api.js

Author: yash-pouranik

scripts/loadtest-public-api.js

@@ -20,6 +20,12 @@ const target = (args.target || process.env.PUBLIC_API_URL || 'http://localhost:5
 const connections = parseInt(args.connections || '50', 10);
 const duration = parseInt(args.duration || '10', 10);
 const apiKey = args.apiKey || args.pk || args.sk || process.env.TEST_API_KEY || process.env.PUBLIC_KEY || 'pk_live_demo_key';
+const collection = args.collection || 'posts';
+const email = args.email || 'loadtest@example.com';
+const password = args.password || 'password123';
+const skipAuth = args.skipAuth === 'true' || args.skipAuth === true;
+const skipData = args.skipData === 'true' || args.skipData === true;
+
 const keyType = apiKey.startsWith('sk_live_') ? 'Secret Key (sk_live_)' : apiKey.startsWith('pk_live_') ? 'Publishable Key (pk_live_)' : 'Custom Key';
 
 console.log('====================================================');
@@ -29,6 +35,7 @@ console.log(`📍 Target URL  : ${target}`);
 console.log(`⚡ Connections : ${connections} VUs (Virtual Users)`);
 console.log(`⏱️ Duration   : ${duration} seconds`);
 console.log(`🔑 Key Type    : ${keyType}`);
+console.log(`📁 Collection  : ${collection}`);
 console.log('====================================================\n');
 
 async function runBenchmark(name, opts) {
@@ -60,49 +67,55 @@ async function runBenchmark(name, opts) {
 }
 
 async function startLoadTest() {
-    // Test 1: Health Check Endpoint
+    // Test 1: Health Check Endpoint (Works without any project setup)
     await runBenchmark('1. Health Check Endpoint (GET /api/health)', {
         url: `${target}/api/health`,
         method: 'GET',
     });
 
-    // Test 2: API Key Auth & Data Collection Read (RLS + DB Query)
-    await runBenchmark('2. Data Collection Read (GET /api/data/posts)', {
-        url: `${target}/api/data/posts?limit=10`,
-        method: 'GET',
-        headers: {
-            'x-api-key': apiKey,
-            'accept': 'application/json',
-        },
-    });
+    // Test 2: Data Collection Read
+    if (!skipData) {
+        await runBenchmark(`2. Data Collection Read (GET /api/data/${collection})`, {
+            url: `${target}/api/data/${collection}?limit=10`,
+            method: 'GET',
+            headers: {
+                'x-api-key': apiKey,
+                'accept': 'application/json',
+            },
+        });
+    }
 
-    // Test 3: User Auth Login (Bcrypt Hash & JWT Signing Load)
-    await runBenchmark('3. User Auth Login (POST /api/userAuth/login)', {
-        url: `${target}/api/userAuth/login`,
-        method: 'POST',
-        headers: {
-            'x-api-key': apiKey,
-            'content-type': 'application/json',
-        },
-        body: {
-            email: 'loadtest@example.com',
-            password: 'password123',
-        },
-    });
+    // Test 3: User Auth Login
+    if (!skipAuth) {
+        await runBenchmark('3. User Auth Login (POST /api/userAuth/login)', {
+            url: `${target}/api/userAuth/login`,
+            method: 'POST',
+            headers: {
+                'x-api-key': apiKey,
+                'content-type': 'application/json',
+            },
+            body: {
+                email,
+                password,
+            },
+        });
+    }
 
-    // Test 4: Data Write / Document Insert (POST /api/data/posts)
-    await runBenchmark('4. Document Insert (POST /api/data/posts)', {
-        url: `${target}/api/data/posts`,
-        method: 'POST',
-        headers: {
-            'x-api-key': apiKey,
-            'content-type': 'application/json',
-        },
-        body: {
-            title: 'Load Test Post',
-            content: 'Benchmarking database write performance',
-        },
-    });
+    // Test 4: Data Write / Document Insert
+    if (!skipData) {
+        await runBenchmark(`4. Document Insert (POST /api/data/${collection})`, {
+            url: `${target}/api/data/${collection}`,
+            method: 'POST',
+            headers: {
+                'x-api-key': apiKey,
+                'content-type': 'application/json',
+            },
+            body: {
+                title: 'Load Test Post',
+                content: 'Benchmarking database write performance',
+            },
+        });
+    }
 
     console.log('🎉 Load Testing Complete!');
 }