hspec-core-2.4.0: test/Mock.hs

module Mock where

import           Prelude ()
import           Test.Hspec.Core.Compat

newtype Mock = Mock (IORef Int)

newMock :: IO Mock
newMock = Mock <$> newIORef 0

mockAction :: Mock -> IO ()
mockAction (Mock ref) = modifyIORef ref succ

mockCounter  :: Mock -> IO Int
mockCounter  (Mock ref) = readIORef ref