Haskell管道和HSpec测试

Question

我为一个使用Pipes的项目写了一个程序，我很喜欢!然而，我正在努力对代码进行单元测试。

我有一系列类型为Pipe In Out IO ()的函数(例如)，我希望用HSpec进行测试。我该怎么做呢?

例如，假设我有这样一个域:

data Person = Person String Int | Unknown deriving (Show, Eq)
data Classification = Friend | Foe | Undecided deriving Show

和this Pipe:

classify :: Pipe Person (Person, Classification) IO ()
classify = do
    p@(Person name _) <- await
    case name of 
      "Alex" -> yield (p, Friend)
      "Bob" -> yield (p, Foe)
      _ -> yield (p, Undecided)

我想写一个规范:

main = hspec $ do
  describe "readFileP" $ 
    it "yields all the lines of a file"
      pendingWith "How can I test this Pipe? :("

Answer 1

您可以使用temporary包的函数创建包含预期数据的临时文件，然后测试数据是否被管道正确读取。

顺便说一下，您的Pipe正在使用执行惰性I/O的readFile。惰性I/O和像管道这样的流库不能很好地混合在一起，事实上，后者主要是作为前者的替代品而存在的!

也许你应该使用执行严格I/O的函数，如openFile和getLine。

严格I/O的一个烦恼是它迫使您更仔细地考虑资源分配。如何确保每个文件句柄在结束时关闭，或者在发生错误的情况下?实现这一目标的一种可能方法是在ResourceT IO单子中工作，而不是直接在IO中工作。

Answer 2

技巧是使用Pipes的ListT单子转换器中的toListM。

import Pipes
import qualified Pipes.Prelude as P
import Test.Hspec
data Person = Person String Int | Unknown deriving (Show, Eq)
data Classification = Friend | Foe | Undecided deriving (Show, Eq)
classify :: Pipe Person (Person, Classification) IO ()
classify = do
  p@(Person name _) <- await
  case name of 
    "Alex" -> yield (p, Friend)
    "Bob" -> yield (p, Foe)
    _ -> yield (p, Undecided)

测试，使用ListT转换器将管道转换为ListT并使用HSpec进行断言:

main = hspec $ do
  describe "classify" $ do
    it "correctly finds friends" $ do
      [(p, cl)] <- P.toListM $ each [Person "Alex" 31] >-> classify
      p `shouldBe` (Person "Alex" 31)
      cl `shouldBe` Friend

注意，您不必使用each，这可以是一个调用yield的简单生成器。