From 83d63d6d3248666ebb23c7c29f0c58775f277482 Mon Sep 17 00:00:00 2001
From: Arnaud Bailly <arnaud@pankzsoft.com>
Date: Tue, 2 Dec 2025 04:43:49 +0100
Subject: adapt Bank interface to async

---
 pbt/ts/src/bank.test.ts | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

(limited to 'pbt/ts/src/bank.test.ts')

diff --git a/pbt/ts/src/bank.test.ts b/pbt/ts/src/bank.test.ts
index f6aa7b0..5939034 100644
--- a/pbt/ts/src/bank.test.ts
+++ b/pbt/ts/src/bank.test.ts
@@ -61,20 +61,20 @@ const genOperations = (accounts: Map<Account, number>) =>
 
 const correctBalance =
   (initialAccounts: Map<Account, number>) =>
-  (ops: Operation[]): Promise<boolean> => {
+  async (ops: Operation[]): Promise<boolean> => {
     const expectedBalances = new Map(initialAccounts);
     const api = new SimpleBank(initialAccounts);
 
     for (const op of ops) {
       switch (op.op) {
         case 'GetBalance': {
-          const retrieved = toNullable(api.balance(op.account));
+          const retrieved = toNullable(await api.balance(op.account));
           const expected = expectedBalances.get(op.account);
           if (retrieved !== expected) {
             console.log(
               `Balance mismatch for account ${op.account}: expected ${expected}, got ${retrieved}`,
             );
-            return Promise.resolve(false);
+            return false;
           }
           break;
         }
@@ -95,24 +95,24 @@ const correctBalance =
 
     // finally check all balances
     for (const [account, expectedBal] of expectedBalances.entries()) {
-      const retrieved = toNullable(api.balance(account));
+      const retrieved = toNullable(await api.balance(account));
       if (retrieved !== expectedBal) {
         console.log(
           `Final balance mismatch for account ${account}: expected ${expectedBal}, got ${retrieved}`,
         );
-        return Promise.resolve(false);
+        return false;
       }
     }
 
-    return Promise.resolve(true);
+    return true;
   };
 
 describe('Bank Simulator', () => {
   test('can retrieve balance given account exists', async () => {
-    const prop = forall(genAccounts, accounts => {
+    const prop = forall(genAccounts, async accounts => {
       const api = new SimpleBank(accounts);
-      return property(keyValueOf(accounts), ([accountNr, bal]) => {
-        const retrieved = toNullable(api.balance(accountNr));
+      return property(keyValueOf(accounts), async ([accountNr, bal]) => {
+        const retrieved = toNullable(await api.balance(accountNr));
         return Promise.resolve(retrieved === bal);
       });
     });
-- 
cgit v1.3.1