fix hyperlane gas tests (#1359)

* fix hyperlane gas tests
kadena-io · Apr 2, 2024 · ba23e50 · ba23e50
1 parent 4245018
commit ba23e50
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/src/Pact/GasModel/GasTests.hs b/src/Pact/GasModel/GasTests.hs
@@ -42,7 +42,8 @@ import Pact.JSON.Legacy.Value
 
 -- | Gas benchmark tests for Pact native functions
 allNatives :: [NativeDefName]
-allNatives = map fst (concatMap snd natives) <> nonNatives
+allNatives = map fst (concatMap snd pact412Natives) <> nonNatives
+
 
 -- | Non-native concepts to benchmark
 nonNatives :: [NativeDefName]

diff --git a/src/Pact/Native.hs b/src/Pact/Native.hs
@@ -21,6 +21,9 @@
 module Pact.Native
     ( natives
     , nativeDefs
+    , hyperlaneDefs
+    , hyperlaneAmendedDefs
+    , pact412Natives
     , pact412NativeDefs
     , moduleToMap
     , distinctDef
@@ -143,10 +146,13 @@ natives =
 
 -- | Production native modules as a dispatch map.
 nativeDefs :: HM.HashMap Text Ref
-nativeDefs = mconcat $ map moduleToMap (hyperlaneDefs:natives)
+nativeDefs = mconcat $ map moduleToMap natives
+
+pact412Natives :: [NativeModule]
+pact412Natives = hyperlaneAmendedDefs:natives
 
 pact412NativeDefs :: HM.HashMap Text Ref
-pact412NativeDefs = mconcat $ map moduleToMap $ hyperlaneAmendedDefs:natives
+pact412NativeDefs = foldMap moduleToMap pact412Natives
 
 moduleToMap :: NativeModule -> HM.HashMap Text Ref
 moduleToMap = HM.fromList . map (asString *** Direct) . snd

diff --git a/tests/GasModelSpec.hs b/tests/GasModelSpec.hs
@@ -89,9 +89,10 @@ untestedNativesCheck = do
      , "verify-spv"
      , "public-chain-data"
      , "dec"
+     , "hyperlane-decode-token-message"
+     , "hyperlane-token-erc20","hyperlane-token-msg"
      , "list"
      , "continue"
-     , "hyperlane-decode-token-message"
      ])
 
 allGasTestsAndGoldenShouldPass :: Spec
@@ -138,14 +139,16 @@ goldenSizeOfPactValues = do
 allNativesInGasTable :: Spec
 allNativesInGasTable = do
   it "all native functions should be in gas table" $ do
-    let justNatives = map (asString . fst) (concatMap snd natives)
+    let justNatives = map (asString . fst) (concatMap snd pact412Natives)
         absent li name = case (Map.lookup name defaultGasTable) of
           Nothing -> name : li
           Just _ -> li
         absentNatives = foldl' absent [] justNatives
     (S.fromList absentNatives)
     `shouldBe`
-    (S.fromList ["CHARSET_ASCII", "CHARSET_LATIN1", "public-chain-data", "list"])
+    (S.fromList ["CHARSET_ASCII", "CHARSET_LATIN1"
+                ,"hyperlane-token-erc20", "hyperlane-token-msg"
+                ,"public-chain-data", "list"])
 
 -- | Use this to run a single named test.
 _runNative :: NativeDefName -> IO (Maybe [(T.Text,Gas)])