Use SinglePoolCertificateSequence in prop_readPoolLifeCycleStatus.

cardano-foundation · Aug 13, 2020 · 5c351d7 · 5c351d7
1 parent fd9dc69
commit 5c351d7
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/lib/core/test/unit/Cardano/Pool/DB/Properties.hs b/lib/core/test/unit/Cardano/Pool/DB/Properties.hs
@@ -614,11 +614,10 @@ prop_multiple_putPoolRetirement_single_readPoolRetirement
 --
 prop_readPoolLifeCycleStatus
     :: DBLayer IO
-    -> PoolId
-    -> [PoolCertificate]
+    -> SinglePoolCertificateSequence
     -> Property
 prop_readPoolLifeCycleStatus
-    DBLayer {..} sharedPoolId certificatesManyPoolIds =
+    DBLayer {..} (SinglePoolCertificateSequence sharedPoolId certificates) =
         monadicIO (setup >> prop)
   where
     setup = run $ atomically cleanDB
@@ -669,8 +668,6 @@ prop_readPoolLifeCycleStatus
         , ii <- [0 .. 3]
         ]
 
-    certificates = setPoolId sharedPoolId <$> certificatesManyPoolIds
-
     toRegistrationCertificate = \case
         Registration cert -> Just cert
         Retirement _ -> Nothing
@@ -685,12 +682,6 @@ prop_readPoolLifeCycleStatus
         Retirement cert ->
             putPoolRetirement cpt cert
 
-    setPoolId newPoolId = \case
-        Registration cert -> Registration
-            $ set #poolId newPoolId cert
-        Retirement cert -> Retirement
-            $ set #poolId newPoolId cert
-
 prop_rollbackRegistration
     :: DBLayer IO
     -> SlotNo