From 611e6fa2a6c6af9622dc9afeeade8589cfd0ba34 Mon Sep 17 00:00:00 2001
From: Emily Martins <emi@haskell.fyi>
Date: Tue, 15 Mar 2022 17:25:44 +0100
Subject: [PATCH] move `scriptTest` to Util module

---
 agora-test/Spec/Stake.hs | 15 +++------------
 agora-test/Spec/Util.hs  | 25 +++++++++++++++++++++++++
 agora.cabal              |  2 ++
 3 files changed, 30 insertions(+), 12 deletions(-)
 create mode 100644 agora-test/Spec/Util.hs

diff --git a/agora-test/Spec/Stake.hs b/agora-test/Spec/Stake.hs
index 75d63c1..c3ba408 100644
--- a/agora-test/Spec/Stake.hs
+++ b/agora-test/Spec/Stake.hs
@@ -7,13 +7,11 @@ import Prelude
 --------------------------------------------------------------------------------
 
 import Test.Tasty (TestTree, testGroup)
-import Test.Tasty.HUnit (assertFailure, testCase)
 
 --------------------------------------------------------------------------------
 
 import Plutarch (compile)
-import Plutarch.Evaluate (evalScript)
-import Plutus.V1.Ledger.Scripts (Script)
+import Plutarch.Builtin (pforgetData)
 
 --------------------------------------------------------------------------------
 
@@ -21,10 +19,11 @@ import Agora.Stake (stakePolicy)
 
 --------------------------------------------------------------------------------
 
-import Plutarch.Builtin (pforgetData)
 import Spec.Sample.Stake qualified as Stake
+import Spec.Util (scriptTest)
 
 --------------------------------------------------------------------------------
+
 tests :: [TestTree]
 tests =
   [ testGroup
@@ -32,11 +31,3 @@ tests =
       [ scriptTest "minting" (compile $ stakePolicy Stake.stake # pforgetData (pconstantData ()) # pconstant Stake.stakeCreation)
       ]
   ]
-
-scriptTest :: String -> Script -> TestTree
-scriptTest name script = testCase name $ do
-  let (res, _budget, traces) = evalScript script
-  case res of
-    Left e -> do
-      assertFailure (show e <> " Traces: " <> show traces)
-    Right _v -> pure ()
diff --git a/agora-test/Spec/Util.hs b/agora-test/Spec/Util.hs
new file mode 100644
index 0000000..9171641
--- /dev/null
+++ b/agora-test/Spec/Util.hs
@@ -0,0 +1,25 @@
+module Spec.Util (scriptTest) where
+
+--------------------------------------------------------------------------------
+
+import Prelude
+
+--------------------------------------------------------------------------------
+
+import Test.Tasty (TestTree)
+import Test.Tasty.HUnit (assertFailure, testCase)
+
+--------------------------------------------------------------------------------
+
+import Plutarch.Evaluate (evalScript)
+import Plutus.V1.Ledger.Scripts (Script)
+
+--------------------------------------------------------------------------------
+
+scriptTest :: String -> Script -> TestTree
+scriptTest name script = testCase name $ do
+  let (res, _budget, traces) = evalScript script
+  case res of
+    Left e -> do
+      assertFailure (show e <> " Traces: " <> show traces)
+    Right _v -> pure ()
diff --git a/agora.cabal b/agora.cabal
index 7dad5f8..8d17d47 100644
--- a/agora.cabal
+++ b/agora.cabal
@@ -152,6 +152,8 @@ test-suite agora-test
     Spec.Sample.Stake
     Spec.Stake
 
+    Spec.Util
+
   build-depends:
     , agora