Use arbitraryBoundedEnum to avoid repeating the definition of `Sort…

…Order`.
cardano-foundation · Aug 1, 2019 · a99a8c9 · a99a8c9
1 parent 4f00014
commit a99a8c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/core/test/unit/Cardano/Wallet/DB/StateMachine.hs b/lib/core/test/unit/Cardano/Wallet/DB/StateMachine.hs
@@ -107,6 +107,7 @@ import Test.QuickCheck
     , Args (..)
     , Gen
     , Property
+    , arbitraryBoundedEnum
     , collect
     , elements
     , frequency
@@ -538,7 +539,7 @@ generator (Model _ wids) = Just $ frequency $ fmap (fmap At) <$> concat
     genPrivKey = elements ["pk1", "pk2", "pk3"]
 
     genSortOrder :: Gen SortOrder
-    genSortOrder = QC.elements [Ascending, Descending]
+    genSortOrder = arbitraryBoundedEnum
 
     genRange :: Gen (Range SlotId)
     genRange = Range <$> genSId <*> genSId