update test configs for consumed txout

Modified-by: Cmdv <vincent@cmdv.me>

Author: Cmdv

cardano-chain-gen/test/Test/Cardano/Db/Mock/Config.hs

@@ -19,6 +19,7 @@ module Test.Cardano.Db.Mock.Config (
   getPoolLayer,
   mkConfig,
   mkSyncNodeConfig,
+  mkCustomSyncNodeConfig,
   mkConfigDir,
   mkFingerPrint,
   mkMutableDir,
@@ -242,6 +243,11 @@ mkSyncNodeConfig configFilePath cmdLineArgs =
     configFilename = claConfigFilename cmdLineArgs
     configDir = mkConfigDir configFilePath
 
+mkCustomSyncNodeConfig :: FilePath -> CommandLineArgs -> (SyncNodeConfig -> SyncNodeConfig) -> IO SyncNodeConfig
+mkCustomSyncNodeConfig cfgDir args updateFn = do
+  initConfigFile <- mkSyncNodeConfig cfgDir args
+  pure $ updateFn initConfigFile
+
 mkShelleyCredentials :: FilePath -> IO [ShelleyLeaderCredentials StandardCrypto]
 mkShelleyCredentials bulkFile = do
   eitherM (panic . textShow) pure $ runExceptT $ readLeaderCredentials (Just protFiles)

cardano-chain-gen/test/Test/Cardano/Db/Mock/Unit/Babbage/Config/MigrateConsumedPruneTxOut.hs

@@ -18,10 +18,10 @@ import Test.Tasty.HUnit (Assertion ())
 
 configRemoveJsonbFromSchemaEnabled :: IOManager -> [(Text, Text)] -> Assertion
 configRemoveJsonbFromSchemaEnabled ioManager metadata = do
-  syncNodeConfig <- mksNodeConfig
+  syncNodeConfig <- mkCustomSyncNodeConfig cfgDir args config
   withCustomConfigAndDropDB args (Just syncNodeConfig) cfgDir testLabel action ioManager metadata
   where
-    action = \_interpreter _mockServer dbSync -> do
+    action _interpreter _mockServer dbSync = do
       startDBSync dbSync
       threadDelay 7_000_000
       assertEqQuery
@@ -36,21 +36,16 @@ configRemoveJsonbFromSchemaEnabled ioManager metadata = do
 
     cfgDir = conwayConfigDir
 
-    mksNodeConfig :: IO SyncNodeConfig
-    mksNodeConfig = do
-      initConfigFile <- mkSyncNodeConfig cfgDir args
-      let dncInsertOptions' = dncInsertOptions initConfigFile
-      pure $
-        initConfigFile
-          { dncInsertOptions = dncInsertOptions' {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig True}
-          }
+    config :: SyncNodeConfig -> SyncNodeConfig
+    config cfg = do
+      cfg {dncInsertOptions = (dncInsertOptions cfg) {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig True}}
 
 configRemoveJsonbFromSchemaDisabled :: IOManager -> [(Text, Text)] -> Assertion
 configRemoveJsonbFromSchemaDisabled ioManager metadata = do
-  syncNodeConfig <- mksNodeConfig
+  syncNodeConfig <- mkCustomSyncNodeConfig cfgDir args config
   withCustomConfigAndDropDB args (Just syncNodeConfig) cfgDir testLabel action ioManager metadata
   where
-    action = \_interpreter _mockServer dbSync -> do
+    action _interpreter _mockServer dbSync = do
       startDBSync dbSync
       threadDelay 7_000_000
       assertEqQuery
@@ -62,62 +57,49 @@ configRemoveJsonbFromSchemaDisabled ioManager metadata = do
 
     args = initCommandLineArgs {claFullMode = False}
     testLabel = "conwayConfigRemoveJsonbFromSchemaDisabled"
-
     cfgDir = conwayConfigDir
-
-    mksNodeConfig :: IO SyncNodeConfig
-    mksNodeConfig = do
-      initConfigFile <- mkSyncNodeConfig cfgDir args
-      let dncInsertOptions' = dncInsertOptions initConfigFile
-      pure $
-        initConfigFile
-          { dncInsertOptions = dncInsertOptions' {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig False}
-          }
+    config :: SyncNodeConfig -> SyncNodeConfig
+    config cfg =
+      cfg {dncInsertOptions = (dncInsertOptions cfg) {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig False}}
 
 configJsonbInSchemaShouldRemoveThenAdd :: IOManager -> [(Text, Text)] -> Assertion
 configJsonbInSchemaShouldRemoveThenAdd ioManager metadata = do
-  syncNodeConfig <- mksNodeConfig
+  syncNodeConfig <- mkCustomSyncNodeConfig cfgDir args config
   withCustomConfigAndDropDB args (Just syncNodeConfig) cfgDir testLabel action ioManager metadata
   where
-    action = \_interpreter _mockServer dbSync -> do
-      startDBSync dbSync
+    action _interpreter _mockServer dbSyncEnv = do
+      startDBSync dbSyncEnv
       threadDelay 7_000_000
       assertEqQuery
-        dbSync
+        dbSyncEnv
         DB.queryJsonbInSchemaExists
         False
         "There should be no jsonb types in database if option has been enabled"
-      stopDBSync dbSync
+      stopDBSync dbSyncEnv
       let newDbSyncEnv =
-            dbSync
+            dbSyncEnv
               { dbSyncConfig =
-                  (dbSyncConfig dbSync)
+                  (dbSyncConfig dbSyncEnv)
                     { dncInsertOptions =
-                        (dncInsertOptions $ dbSyncConfig dbSync)
+                        (dncInsertOptions $ dbSyncConfig dbSyncEnv)
                           { sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig False
                           }
                     }
               }
       startDBSync newDbSyncEnv
       threadDelay 7_000_000
       assertEqQuery
-        dbSync
+        dbSyncEnv
         DB.queryJsonbInSchemaExists
         True
         "There should be jsonb types in database if option has been disabled"
       -- Expected to fail
-      checkStillRuns dbSync
+      checkStillRuns dbSyncEnv
 
     args = initCommandLineArgs {claFullMode = False}
     testLabel = "configJsonbInSchemaShouldRemoveThenAdd"
-
     cfgDir = conwayConfigDir
 
-    mksNodeConfig :: IO SyncNodeConfig
-    mksNodeConfig = do
-      initConfigFile <- mkSyncNodeConfig cfgDir args
-      let dncInsertOptions' = dncInsertOptions initConfigFile
-      pure $
-        initConfigFile
-          { dncInsertOptions = dncInsertOptions' {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig True}
-          }
+    config :: SyncNodeConfig -> SyncNodeConfig
+    config cfg =
+      cfg {dncInsertOptions = (dncInsertOptions cfg) {sioRemoveJsonbFromSchema = RemoveJsonbFromSchemaConfig True}}