feat: added Enum, Bounded instance of coin selection strategy typ…

…e to better enumerate through it Related to #294
geniusyield · Apr 27, 2024 · 767b8c8 · 767b8c8
1 parent 3105957
commit 767b8c8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/GeniusYield/Transaction/CoinSelection.hs b/src/GeniusYield/Transaction/CoinSelection.hs
@@ -94,7 +94,7 @@ data GYCoinSelectionStrategy
     = GYLargestFirstMultiAsset
     | GYRandomImproveMultiAsset
     | GYLegacy
-    deriving stock (Eq, Show)
+    deriving stock (Eq, Show, Enum, Bounded)
 
 {- | Select additional inputs from the set of own utxos given, such that when combined with given existing inputs,
 they cover for all the given outputs, as well as extraLovelace.

diff --git a/tests-privnet/GeniusYield/Test/Privnet/Stake.hs b/tests-privnet/GeniusYield/Test/Privnet/Stake.hs
@@ -58,7 +58,7 @@ withdrawRewardsSteps strat user@User{..} rewards info ctx = do
 tests :: IO Setup -> TestTree
 tests setup = testGroup "stake"
   [ testCaseSteps "exercising stake credential registration, delegation, rewards claiming & de-registration" $ \info -> withSetup setup info $ \ctx -> do
-    for_ [GYRandomImproveMultiAsset, GYLegacy] $ \strat -> do
+    for_ [minBound .. maxBound] $ \strat -> do
       newUser <- registerStakeCredentialSteps strat info ctx
       sps <- ctx & ctxGetParams & gyGetStakePools'
       info $ "Total stake pools: " <> show sps <> "\n"

diff --git a/tests/GeniusYield/Test/Stake.hs b/tests/GeniusYield/Test/Stake.hs
@@ -25,7 +25,7 @@ stakeTests config =
           -- Check if the withdrawal amount is positive.
           stakeAddrInfo <- gyGetStakeAddressInfo stakeAddr
           assertBool "No positive rewards available for withdrawal" $ gyStakeAddressInfoAvailableRewards stakeAddrInfo > 0
-          for_ [GYRandomImproveMultiAsset, GYLegacy] $ \strat ->
+          for_ [minBound .. maxBound] $ \strat ->
             testWithdrawalWithStrategy strat stakeAddrInfo addr stakeAddr config provider
     ]