add tests for pool staments rollbacks

Author: Cmdv

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

@@ -34,12 +34,12 @@ import Cardano.Mock.Forging.Types (PoolIndex (..), StakeIndex (..), UTxOIndex (.
 import Cardano.Prelude
 import Data.Maybe.Strict (StrictMaybe (..))
 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 Test.Cardano.Db.Mock.Validate (assertBlockNoBackoff, assertEqQuery, assertTxCount)
 import Test.Tasty.HUnit (Assertion (), assertBool, assertEqual)
 import Prelude (error, head, last)
+import Test.Cardano.Db.Mock.Config (claFullMode, withFullConfigDropDB, conwayConfigDir, startDBSync, stopDBSync, withCustomConfigDropDB, configPoolStats, queryDBSync, initCommandLineArgs)
+import Test.Cardano.Db.Mock.Examples
+import Test.Cardano.Db.Mock.Validate (assertBlockNoBackoff, assertTxCount, assertEqQuery)
+import Test.Cardano.Db.Mock.UnifiedApi
 
 simpleRollback :: IOManager -> [(Text, Text)] -> Assertion
 simpleRollback =
@@ -391,10 +391,10 @@ poolStatBasicTest =
       void $ forgeAndSubmitBlocks interpreter mockServer 200
       assertBlockNoBackoff dbSync 201
 
-      poolStatCount <- queryDBSync dbSync Db.queryPoolStatCount
+      poolStatCount <- queryDBSync dbSync DB.queryPoolStatCount
 
       -- Verify pool stats are created and no duplicates exist
-      duplicateCount <- queryDBSync dbSync Db.queryPoolStatDuplicates
+      duplicateCount <- queryDBSync dbSync DB.queryPoolStatDuplicates
       assertEqual "Should have no duplicate pool stats" 0 duplicateCount
       assertBool "Should have some pool stats" (poolStatCount > 0)
   where
@@ -429,7 +429,7 @@ poolStatRollbackNoDuplicates =
       assertBlockNoBackoff dbSync 351 -- Should stay same due to rollbackExpand commentComment on line R349ResolvedCode has comments. Press enter to view.
 
       -- The main test: no duplicates after rollback + re-sync
-      duplicateCount <- queryDBSync dbSync Db.queryPoolStatDuplicates
+      duplicateCount <- queryDBSync dbSync DB.queryPoolStatDuplicates
       assertEqual "Should have no duplicate pool stats after rollback" 0 duplicateCount
   where
     args = initCommandLineArgs {claFullMode = False}
@@ -451,15 +451,15 @@ poolStatRollbackGeneral =
       assertBlockNoBackoff dbSync totalBeforeRollback
 
       -- Check initial pool stat count
-      initialCount <- queryDBSync dbSync Db.queryPoolStatCount
+      initialCount <- queryDBSync dbSync DB.queryPoolStatCount
 
       -- Forge more blocks to create additional pool stats
       epochBlks2 <- fillEpochs interpreter mockServer 1
       let totalAfterEpoch = totalBeforeRollback + length epochBlks2
       assertBlockNoBackoff dbSync totalAfterEpoch
 
       -- Verify pool stats increased
-      afterCount <- queryDBSync dbSync Db.queryPoolStatCount
+      afterCount <- queryDBSync dbSync DB.queryPoolStatCount
       assertBool "Pool stats should have increased" (afterCount > initialCount)
 
       -- Rollback to previous point
@@ -470,13 +470,13 @@ poolStatRollbackGeneral =
       epochBlks3 <- fillEpochs interpreter mockServer 1
       let finalTotal = totalBeforeRollback + length epochBlks3 + 1
       assertBlockNoBackoff dbSync finalTotal
-      finalCount <- queryDBSync dbSync Db.queryPoolStatCount
+      finalCount <- queryDBSync dbSync DB.queryPoolStatCount
 
       -- Verify count matches and no constraint violations occurred
       assertEqual "Pool stat count should match after rollback" afterCount finalCount
 
       -- Also verify no duplicates
-      duplicateCount <- queryDBSync dbSync Db.queryPoolStatDuplicates
+      duplicateCount <- queryDBSync dbSync DB.queryPoolStatDuplicates
       assertEqual "Should have no duplicate pool stats" 0 duplicateCount
   where
     args = initCommandLineArgs {claFullMode = False}

cardano-db/src/Cardano/Db/Statement/Pool.hs

@@ -208,6 +208,53 @@ insertBulkPoolStat :: [SCP.PoolStat] -> DbM ()
 insertBulkPoolStat poolStats =
   runSession mkDbCallStack $ HsqlSes.statement poolStats insertBulkPoolStatStmt
 
+--------------------------------------------------------------------------------
+queryPoolStatCountStmt :: HsqlStmt.Statement () Int64
+queryPoolStatCountStmt =
+  HsqlStmt.Statement sql encoder decoder True
+  where
+    poolStatTable = tableName (Proxy @SCP.PoolStat)
+
+    sql =
+      TextEnc.encodeUtf8 $
+        Text.concat
+          [ "SELECT COUNT(*)::bigint FROM "
+          , poolStatTable
+          ]
+
+    encoder = mempty
+    decoder = HsqlD.singleRow (HsqlD.column $ HsqlD.nonNullable HsqlD.int8)
+
+queryPoolStatCount :: DbM Int64
+queryPoolStatCount =
+  runSession mkDbCallStack $ HsqlSes.statement () queryPoolStatCountStmt
+
+--------------------------------------------------------------------------------
+queryPoolStatDuplicatesStmt :: HsqlStmt.Statement () Int64
+queryPoolStatDuplicatesStmt =
+  HsqlStmt.Statement sql encoder decoder True
+  where
+    poolStatTable = tableName (Proxy @SCP.PoolStat)
+
+    sql =
+      TextEnc.encodeUtf8 $
+        Text.concat
+          [ "SELECT COUNT(*)::bigint FROM ("
+          , "  SELECT pool_hash_id, epoch_no"
+          , "  FROM "
+          , poolStatTable
+          , "  GROUP BY pool_hash_id, epoch_no"
+          , "  HAVING COUNT(*) > 1"
+          , ") AS duplicates"
+          ]
+
+    encoder = mempty
+    decoder = HsqlD.singleRow (HsqlD.column $ HsqlD.nonNullable HsqlD.int8)
+
+queryPoolStatDuplicates :: DbM Int64
+queryPoolStatDuplicates =
+  runSession mkDbCallStack $ HsqlSes.statement () queryPoolStatDuplicatesStmt
+
 --------------------------------------------------------------------------------
 -- PoolOwner
 --------------------------------------------------------------------------------