IO 작업 조롱 : getArgs 및 putStrLn

Question

명령 줄 인수를 사용하여 화면에 출력하는 작은 함수 (또는 IO 동작)를 테스트하려고합니다. 내 원래 (검증 할) 기능은 다음과 같습니다

-- In Library.hs 
module Library where 

import System.Environment (getArgs) 

run :: IO() 
run = do 
    args <- getArgs 
    putStrLn $ head args 

this answer about mocking보고 후, 나는 타입 클래스 제약 형식을 사용하여 getArgs 및 putStrLn을 조롱 수있는 방법을 마련했다. 그래서 위의 함수가된다 :

-- In Library.hs 
module Library where 

class Monad m => SystemMonad m where 
    getArgs :: m [String] 
    putStrLn :: String -> m() 

instance SystemMonad IO where 
    getArgs = System.Environment.getArgs 
    putStrLn = Prelude.putStrLn 

run :: SystemMonad m => m() 
run = do 
    args <- Library.getArgs 
    Library.putStrLn $ head args 

이 Library., Prelude.

및 System.Environment.이 Ambigious Occurence의 컴파일러 불만을 방지한다. 내 테스트 파일은 다음과 같습니다.

-- In LibrarySpec.hs 
{-# LANGUAGE TypeSynonymInstances #-} 
{-# LANGUAGE FlexibleInstances #-} 

import Library 
import Test.Hspec 
import Control.Monad.State 

data MockArgsAndResult = MockArgsAndResult [String] String 
    deriving(Eq, Show) 

instance SystemMonad (State MockArgsAndResult) where 
    getArgs = do 
     MockArgsAndResult args _ <- get 
     return args 
    putStrLn string = do 
     MockArgsAndResult args _ <- get 
     put $ MockArgsAndResult args string 
     return() 

main :: IO() 
main = hspec $ do 
    describe "run" $ do 
    it "passes the first command line argument to putStrLn" $ do 
     (execState run (MockArgsAndResult ["first", "second"] "")) `shouldBe` (MockArgsAndResult ["first", "second"] "first") 

효과적으로 2 개의 필드를 포함하는 State 모나드를 사용하고 있습니다.

1. 모의 getArgs이 모의 putStrLn가 전달 된 것을두고 있음을
2. 에서 문자열을 읽는 명령 행 인수에 대한 목록입니다.

위의 코드는 작동하며 내가 테스트하고 싶은 것을 테스트하는 것으로 보입니다. 그러나,이 더 나은/청소기/더 많은 관용적 인 방법이 있는지 궁금하네요. 한 가지는, 같은 상태를 사용하여 테스트에 물건을 넣고 (내 가짜 명령 행 인수) putStrLn에 전달 된 내용입니다.

더 좋은 방법이 있습니까? 내가하고있는 일을하고 있니? 나는 Javascript 환경에서 조롱하는 것에 더 익숙하다. 하스켈에 대한 나의 지식은 꽤 기초적이다. (나는 실제적인 이해가 아니라 시행 착오의 공정한 비트에 의해 위의 해결책에 도달했다.)

Answer 1

더 좋은 방법은 순수 함수로 계산의 마음을 분리하여 getArgs 및 putStrLn의 모의 버전을 제공 할 필요 방지하는 것입니다.

공동 이 예 nsider :

main = do 
    args <- getArgs 
    let n = length $ filter (\w -> length w < 5) args 
    putStrLn $ "Number of small words: " ++ show n 

하나는 계산의 마음이 유형 [String] -> Int의 순수 기능입니다 작은 단어의 수를 세는 것을 말할 수 있습니다.

main = do 
    args <- getArgs 
    let n = countSmallWords args 
    putStrLn $ "Number of small words: " ++ show n 

countSmallWords :: [String] -> Int 
countSmallWords ws = ... 

지금 우리가 countSmallWords을 테스트하고 순수한 기능이 있기 때문에이 용이하다 : 이것은 우리가이 같은 프로그램을 리팩토링해야하는 것이 좋습니다.