benchmarking

agora-bench/Main.hs

@@ -1,42 +1,33 @@
 module Main (main) where
 
-import Agora.AuthorityToken (authorityTokenPolicy)
-import Agora.Effect.TreasuryWithdrawal (treasuryWithdrawalValidator)
-import Agora.Governor (Governor (..))
-import Agora.Governor.Scripts (governorPolicy, governorValidator)
-import Agora.Proposal.Scripts (proposalPolicy, proposalValidator)
-import Agora.Stake.Scripts (stakePolicy, stakeValidator)
-import Agora.Treasury (treasuryValidator)
-import Bench
-import Data.Foldable (for_)
-import Plutus.V1.Ledger.Api (CurrencySymbol)
-import Sample.Shared
+import Bench (specificationTreeToBenchmarks)
+import Spec.AuthorityToken qualified as AuthorityToken
+import Spec.Effect.GovernorMutation qualified as GovernorMutation
+import Spec.Effect.TreasuryWithdrawal qualified as TreasuryWithdrawal
+import Spec.Governor qualified as Governor
+import Spec.Proposal qualified as Proposal
+import Spec.Spec (group)
+import Spec.Stake qualified as Stake
+import Spec.Treasury qualified as Treasury
 import Prelude
 
 --------------------------------------------------------------------------------
 
 main :: IO ()
 main = do
-  let benchmarks =
-        mconcat
-          [ -- GATs
-            benchmarkSize "authorityTokenPolicy" $ compile $ authorityTokenPolicy authorityToken
-          , -- Governor
-            benchmarkSize "governorValidator" $ compile $ governorValidator governor
-          , benchmarkSize "governorPolicy" $ compile $ governorPolicy governor
-          , -- Stake
-            benchmarkSize "stakeValidator" $ compile $ stakeValidator stake
-          , benchmarkSize "stakePolicy" $ compile $ stakePolicy governor.gtClassRef
-          , -- Proposal
-            benchmarkSize "proposalValidator" $ compile $ proposalValidator proposal
-          , benchmarkSize "proposalPolicy" $ compile $ proposalPolicy govAssetClass
-          , -- Treasury
-            benchmarkSize "treasuryValidator" $ compile $ treasuryValidator gatCS
-          , -- Effect validators
-            benchmarkSize "treasuryWithdrawalValidator" $ compile $ treasuryWithdrawalValidator gatCS
-          ]
-
-  for_ benchmarks print
-
-gatCS :: CurrencySymbol
-gatCS = "73475cb40a568e8da8a045ced110137e159f890ac4da883b6b17dc651b3a8049" -- arbitrary CS
+  mapM_ print $
+    specificationTreeToBenchmarks $
+      group
+        "Benchmark"
+        [ group
+            "Effects"
+            [ group "Treasury Withdrawal Effect" TreasuryWithdrawal.specs
+            , group "Governor Mutation Effect" GovernorMutation.specs
+            ]
+        , group "Stake" Stake.specs
+        , group "Proposal" Proposal.specs
+        , group "AuthorityToken" AuthorityToken.specs
+        , group "Treasury" Treasury.specs
+        , group "AuthorityToken" AuthorityToken.specs
+        , group "Governor" Governor.specs
+        ]