Test for multichain rollback + rollback for wildcard fix

Author: DZakh

codegenerator/cli/npm/envio/src/FetchState.res

@@ -1161,11 +1161,18 @@ let rollbackPartition = (
   ~firstChangeEvent: blockNumberAndLogIndex,
   ~addressesToRemove,
 ) => {
+  let shouldRollbackFetched = p.latestFetchedBlock.blockNumber >= firstChangeEvent.blockNumber
+  let latestFetchedBlock = shouldRollbackFetched
+    ? {
+        blockNumber: firstChangeEvent.blockNumber - 1,
+        blockTimestamp: 0,
+      }
+    : p.latestFetchedBlock
   switch p {
   | {selection: {dependsOnAddresses: false}} =>
     Some({
       ...p,
-      // FIXME: Should rollback latestFetchedBlock???
+      latestFetchedBlock,
       status: {
         fetchingStateId: None,
       },
@@ -1183,21 +1190,14 @@ let rollbackPartition = (
     if rollbackedAddressesByContractName->Js.Dict.keys->Array.length === 0 {
       None
     } else {
-      let shouldRollbackFetched = p.latestFetchedBlock.blockNumber >= firstChangeEvent.blockNumber
-
       Some({
         id: p.id,
         selection: p.selection,
         status: {
           fetchingStateId: None,
         },
         addressesByContractName: rollbackedAddressesByContractName,
-        latestFetchedBlock: shouldRollbackFetched
-          ? {
-              blockNumber: firstChangeEvent.blockNumber - 1,
-              blockTimestamp: 0,
-            }
-          : p.latestFetchedBlock,
+        latestFetchedBlock,
       })
     }
   }

scenarios/test_codegen/test/helpers/Mock.res

@@ -217,7 +217,7 @@ module Indexer = {
 
   type chainConfig = {chain: Types.chain, sources: array<Source.t>, startBlock?: int}
 
-  let make = async (~chains: array<chainConfig>) => {
+  let make = async (~chains: array<chainConfig>, ~multichain=InternalConfig.Unordered) => {
     DbHelpers.resetPostgresClient()
 
     let config = RegisterHandlers.registerAllHandlers()
@@ -265,6 +265,7 @@ module Indexer = {
       persistence,
       enableRawEvents: false,
       chainMap,
+      multichain,
     }
 
     let gsManagerRef = ref(None)
@@ -398,6 +399,9 @@ module Source = {
     {
       getHeightOrThrowCalls,
       resolveGetHeightOrThrow: height => {
+        if getHeightOrThrowResolveFns->Utils.Array.isEmpty {
+          Js.Exn.raiseError("getHeightOrThrowResolveFns is empty")
+        }
         getHeightOrThrowResolveFns->Array.forEach(resolve => resolve(height))
       },
       rejectGetHeightOrThrow: exn => {
@@ -497,7 +501,16 @@ module Source = {
                             isWildcard: false,
                             filterByAddresses: false,
                             dependsOnAddresses: false,
-                            handler: item.handler->(
+                            handler: (
+                              ({context} as args) => {
+                                // We don't want preload optimization for the tests
+                                if context.isPreload {
+                                  Promise.resolve()
+                                } else {
+                                  item.handler(args)
+                                }
+                              }
+                            )->(
                               Utils.magic: Types.HandlerTypes.loader<unit, unit> => option<
                                 Internal.handler,
                               >