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来提供＆＃34;供应＆＃34;测试查询时使用的getSomething数据或转换。

编辑：在hspec中使用示例。

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

1 个答案: