Haskell管道和HSpec测试

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

我有一系列类型为

管道输入输出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)

---

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

顺便说一句，您的

管道

正在使用执行惰性I/O的

readFile

。惰性I/O和管道之类的流媒体库混合得不好，事实上后者主要作为前者的替代品存在

也许您应该改用执行严格I/O的函数，比如

openFile

和

getLine

---

一个恼人的严格的I/O是迫使你更仔细地考虑资源分配。如何确保每个文件句柄在结束时关闭，或者在出现错误时关闭？实现这一点的一种可能方法是在monad中工作，而不是直接在

IO

中工作。诀窍是使用管道

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)

导入管道
进口合格管道。前奏部分为P
进口测试
data Person=人员字符串Int |未知派生（显示，等式）
数据分类=朋友|敌人|未决定的推导（显示，等式）
分类：：管道人员（人员，分类）IO（）
分类=做
p@（人名）收益率（p，朋友）
“Bob”->收益率（p，Foe）
_->产量（p，未决定）

该测试使用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

main=hspec$do
描述“分类”$do
它“正确地找到朋友”$do
[（p，cl）]->分类
p`shouldBe`（人称“Alex”31）
我应该是你的朋友

---

注意，您不必使用

每个

，这可以是一个简单的生产者，通过调用

产量

将管道转换为生产者或效果，然后使用或简单地使用

运行效果

来具体化值。显然，您必须决定如何创建测试数据并将其提供给管道。

runEffect

只会给我一个

mr

，在本例中是

IO（）

。不确定这会有什么帮助？不是

runEffect classify

，而是

runEffect（giveDataToClassify classify）

-就像我说的，你的管道接受输入，你必须决定输入是什么，只需以适当的方式将管道与创建输出而不需要输入的管道组合即可（我认为在

pipes

中这是一个

Producer

）。例如，

toListM$mapM\yield[Person“Bob”10，Person“June”20]>->分类

做你想做的事吗？注意

\xs->toListM$mapM\yield xs>->分类的类型是
[Person]>IO[（Person，Classification）]
在我看来，这似乎是一个与HSpec兼容的表单。我知道我需要将数据馈送到管道。我的问题是
runEffect
调用位于
IO
monad中，它无法计算为HSpec的
it
所需的
期望值您有一个
示例期望
（即
示例（IO（））
）的实例，其语义是抛出一个
HUnitFailure
异常表示测试失败，抛出一个
结果
异常（似乎？）表示测试成功符合要求。我觉得奇怪的是，没有
示例a=>示例（IO a）
实例，甚至没有
示例（IO结果）
-似乎这些对您很有用。也许您应该尝试自己编写它们？我的代码对IO没有类似的功能（实际上是DB调用等）。本问题中IO的概念仅用于演示目的，根本不是问题的目的。我想知道我是否可以使用HSpec测试管道代码，以及如何使用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