Fix golden tests.

input-output-hk · Mar 13, 2019 · 358269a · 358269a
1 parent 9d80227
commit 358269a
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/cardano-chain.cabal b/cardano-chain.cabal
@@ -51,6 +51,7 @@ library
                        Cardano.Chain.Common.AddressHash
                        Cardano.Chain.Common.Attributes
                        Cardano.Chain.Common.BlockCount
+                       Cardano.Chain.Common.BlockCount.JSON
                        Cardano.Chain.Common.ChainDifficulty
                        Cardano.Chain.Common.Lovelace
                        Cardano.Chain.Common.LovelacePortion

diff --git a/src/Cardano/Chain/Common/BlockCount.hs b/src/Cardano/Chain/Common/BlockCount.hs
@@ -9,11 +9,13 @@ where
 
 import Cardano.Prelude
 
-import Data.Aeson.TH (defaultOptions, deriveJSON)
+import Data.Aeson.TH (defaultOptions, deriveJSON, fieldLabelModifier)
 import Formatting.Buildable (Buildable)
 
 import Cardano.Binary.Class (Bi(..))
 
+import Cardano.Chain.Common.BlockCount.JSON (unToGet)
+
 newtype BlockCount = BlockCount
     { unBlockCount :: Word64
     } deriving ( Eq
@@ -34,4 +36,4 @@ instance Bi BlockCount where
     decode = BlockCount <$> decode
     encodedSizeExpr size pxy = size (unBlockCount <$> pxy)
 
-deriveJSON defaultOptions ''BlockCount
+deriveJSON (defaultOptions { fieldLabelModifier = unToGet }) ''BlockCount
diff --git a/src/Cardano/Chain/Common/BlockCount/JSON.hs b/src/Cardano/Chain/Common/BlockCount/JSON.hs
@@ -0,0 +1,9 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module Cardano.Chain.Common.BlockCount.JSON (unToGet) where
+
+import Cardano.Prelude
+
+unToGet :: [Char] -> [Char]
+unToGet "unBlockCount" = "getBlockCount"
+unToGet x = x