fix(loadgen): Handle changes to Treasury from PR 4209

The name of the Treasury changed to VaultFactory in Agoric/agoric-sdk#4209. Pick whichever is available for backwards compatibility
Agoric · Jan 8, 2022 · 01bf114 · 01bf114
1 parent 7d8ef17
commit 01bf114
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/loadgen/contract/agent-create-vault.js b/loadgen/contract/agent-create-vault.js
@@ -23,16 +23,24 @@ export default async function startAgent([key, home]) {
     runPurse,
     bldPurse,
     treasuryInstance,
+    vaultFactoryInstance,
   } = await allValues({
     runBrand: E(agoricNames).lookup('brand', issuerPetnames.RUN),
     // bldBrand: E(agoricNames).lookup('brand', issuerPetnames.BLD),
     bldBrand: E(E(wallet).getIssuer(issuerPetnames.BLD)).getBrand(),
     runPurse: E(wallet).getPurse(pursePetnames.RUN),
     bldPurse: E(wallet).getPurse(pursePetnames.BLD),
-    treasuryInstance: E(home.agoricNames).lookup('instance', 'Treasury'),
+    treasuryInstance: E(agoricNames)
+      .lookup('instance', 'Treasury')
+      .catch(() => {}),
+    vaultFactoryInstance: E(agoricNames)
+      .lookup('instance', 'VaultFactory')
+      .catch(() => {}),
   });
 
-  const treasuryPublicFacet = E(zoe).getPublicFacet(treasuryInstance);
+  const treasuryPublicFacet = E(zoe).getPublicFacet(
+    vaultFactoryInstance || treasuryInstance,
+  );
 
   const bldBalance = await E(bldPurse).getCurrentAmount();
   if (AmountMath.isEmpty(bldBalance)) {