feat: Zoe remember the new bundleId when upgrading a contract

Agoric · Feb 24, 2024 · 5a30b2d · 5a30b2d
1 parent fc2d0f0
commit 5a30b2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/packages/vats/src/proposals/probeZcfBundle.js b/packages/vats/src/proposals/probeZcfBundle.js
@@ -48,7 +48,7 @@ export const probeZcfBundleCap = async (
   await E(adminNode).upgrade(zoeBundleCap, {});
 
   // STEP 4: restart WF ////////////////////////
-  await E(walletAdminFacet).upgradeContract(walletRef.bundleID, privateArgs);
+  await E(walletAdminFacet).restartContract(privateArgs);
 
   // //////  See which zcf bundle was used //////////
 };

diff --git a/packages/zoe/src/zoeService/startInstance.js b/packages/zoe/src/zoeService/startInstance.js
@@ -266,6 +266,7 @@ export const makeStartInstance = (
           contractBundleCap: newContractBundleCap,
           privateArgs: newPrivateArgs,
         };
+        state.contractBundleCap = newContractBundleCap;
         return E.when(getFreshZcfBundleCap(), bCap =>
           E(state.adminNode).upgrade(bCap, { vatParameters }),
         );