chore: makeFakeLocalchainBridge accepts makeAddressFn

Agoric · Dec 18, 2024 · 7bbe21d · 7bbe21d
1 parent b8cbd92
commit 7bbe21d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/vats/tools/fake-bridge.js b/packages/vats/tools/fake-bridge.js
@@ -291,9 +291,14 @@ export const fakeLocalChainBridgeQueryHandler = message => {
 /**
  * @param {import('@agoric/zone').Zone} zone
  * @param {(obj) => void} [onToBridge]
+ * @param {(number) => string} [makeAddressFn]
  * @returns {ScopedBridgeManager<'vlocalchain'>}
  */
-export const makeFakeLocalchainBridge = (zone, onToBridge = () => {}) => {
+export const makeFakeLocalchainBridge = (
+  zone,
+  onToBridge = () => {},
+  makeAddressFn,
+) => {
   /** @type {Remote<BridgeHandler>} */
   let hndlr;
   let lcaExecuteTxSequence = 0;
@@ -306,7 +311,9 @@ export const makeFakeLocalchainBridge = (zone, onToBridge = () => {}) => {
       trace('toBridge', type, method, params);
       switch (type) {
         case 'VLOCALCHAIN_ALLOCATE_ADDRESS': {
-          const address = `${LOCALCHAIN_DEFAULT_ADDRESS}${accountsCreated || ''}`;
+          const address = makeAddressFn
+            ? makeAddressFn(accountsCreated)
+            : `${LOCALCHAIN_DEFAULT_ADDRESS}${accountsCreated || ''}`;
           accountsCreated += 1;
           return address;
         }