通常导出大规模代数数据类型的任意?

时间:2016-06-15 18:29:13

标签: haskell generics quickcheck

我有一个我打字的协议:

data ProtocolPacket
  = Packet1 Word8 Text Int8
  | Packet2 Text
  | Packet3 Int Text Text Text Text
  | Packet4 Int Double Double Double Int16 Int16 Int16
  ...
  deriving (Show,Eq)

此外,我还为每个数据包实现了序列化/反序列化代码。当然,我想在Quickcheck中测试这个协议,并确保对任何输入组合的任何数据包进行序列化和反序列化都会让我准确地回复我的内容。所以我继续为Arbitrary实现这些数据包。像这样输入类:

instance Arbitrary ProtocolPacket where
  arbitrary = do
  packetID <- choose (0x00,...) :: Gen Word8
  case packetID of
    0x00 -> do
      a <- arbitrary
      b <- arbitrary
      c <- arbitrary
      return $ Packet1 a b c
    0x01 -> do
      a <- arbitrary
      return $ Packet2 a
    0x02 -> do
      a <- arbitrary
      b <- arbitrary
      c <- arbitrary
      d <- arbitrary
      e <- arbitrary
      return $ Packet3 a b c d e
    0x03 -> do
      a <- arbitrary
      b <- arbitrary
      c <- arbitrary
      d <- arbitrary
      e <- arbitrary
      f <- arbitrary
      g <- arbitrary
      return $ Packet4 a b c d e f g
    ...

假设我已经开始并为所有相关的数据构造函数参数定义了Arbitrary,这些参数没有开箱即用的Arbitrary,这些代码需要手工完成由我写的,以便用有意义的数据填充数据包字段。但那就是它。

但是正如你所看到的,我正在重复自己很多来做一些只是咕噜咕噜的工作。这是我实际处理的一小部分样本。理想情况下,我希望能够做到这一点:

{-# LANGUAGE DeriveGeneric #-}
import GHC.Generics

data ProtocolPacket
  = Packet1 Word8 Text Int8
  | Packet2 Text
  | Packet3 Int Text Text Text Text
  | Packet4 Int Double Double Double Int16 Int16 Int16
  ...
  deriving (Show,Eq,Generic)

instance Arbitrary ProtocolPacket

就像我可以使用FromJSONToJSON一样,但这并不起作用。有没有一种方法呢?

1 个答案:

答案 0 :(得分:6)

Daniel Wagner在评论中提到generic-random能够做到这一点。这是我正在寻找的图书馆,但文档并没有让我明白是这样的。在撰写本文时,Brent Yorgey在他的博客上发布了一个非常清晰的教程,详细介绍了如何使用generic-random来做我所询问的内容等等。 The blog post can be found here.

就我而言,解决方案很简单。使用Generic.Random.Generic中的generic-random

{-# LANGUAGE DeriveGeneric #-}
import Generic.Random.Generic
import GHC.Generics

data ProtocolPacket
  = Packet1 Word8 Text Int8
  | Packet2 Text
  | Packet3 Int Text Text Text Text
  | Packet4 Int Double Double Double Int16 Int16 Int16
  ...
  deriving (Show,Eq,Generic)

instance Arbitrary ProtocolPacket where
  arbitrary = genericArbitrary