Question

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

我有一系列Pipe In Out IO ()类型的函数（例如），我想用HSpec测试。我怎么能这样做？

例如，假设我有这个域：

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

和这个管道：

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 monad，而不是直接使用IO。

Answer 2

诀窍是使用来自Pipes toListM monad transformer的ListT。

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。

Haskell管道和HSpec测试

2 个答案: