feat(loadgen): handle AMM method renames (#84)

Handles the upcoming rename in Agoric/agoric-sdk#5377
Agoric · May 24, 2022 · 1a9c9c2 · 1a9c9c2
1 parent 7c8f64a
commit 1a9c9c2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/loadgen/contract/agent-prepare-loadgen.js b/loadgen/contract/agent-prepare-loadgen.js
@@ -257,9 +257,9 @@ export default async function startAgent({
       brand: secondaryBrandP,
       purse: secondaryPurse,
     } = E.get(tokenKit);
-    const liquidityIssuer = E(amm).addPool(
-      secondaryIssuer,
-      issuerPetnames[tokenBrandPetname],
+    const liquidityIssuer = fallback(
+      E(amm).addIssuer(secondaryIssuer, issuerPetnames[tokenBrandPetname]),
+      E(amm).addPool(secondaryIssuer, issuerPetnames[tokenBrandPetname]),
     );
     const liquidityBrandP = E(liquidityIssuer).getBrand();
 
@@ -312,7 +312,7 @@ export default async function startAgent({
     });
 
     const addLiquiditySeat = E(zoe).offer(
-      E(amm).makeAddLiquidityInvitation(),
+      fallback(E(amm).addPoolInvitation(), E(amm).makeAddLiquidityInvitation()),
       proposal,
       harden({
         Secondary: secondaryAMMPayment,

diff --git a/loadgen/types.js b/loadgen/types.js
@@ -28,7 +28,9 @@ export {};
  *   XYKAMMPublicFacet,
  *   'getPriceAuthorities' | 'makeSwapInInvitation' | 'makeAddLiquidityInvitation'
  * > & {
- *   addPool: (issuer: ERef<Issuer>, keyword: Keyword) => Promise<Issuer>
+ *   addPool?: (issuer: ERef<Issuer>, keyword: Keyword) => Promise<Issuer>;
+ *   addIssuer?: (issuer: ERef<Issuer>, keyword: Keyword) => Promise<Issuer>;
+ *   addPoolInvitation?: () => Promise<Invitation>;
  * } } AttenuatedAMM
  */