chore(wallet-tools): notifier -> async iterator

Agoric · Feb 22, 2024 · 9e3d865 · 9e3d865
1 parent 3c7e212
commit 9e3d865
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/contract/test/wallet-tools.js b/contract/test/wallet-tools.js
@@ -1,6 +1,7 @@
 // @ts-check
 import { E, Far } from '@endo/far';
 import { makePromiseKit } from '@endo/promise-kit';
+import { makeNotifier } from '@agoric/notifier';
 import { allValues, mapValues } from '../src/objectTools.js';
 
 export { allValues, mapValues };
@@ -162,15 +163,20 @@ export const mockWalletFactory = (
       };
     }
 
+    async function* purseUpdates(brand) {
+      const n = makeNotifier(
+        E(
+          purseByBrand.get(brand) || Fail`${q(brand)}`,
+        ).getCurrentAmountNotifier(),
+      );
+      for await (const amount of n) {
+        yield amount;
+      }
+    }
     return {
       deposit: depositFacet,
       offers: Far('Offers', { executeOffer, addIssuer, tryExit }),
-      peek: Far('Wallet Peek', {
-        purseNotifier: brand =>
-          E(
-            purseByBrand.get(brand) || Fail`${q(brand)}`,
-          ).getCurrentAmountNotifier(),
-      }),
+      peek: Far('Wallet Peek', { purseUpdates }),
     };
   };
 
@@ -187,6 +193,7 @@ export const seatLike = updates => {
     payouts: makePromiseKit(),
   };
   (async () => {
+    await null;
     try {
       // XXX an error here is somehow and unhandled rejection
       for await (const update of updates) {