Ignore extra pkhs when fetching networks for contracts

src/utils/multisig/helper.test.ts

@@ -151,5 +151,25 @@ describe("multisig helpers", () => {
         ])
       );
     });
+
+    it("filters out extra pkhs from api response", async () => {
+      mockedContractsGet.mockImplementation((...args) => {
+        if (args[1].baseUrl === MAINNET.tzktApiUrl) {
+          return Promise.resolve(["pkh1", "pkh3", "pkh4"] as any);
+        } else {
+          return Promise.resolve(["pkh2", "pkh5"] as any);
+        }
+      });
+
+      const result = await getNetworksForContracts([MAINNET, GHOSTNET], ["pkh1", "pkh2", "pkh3"]);
+
+      expect(result).toEqual(
+        new Map([
+          ["pkh1", MAINNET.name],
+          ["pkh2", GHOSTNET.name],
+          ["pkh3", MAINNET.name],
+        ])
+      );
+    });
   });
 });

src/utils/multisig/helpers.ts

@@ -57,7 +57,11 @@ export const getNetworksForContracts = async (
       ])
     )
   );
-  accountsWithNetwork.flat().forEach(([pkh, network]) => result.set(pkh, network));
+  accountsWithNetwork.flat().forEach(([pkh, network]) => {
+    if (contractPkhs.includes(pkh)) {
+      result.set(pkh, network);
+    }
+  });
 
   return result;
 };

src/utils/redux/migrations.test.ts

@@ -44,6 +44,7 @@ describe("migrations", () => {
         new Map([
           [mainnetPkh, "mainnet"],
           [ghostnetPkh, "ghostnet"],
+          [mockContractAddress(3).pkh, "ghostnet"], // not in the store
         ])
       );
 

src/utils/redux/migrations.ts

@@ -46,10 +46,9 @@ export const mainStoreMigrations = {
       state.networks.available,
       contractPkhs
     );
-    const contractAccounts = [...contractsWithNetworks.entries()].map(([pkh, network]) => [
-      pkh,
-      { ...state.contacts[pkh], network },
-    ]);
+    const contractAccounts = contractPkhs
+      .filter(pkh => contractsWithNetworks.has(pkh))
+      .map(pkh => [pkh, { ...state.contacts[pkh], network: contractsWithNetworks.get(pkh) }]);
 
     return produce(state, (draft: any) => {
       draft.contacts = fromPairs([...implicitAccounts, ...contractAccounts]);
@@ -84,4 +83,5 @@ export const accountsMigrations = {
       });
     }),
   5: identity,
+  6: identity,
 } as any;