fix tests

Author: Cmdv

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

@@ -35,6 +35,7 @@ module Test.Cardano.Db.Mock.Config (
   configMetadataEnable,
   configMetadataDisable,
   configMetadataKeys,
+  configPoolStats,
   mkFingerPrint,
   mkMutableDir,
   mkDBSyncEnv,
@@ -353,6 +354,10 @@ configMetadataKeys :: SyncNodeConfig -> SyncNodeConfig
 configMetadataKeys cfg = do
   cfg {dncInsertOptions = (dncInsertOptions cfg) {sioMetadata = MetadataKeys $ 1 :| []}}
 
+configPoolStats :: SyncNodeConfig -> SyncNodeConfig
+configPoolStats cfg = do
+  cfg {dncInsertOptions = (dncInsertOptions cfg) {sioPoolStats = PoolStatsConfig True}}
+
 initCommandLineArgs :: CommandLineArgs
 initCommandLineArgs =
   CommandLineArgs

cardano-chain-gen/test/Test/Cardano/Db/Mock/Unit/Conway/Rollback.hs

@@ -13,6 +13,7 @@ module Test.Cardano.Db.Mock.Unit.Conway.Rollback (
   poolStatRollback,
 ) where
 
+import qualified Cardano.Db as Db
 import Cardano.Ledger.Coin (Coin (..))
 import Cardano.Ledger.Conway.TxCert (ConwayDelegCert (..), Delegatee (..))
 import Cardano.Mock.ChainSync.Server (IOManager (), addBlock, rollback)
@@ -26,11 +27,10 @@ import Ouroboros.Network.Block (blockPoint)
 import Test.Cardano.Db.Mock.Config
 import Test.Cardano.Db.Mock.Examples (mockBlock0, mockBlock1, mockBlock2)
 import Test.Cardano.Db.Mock.UnifiedApi
+import qualified Test.Cardano.Db.Mock.UnifiedApi as Api
 import Test.Cardano.Db.Mock.Validate (assertBlockNoBackoff, assertTxCount, runQuery)
 import Test.Tasty.HUnit (Assertion (), assertBool, assertEqual)
 import Prelude (last)
-import qualified Test.Cardano.Db.Mock.UnifiedApi as Api
-import qualified Cardano.Db as Db
 
 simpleRollback :: IOManager -> [(Text, Text)] -> Assertion
 simpleRollback =
@@ -297,37 +297,47 @@ rollbackFullTx =
 
 poolStatRollback :: IOManager -> [(Text, Text)] -> Assertion
 poolStatRollback =
-  withFullConfigAndDropDB conwayConfigDir testLabel $ \interpreter mockServer dbSync -> do
-    startDBSync dbSync
-
-    -- Create pools and stake to generate pool stats
-    void $ Api.registerAllStakeCreds interpreter mockServer
-    void $ Api.fillEpochs interpreter mockServer 2
-
-    -- Create rollback point
-    blks <- Api.forgeAndSubmitBlocks interpreter mockServer 10
-    assertBlockNoBackoff dbSync (2 + length blks)
-
-    -- Check initial pool stat count
-    initialCount <- runQuery dbSync Db.queryPoolStatCount
-
-    -- Forge more blocks to create additional pool stats
-    void $ Api.fillEpochs interpreter mockServer 1
-    assertBlockNoBackoff dbSync (3 + length blks)
-
-    -- Verify pool stats increased
-    afterCount <- runQuery dbSync Db.queryPoolStatCount
-    assertBool "Pool stats should have increased" (afterCount > initialCount)
-
-    -- Rollback to previous point
-    atomically $ rollback mockServer (blockPoint $ last blks)
-    assertBlockNoBackoff dbSync (3 + length blks) -- Delayed rollback
-
-    -- Re-sync the same blocks - should not create duplicates
-    void $ Api.fillEpochs interpreter mockServer 1
-    finalCount <- runQuery dbSync Db.queryPoolStatCount
-
-    -- Verify count matches and no constraint violations occurred
-    assertEqual "Pool stat count should match after rollback" afterCount finalCount
+  withCustomConfig args (Just configPoolStats) conwayConfigDir testLabel $
+    \interpreter mockServer dbSync -> do
+      startDBSync dbSync
+
+      -- Create pools and stake to generate pool stats
+      void $ Api.registerAllStakeCreds interpreter mockServer
+      epochBlks1 <- Api.fillEpochs interpreter mockServer 2
+
+      -- Create rollback point
+      rollbackBlks <- Api.forgeAndSubmitBlocks interpreter mockServer 10
+      let totalBeforeRollback = length epochBlks1 + length rollbackBlks + 1
+      assertBlockNoBackoff dbSync totalBeforeRollback
+
+      -- Check initial pool stat count
+      initialCount <- runQuery dbSync Db.queryPoolStatCount
+
+      -- Forge more blocks to create additional pool stats
+      epochBlks2 <- Api.fillEpochs interpreter mockServer 1
+      let totalAfterEpoch = totalBeforeRollback + length epochBlks2
+      assertBlockNoBackoff dbSync totalAfterEpoch
+
+      -- Verify pool stats increased
+      afterCount <- runQuery dbSync Db.queryPoolStatCount
+      assertBool "Pool stats should have increased" (afterCount > initialCount)
+
+      -- Rollback to previous point
+      atomically $ rollback mockServer (blockPoint $ last rollbackBlks)
+      assertBlockNoBackoff dbSync totalAfterEpoch -- Delayed rollback
+
+      -- Re-sync the same blocks - should not create duplicates
+      epochBlks3 <- Api.fillEpochs interpreter mockServer 1
+      let finalTotal = totalBeforeRollback + length epochBlks3 + 1
+      assertBlockNoBackoff dbSync finalTotal
+      finalCount <- runQuery dbSync Db.queryPoolStatCount
+
+      -- Verify count matches and no constraint violations occurred
+      assertEqual "Pool stat count should match after rollback" afterCount finalCount
   where
+    args =
+      initCommandLineArgs
+        { claFullMode = False
+        }
+
     testLabel = "conwayPoolStatRollback"