Also test with mainnet fee policy in prop_distributeSurplus_onSuccess.

In response to review feedback: #3238 (comment)
cardano-foundation · Apr 20, 2022 · 38be46b · 38be46b
1 parent 8eccfc5
commit 38be46b
Showing 1 changed file with 21 additions and 10 deletions.
diff --git a/lib/shelley/test/unit/Cardano/Wallet/Shelley/TransactionSpec.hs b/lib/shelley/test/unit/Cardano/Wallet/Shelley/TransactionSpec.hs
@@ -2434,16 +2434,27 @@ prop_distributeSurplusDelta_coversCostIncreaseAndConservesSurplus
 --------------------------------------------------------------------------------
 
 instance Arbitrary FeePolicy where
-    arbitrary = do
-        intercept <- frequency
-            [ (1, pure 0.0)
-            , (3, getPositive <$> arbitrary)
-            ]
-        slope <- frequency
-            [ (1, pure 0.0)
-            , (3, getPositive <$> arbitrary)
-            ]
-        pure $ LinearFee LinearFunction {intercept, slope}
+    arbitrary = frequency
+        [ (1, feePolicyMainnet)
+        , (7, feePolicyGeneral)
+        ]
+      where
+        feePolicyMainnet :: Gen FeePolicy
+        feePolicyMainnet = pure $ LinearFee $ LinearFunction
+            {intercept = 150_000, slope = 44}
+
+        feePolicyGeneral :: Gen FeePolicy
+        feePolicyGeneral = do
+            intercept <- frequency
+                [ (1, pure 0.0)
+                , (3, getPositive <$> arbitrary)
+                ]
+            slope <- frequency
+                [ (1, pure 0.0)
+                , (3, getPositive <$> arbitrary)
+                ]
+            pure $ LinearFee LinearFunction {intercept, slope}
+
     shrink (LinearFee LinearFunction {intercept, slope}) =
         LinearFee . uncurry LinearFunction
             <$> shrink (intercept, slope)