使用 Hspec 对 IO 操作进行单元测试

Question

我发现了类似问题的其他问题，但在这个特定场景中没有任何问题可以回答我的问题。此外，Haskell.

中似乎很少有资源简洁地涵盖单元测试 IO 操作的主题

假设我的数据库通信有这个类型类：

data Something = Something String deriving Show

class MonadIO m => MonadDB m where
  getSomething :: String -> m Something
  getSomething s = do
    ... -- assume a DB call is made and an otherwise valid function

instance MonadDB IO

以及使用它的函数：

getIt :: MonadDB m => m (Int, Something)
getIt = do
  s@(Something str) <- getSomething "hi"
  return (length str, s) -- excuse the contrived example

我想用 hspec 测试这个 getIt 函数但不与数据库对话，这可能意味着替换它使用的 MonadDB，但我如何实现它？

Answer 1

这对你有用吗？

#!/usr/bin/env stack
-- stack exec --package transformers --package hspec -- ghci
import Control.Monad.IO.Class
import Control.Monad.Trans.Identity

import Data.Char
import Test.Hspec

data Something = Something String deriving (Eq, Show)

class MonadIO m => MonadDB m where
  getSomething :: String -> m Something
  getSomething s = return $ Something (map toUpper s)

instance MonadDB IO

instance MonadIO m => MonadDB (IdentityT m)

getIt :: MonadDB m => m (Int, Something)
getIt = do
  s@(Something str) <- getSomething "hi"
  return (length str, s)

main :: IO ()
main = hspec $ do
  describe "Some tests" $ do
    it "test getIt" $ do
      runIdentityT getIt `shouldReturn` (2, Something "HI")

    it "test getIt should fail" $ do
      runIdentityT getIt `shouldReturn` (1, Something "HI")

您也可以使用 ReaderT 或 StateT 到 "supply" 数据或 getSomething 的转换以在测试查询时使用。

编辑：hspec 中的示例使用。

使用 Hspec 对 IO 操作进行单元测试

Unit testing IO actions with Hspec

haskell

hspec