Only move genScriptRegistry function

input-output-hk · Feb 6, 2023 · 2f3442c · 2f3442c
1 parent 31855b6
commit 2f3442c
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/hydra-node/src/Hydra/Chain/Direct/ScriptRegistry.hs b/hydra-node/src/Hydra/Chain/Direct/ScriptRegistry.hs
@@ -62,6 +62,26 @@ data ScriptRegistry = ScriptRegistry
   }
   deriving (Eq, Show, Generic, ToJSON, FromJSON)
 
+genScriptRegistry :: Gen ScriptRegistry
+genScriptRegistry = do
+  txId <- arbitrary
+  txOut <- genTxOutAdaOnly
+  pure $
+    ScriptRegistry
+      { initialReference =
+          ( TxIn txId (TxIx 0)
+          , txOut{txOutReferenceScript = mkScriptRef Initial.validatorScript}
+          )
+      , commitReference =
+          ( TxIn txId (TxIx 1)
+          , txOut{txOutReferenceScript = mkScriptRef Commit.validatorScript}
+          )
+      , headReference =
+          ( TxIn txId (TxIx 2)
+          , txOut{txOutReferenceScript = mkScriptRef Head.validatorScript}
+          )
+      }
+
 data NewScriptRegistryException = MissingScript
   { scriptName :: Text
   , scriptHash :: ScriptHash
@@ -147,26 +167,6 @@ queryScriptRegistry networkId nodeSocket txId = do
  where
   candidates = [TxIn txId ix | ix <- [TxIx 0 .. TxIx 10]] -- Arbitrary but, high-enough.
 
-genScriptRegistry :: Gen ScriptRegistry
-genScriptRegistry = do
-  txId <- arbitrary
-  txOut <- genTxOutAdaOnly
-  pure $
-    ScriptRegistry
-      { initialReference =
-          ( TxIn txId (TxIx 0)
-          , txOut{txOutReferenceScript = mkScriptRef Initial.validatorScript}
-          )
-      , commitReference =
-          ( TxIn txId (TxIx 1)
-          , txOut{txOutReferenceScript = mkScriptRef Commit.validatorScript}
-          )
-      , headReference =
-          ( TxIn txId (TxIx 2)
-          , txOut{txOutReferenceScript = mkScriptRef Head.validatorScript}
-          )
-      }
-
 publishHydraScripts ::
   -- | Expected network discriminant.
   NetworkId ->