more debug prints

Author: ananas-block

js/stateless.js/tests/e2e/rpc-multi-trees.test.ts

@@ -212,20 +212,36 @@ describe('rpc-multi-trees', () => {
             const prePayerAccounts = await rpc.getCompressedAccountsByOwner(
                 payer.publicKey,
             );
-
-            const proofs = await rpc.getMultipleCompressedAccountProofs(
-                prePayerAccounts.items.map(account => bn(account.hash)),
+            let hashes = prePayerAccounts.items.map(account =>
+                bn(account.hash),
+            );
+            const proofs = await rpc.getMultipleCompressedAccountProofs(hashes);
+
+            // Create hash -> account map since proof order may not match input order
+            const accountByHash = new Map(
+                prePayerAccounts.items.map(acc => [
+                    bn(acc.hash).toString(),
+                    acc,
+                ]),
             );
 
             proofs.forEach((proof, index) => {
-                const expectedTree =
-                    prePayerAccounts.items[index].treeInfo.tree;
+                // Match proof to account by hash, not by index
+                const matchingAccount = accountByHash.get(
+                    bn(proof.hash).toString(),
+                );
+                if (!matchingAccount) {
+                    throw new Error(
+                        `No matching account found for proof hash ${bn(proof.hash).toString()}`,
+                    );
+                }
+
+                const expectedTree = matchingAccount.treeInfo.tree;
                 const actualTree = proof.treeInfo.tree;
-                const expectedQueue =
-                    prePayerAccounts.items[index].treeInfo.queue;
+                const expectedQueue = matchingAccount.treeInfo.queue;
                 const actualQueue = proof.treeInfo.queue;
-
-                console.log(`Iteration ${round + 1}, Account ${index}:`);
+                console.log(`.hash: ${bn(proof.hash).toString()}`);
+                console.log(`Iteration ${round + 1}, Proof ${index}:`);
                 console.log(
                     `  Expected tree (from getCompressedAccountsByOwner): ${expectedTree.toBase58()}`,
                 );