Extend prop_multiple_putPoolRetirement_single_readPoolRetirement.

Extend `prop_multiple_putPoolRetirement_single_readPoolRetirement` to call `listRetiredPools`.
cardano-foundation · Aug 13, 2020 · fc3d579 · fc3d579
1 parent cd28a2f
commit fc3d579
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/core/test/unit/Cardano/Pool/DB/Properties.hs b/lib/core/test/unit/Cardano/Pool/DB/Properties.hs
@@ -575,6 +575,8 @@ prop_multiple_putPoolRetirement_single_readPoolRetirement
             mapM_ (uncurry putPoolRetirement) certificatePublications
         mRetrievedCertificatePublication <-
             run $ atomically $ readPoolRetirement sharedPoolId
+        poolsMarkedToRetire <-
+            run $ atomically $ listRetiredPools $ EpochNo maxBound
         monitor $ counterexample $ unlines
             [ "\nExpected certificate publication: "
             , show mExpectedCertificatePublication
@@ -588,6 +590,12 @@ prop_multiple_putPoolRetirement_single_readPoolRetirement
         assertWith "retrieved certificate matches expectations" $ (==)
             mRetrievedCertificatePublication
             mExpectedCertificatePublication
+        assertWith "pool is marked to retire at the correct epoch" $
+            case mRetrievedCertificatePublication of
+                Nothing ->
+                    null poolsMarkedToRetire
+                Just (_publicationTime, retirementCert) ->
+                    poolsMarkedToRetire == [retirementCert]
 
     certificatePublications
         :: [(CertificatePublicationTime, PoolRetirementCertificate)]