Fix getWalletIndex function (#1074)

didx-xyz · Sep 27, 2024 · 53e2155 · 53e2155
1 parent 3884838
commit 53e2155
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 7 deletions.
diff --git a/scripts/k6/libs/functions.js b/scripts/k6/libs/functions.js
@@ -8,8 +8,8 @@ import sse from "k6/x/sse";
 // let customDuration = new Trend('custom_duration', true);
 
 // Helper function to generate a unique, zero-based index for even distribution of operations
-export function getWalletIndex(vu, iter) {
-  return (vu - 1) * (iter + 1) + iter;
+export function getWalletIndex(vu, iter, iterationsPerVu) {
+  return (vu - 1) * iterationsPerVu + iter;
 }
 
 function logError(response, requestBody) {
@@ -458,6 +458,7 @@ export function getProofIdByThreadId(holderAccessToken, threadId) {
       // Check if the current object has a matching thread_id
       if (obj.thread_id === threadId) {
         // Return the credential_id if a match is found
+        // console.log(`Log of the request made: ${JSON.stringify(response.request)}`);
         return obj.proof_id;
       }
     }
@@ -494,6 +495,7 @@ export function getProofIdCredentials(holderAccessToken, proofId) {
       return referent;
     }
     // Throw an error if no match is found
+    console.log(`Log of the request made: ${JSON.stringify(response.request)}`);
     throw new Error(`No match found for proofId: ${proofId}\nResponse body: ${JSON.stringify(responseData, null, 2)}`);
   } catch (error) {
     console.error("Error in getProofIdCredentials:", error);

diff --git a/scripts/k6/scenarios/create-credentials.js b/scripts/k6/scenarios/create-credentials.js
@@ -71,7 +71,7 @@ export function setup() {
 export default function (data) {
   const bearerToken = data.bearerToken;
   const holders = data.holders;
-  const walletIndex = getWalletIndex(__VU, __ITER);
+  const walletIndex = getWalletIndex(__VU, __ITER, iterations);
   const wallet = holders[walletIndex];
 
   // console.log(`VU: ${__VU}, Iteration: ${__ITER}, Wallet Index: ${walletIndex}, Issuer Wallet ID: ${wallet.issuer_wallet_id}`);
@@ -128,6 +128,8 @@ export default function (data) {
     },
   });
 
+  // sleep(1);
+
   const credentialId = getCredentialIdByThreadId(wallet.access_token, threadId);
 
   const acceptCredentialResponse = acceptCredential(wallet.access_token, credentialId);

diff --git a/scripts/k6/scenarios/create-invitations.js b/scripts/k6/scenarios/create-invitations.js
@@ -83,7 +83,9 @@ export default function (data) {
   const start = Date.now();
   const bearerToken = data.bearerToken;
   const issuers = data.issuers;
-  const walletIndex = getWalletIndex(__VU, __ITER);
+  const walletIndex = getWalletIndex(__VU, __ITER, iterations);
+
+  // console.log(`VU: ${__VU}, Iteration: ${__ITER}, Wallet Index: ${walletIndex}`);
 
   const holders = data.holders;
   const wallet = holders[walletIndex];

diff --git a/scripts/k6/scenarios/create-proof.js b/scripts/k6/scenarios/create-proof.js
@@ -1,7 +1,7 @@
 /* global __ENV, __ITER, __VU */
 /* eslint-disable no-undefined, no-console, camelcase */
 
-import { check } from "k6";
+import { check, sleep } from "k6";
 import { Counter } from "k6/metrics";
 import { getBearerToken } from "../libs/auth.js";
 import {
@@ -71,7 +71,7 @@ export function setup() {
 export default function (data) {
   const bearerToken = data.bearerToken;
   const tenants = data.tenants;
-  const walletIndex = getWalletIndex(__VU, __ITER);
+  const walletIndex = getWalletIndex(__VU, __ITER, iterations);
   const wallet = tenants[walletIndex];
 
   // console.log(`wallet.issuer_connection_id: ${wallet.issuer_connection_id}`);
@@ -117,6 +117,7 @@ export default function (data) {
 
   // TODO: return object and add check for the response
   const proofId = getProofIdByThreadId(wallet.access_token, threadId);
+  // console.log(`Proof ID: ${proofId}`);
   const referent = getProofIdCredentials(wallet.access_token, proofId);
 
   const acceptProofResponse = acceptProofRequest(wallet.access_token, proofId, referent);

diff --git a/scripts/k6/scenarios/revoke-credentials.js b/scripts/k6/scenarios/revoke-credentials.js
@@ -50,7 +50,7 @@ export function setup() {
 
 export default function (data) {
   const tenants = data.tenants;
-  const walletIndex = getWalletIndex(__VU, __ITER);
+  const walletIndex = getWalletIndex(__VU, __ITER, iterations);
   const wallet = tenants[walletIndex];
 
   const revokeCredentialResponse = revokeCredentialAutoPublish(wallet.issuer_access_token, wallet.credential_exchange_id);