diff --git a/src/ATrade/Types.hs b/src/ATrade/Types.hs index 6397979..9c4b2c8 100644 --- a/src/ATrade/Types.hs +++ b/src/ATrade/Types.hs @@ -49,7 +49,7 @@ import System.ZMQ4.ZAP type TickerId = T.Text data DataType = Unknown - | Price + | LastTradePrice | OpenInterest | BestBid | BestOffer @@ -62,7 +62,7 @@ data DataType = Unknown instance Enum DataType where fromEnum x - | x == Price = 1 + | x == LastTradePrice = 1 | x == OpenInterest = 3 | x == BestBid = 4 | x == BestOffer = 5 @@ -75,7 +75,7 @@ instance Enum DataType where | otherwise = -1 toEnum x - | x == 1 = Price + | x == 1 = LastTradePrice | x == 3 = OpenInterest | x == 4 = BestBid | x == 5 = BestOffer diff --git a/test/ArbitraryInstances.hs b/test/ArbitraryInstances.hs index 0e9c7f2..8af7684 100644 --- a/test/ArbitraryInstances.hs +++ b/test/ArbitraryInstances.hs @@ -120,4 +120,5 @@ instance Arbitrary BrokerServerResponse where | t == 4 -> ResponseError <$> arbitrary instance Arbitrary P.Price where - arbitrary = P.Price <$> arbitrary + arbitrary = P.Price <$> (arbitrary `suchThat` (\p -> abs p < 1000000000 * 10000000)) +