Skip to content

Repository files navigation

Mockcat Logo

Stub Haskell functions. Verify calls when needed.

Mockcat is a test double library for Haskell. Its two core functions are stub and mock.

Defining a stub takes just one line:

-- Stub: Returns True when given "a"
let f :: String -> Bool
    f = stub ("a" ~> True)

f "a"  -- => True

Write the argument on the left of ~> and the return value on the right. No verification—just a pure function.

When you need to verify calls, use mock. With expects, you can declare expectations before execution:

withMockIO $ do
  f <- mock ("a" ~> True)
    `expects` called once   -- Declare "should be called exactly once"

  f "a" `shouldBe` True
  -- Verification runs when exiting the withMockIO scope

Recommended usage:

  • If an ordinary value or lambda is enough, use it directly.
  • Need fixed return values? Use stub (pure, no verification).
  • Need to verify arguments, call counts, or order? Use mock.

Concepts & Terminology

Mockcat separates stubbing return values from verifying calls.

  • Stub: Returns configured values for expected arguments. Use it when you do not need to verify call history.

  • Mock: Adds call history to a stub, allowing you to verify that it was called as expected. Expectations can be verified after execution or declared before execution.


Why Mockcat?

Pure logic can be tested directly with values and functions, so it usually needs neither stubs nor mocks. When you do need a test double, Mockcat provides a small, straightforward way to define its behavior and verify calls when necessary.

Mockcat lets you declaratively describe function behavior and calls without depending on a specific architecture.

It works with ordinary functions, functions returning IO, functions passed as arguments, Service Handle fields, and MTL/Capability typeclasses.

Mockcat follows your architecture—not the other way around.

Before / After

Here is what the same test setup looks like with Mockcat.

Before: Handwritten... 😫 After: Mockcat 🐱✨
Definition (Stub)
"I want to return
this value for this arg"
f :: String -> IO String
f arg = case arg of
"a" -> pure "b"
_ -> error "unexpected"

Even simple branching consumes many lines.
-- Use stub if verification is unneeded (Pure)
let f = stub ("a" ~> "b")

Behaves as a completely pure function.
Verify
(Did it get called
correctly?)
-- Need manual recording mechanism
ref <- newIORef []
let f arg = do
modifyIORef ref (arg:)
...

-- Verification Logic
calls <- readIORef ref
calls shouldBe ["a"]

_ This is just one example. Boilerplate often grows._
withMock $ do
-- Declare expected values at definition
f <- mock ("a" ~> "b")
`expects` called once

-- Just execute (Automatic verification)

Recording is automatic.
Focus on the "Why" and "What", not the "How".

Key Features

  • Haskell Native DSL: No need to memorize redundant data constructors or specialized notation. Write test doubles naturally, just like function definitions (arg ~> return).
  • Architecture Agnostic: Whether using MTL (Typeclasses), Service Handle (Records), or functions—Mockcat adapts to the production design you already have.
  • Verify by "Condition", not just Value: Works even if arguments lack Eq instances. You can verify based on "what properties it should satisfy" (Predicates) rather than just strict equality.
  • Helpful Error Messages: Shows "structural diffs" on failure, highlighting exactly what didn't match.
    function was not called with the expected arguments.
    
      Closest match:
        expected: Record { name = "Alice", age = 20 }
         but got: Record { name = "Alice", age = 21 }
                                             ^^^
      Specific difference in `age`:
        expected: 20
         but got: 21
                  ^^
    
  • Intent-Driven Types: Types help express testing intent without imposing a particular architecture.

Quick Start

Copy and paste the code below to experience Mockcat right now.

Installation

package.yaml:

dependencies:
  - mockcat

Or .cabal:

build-depends:
    mockcat

First Test (No Verification)

import Test.Hspec
import Test.MockCat

spec :: Spec
spec = do
  it "stub demo" $ do
    let f :: String -> Int
        f = stub ("Hello" ~> 42)

    f "Hello" `shouldBe` 42

First Test (With Verification)

import Test.Hspec
import Test.MockCat

spec :: Spec
spec = do
  it "mock demo" $ do
    withMockIO $ do
      -- Declare "should be called exactly once"
      f <- mock ("Hello" ~> (42 :: Int))
        `expects` called once

      f "Hello" `shouldBe` 42
      -- Verification runs when exiting the withMockIO scope

User Guide

Mockcat supports two verification styles, depending on when you want to state the expectations.

1. Declarative Verification (withMock (withMockIO) / expects)

Declare expectations when defining the mock. Verification runs automatically when the scope exits. This keeps the mock definition and its expectations close together.

import Test.Hspec
import Test.MockCat
import Control.Monad.IO.Class (MonadIO(liftIO))

spec :: Spec
spec = do
  it "User Guide (withMock)" $ do
    withMock $ do
      -- Define a mock that returns True for "Hello"
      f <- mock ("Hello" ~> True)
        `expects` called once

      -- Execution
      let result = f "Hello"

      liftIO $ result `shouldBe` True

withMockIO: Simplified IO Testing

withMockIO is an IO-specialized version of withMock. It allows you to run IO actions directly within the mock context without needing liftIO.

import Test.Hspec
import Test.MockCat

spec :: Spec
spec = do
  it "User Guide (withMockIO)" $ do
    withMockIO $ do
      f <- mock ("Hello" ~> True)
        `expects` called once

      let result = f "Hello"

      result `shouldBe` True

Important

When using expects (declarative verification), you MUST wrap the mock definition in parentheses (...). The $ operator pattern used in previous versions (mock $ ... expects ...) will cause compilation errors due to precedence changes.

mock $ any ~> True expects ...mock (any ~> True) expects ...

Note

You can also use expects for declarative verification inside runMockT blocks. This works seamlessly with generated typeclass mocks as well.

runMockT do
  _readFile ("config.txt" ~> pure "value")
    `expects` called once

2. Mocking with Typeclass-Based Designs (makeMock)

For designs that express dependencies via typeclasses (MTL style or Capability pattern), Mockcat can generate mocks from those typeclasses using Template Haskell. This option is for designs that already use typeclasses; you do not need to introduce a typeclass just for testing.

{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeFamilies #-}

class Monad m => FileSystem m where
  readFile :: FilePath -> m String
  writeFile :: FilePath -> String -> m ()

-- [Strict Mode] Default behavior. Consistent with 'mock'.
-- If the return type is `m a`, the stub definition must return a value of type `m a` (e.g., `pure @IO "value"`, `throwIO Error`).
-- Recommended when you prefer explicit descriptions faithful to Haskell's type system.
makeMock [t|FileSystem|]

-- [Auto-Lift Mode] Convenience-focused mode.
-- Automatically wraps pure values into the monad (m String).
makeAutoLiftMock [t|FileSystem|]

Note

If the class definition requires additional extensions (e.g., MultiParamTypeClasses, UndecidableInstances), Mockcat will display a detailed error message during compilation to guide you.

Use runMockT block in your tests.

spec :: Spec
spec = do
  it "filesystem test" do
    result <- runMockT do
      -- [Strict Mode] (if using makeMock)
      _readFile $ "config.txt" ~> pure @IO "debug=true"
      _writeFile $ "log.txt" ~> "start" ~> pure @IO ()

      -- [Auto-Lift Mode] (if using makeAutoLiftMock)
      -- _readFile $ "config.txt" ~> "debug=true"

      -- Run code under test (mock injected)
      myProgram "config.txt"
    
    result `shouldBe` ()

3. Function Mocking and Post-Verification (mock / shouldBeCalled)

Create a mock, exercise the code under test, and then verify the recorded calls with shouldBeCalled. Use this style when verification reads more naturally after execution.

import Test.Hspec
import Test.MockCat

spec :: Spec
spec = do
  it "Function Mocking" $ do
    -- Define a mock that returns True for "Hello" (No 'expects' here)
    f <- mock ("Hello" ~> True)
    
    -- Execution
    f "Hello" `shouldBe` True

    -- Post-Verification (shouldBeCalled)
    f `shouldBeCalled` "Hello"

Warning

Limitation in HPC (Code Coverage) Environments Do not use shouldBeCalled when running tests with stack test --coverage or similar. The code coverage instrumentation by GHC wraps functions, which changes their identity and causes verification to fail. If you need code coverage, please use the expects style (Section 1).

Flexible Matching: You can specify conditions (predicates) instead of concrete values.

{-# LANGUAGE TypeApplications #-}
import Test.Hspec
import Test.MockCat
import Prelude hiding (any)

spec :: Spec
spec = do
  it "Matcher Examples" $ do
    -- Arbitrary string (param any)
    f <- mock (any @String ~> True)
    f "foo" `shouldBe` True

    -- Condition (when)
    g <- mock (when (> (5 :: Int)) "> 5" ~> True)
    g 6 `shouldBe` True

4. Flexible Verification (Matchers)

Even if arguments don't have Eq instances, or you don't want to depend on specific values, you can verify based on intent—"what condition should be met". Mockcat provides matchers for argument properties, not just exact value equality.

Allow Any Value (any)

-- Return True regardless of the argument
f <- mock (any ~> True)

-- Verify that it was called (arguments don't matter)
f `shouldBeCalled` any

Verify with Conditions (when)

You can use conditions (predicates) instead of exact values. This is useful for types without Eq (such as functions), or when checking partial matches.

-- Return False only if the argument starts with "error"
f <- mock do
  onCase $ when (\s -> "error" `isPrefixOf` s) "start with error" ~> False
  onCase $ any ~> True

If you don't need a label (description shown on error), you can use when_.

f <- mock (when_ (> 5) ~> True)

5. Advanced Features - [Advanced]

mock vs stub vs mockM

These functions differ in whether they record calls and where that recording takes place.

Function Verification (shouldBeCalled) IO Dependency Characteristics
stub None Pure Stub. No IO dependency. Sufficient if verification isn't needed.
mock None (External) Mock. Behaves as a pure function. Automatically records history.
mockM Yes (Explicit) Monadic Mock. Used within MockT or IO, allowing explicit handling of side effects (e.g., logging).

Choosing between mock and mockM

Choose between mock and mockM according to the return type of the target function.

  • mock (For Pure Functions):
    • Use this when mocking pure functions like String -> Int.
    • It respects Haskell's lazy evaluation and records the call only when the result is actually evaluated. This prevents counting unnecessary calls that were never executed.

Important

Since mock behaves as a pure function (a -> b), it is subject to GHC's optimizations (CSE / CAF / full laziness).

As a result, even if an expression appears multiple times in your source code, it may be evaluated only once after compilation.

Mockcat records and verifies the actual number of evaluations, not the number of times the expression appears in the source code.

  • mockM (For IO/Monadic Functions):
    • Use this when mocking functions that return IO or other MonadIO instances (such as ReaderT IO), like String -> IO Int.
    • Since recording is built directly into the returned action (IO), call counts remain predictable under concurrency and GHC optimizations.

Tip

If the function returns IO, use mockM. Otherwise, use mock.

Partial Mocking: Mixing with Real Functions

Useful when you want to replace only some methods with mocks while using real implementations for others.

-- [Strict Mode]
makePartialMock [t|FileSystem|]

-- [Auto-Lift Mode]
-- Just like makeAutoLiftMock, there is an Auto-Lift version for Partial Mock.
makeAutoLiftPartialMock [t|FileSystem|]

instance FileSystem IO where ... -- Real instance is also required

test = runMockT do
  _readFile $ "test" ~> pure @IO "content" -- Only mock readFile (Strict)
  -- or
  -- _readFile $ "test" ~> "content" -- (Auto-Lift)

  program -- writeFile runs the real IO instance

Derivation and Custom Instances

When using MockT, you might need to handle type classes that are not directly related to the side effects you are mocking. Mockcat provides macros to help with these cases.

MTL Instances (MonadReader, MonadError, etc.)

MockT provides standard mtl instances (MonadReader, MonadError, MonadState, MonadWriter) out of the box. These instances automatically lift operations to the base monad.

Custom Type Class Derivation (deriveMockInstances)

For custom "Capability" type classes (like MonadLogger, MonadConfig) that should just be lifted to the base monad, use deriveMockInstances.

class Monad m => MonadLogger m where
  logInfo :: String -> m ()

deriveMockInstances [t|MonadLogger|]

This generates an instance for MockT m that calls lift . logInfo.

Explicit No-op Instances (deriveNoopInstance)

Sometimes you want a mock to do nothing for certain methods (especially those returning m ()) without having to define explicit stubs or provide a base implementation.

class Monad m => MonadAuditor m where
  audit :: String -> m ()

deriveNoopInstance [t|MonadAuditor|]

This generates an instance for MockT m where audit simply returns pure ().


Sequential Responses

Cases are matched from top to bottom, like Haskell pattern matching. Only the first matching case is selected. Each case owns its response sequence, and only calls selecting that case advance it. The final response is repeated after the sequence is exhausted.

In short, onCase describes branching by input; andThen describes the sequence of responses for repeated calls to that case.

f <- mock do
  onCase $ "A" ~> 1
    `andThen` 2
    `andThen` 3
  onCase $ any @String ~> 9
    `andThen` 10
    `andThen` 11

-- Calls:   A, B, A, C, A, B, A, C
-- Results: 1, 9, 2, 10, 3, 11, 3, 11

Later overlapping cases are unreachable. To return consecutive values for the same condition, attach them to one case with andThen.

Monadic Return (IO a)

andThen also works with monadic return values when you want different effects or results for consecutive calls.

f <- mock do
  onCase $ "get" ~> pure @IO 1 -- 1st call
    `andThen` pure @IO 2        -- 2nd and later calls

Named Mocks

You can attach labels to display function names in error messages.

f <- mock (label "myAPI") ("arg" ~> True)

Encyclopedia (Feature Reference)

※ Use this section as a dictionary when you get stuck.

Declarative Verification DSL (expects)

In expects blocks, you can describe expectations declaratively using a builder-style syntax. It shares the same vocabulary as shouldBeCalled.

Basic Usage

Start with called and chain conditions.

-- Call count only
mock (any ~> True) `expects` called once

-- With arguments
mock (any ~> True) `expects` (called once `with` "arg")

-- Multiple expectations (in do block)
mock (any ~> True) `expects` do
  called once `with` "A"
  called once `with` "B"

Syntax Reference

Builder Description Example
called [Required] Starts the expectation builder. called ...
times n Expects exact call count. called . times 2
once Alias for times 1. called . once
never Expects 0 calls. called . never
with arg Expects specific argument(s). called with "value"
with matcher Uses a matcher for argument verification. called with when (>5) "gt 5"
inOrder Verify usage order (when used in a list) (See "Order Verification" section)

Verification Matchers (shouldBeCalled)

Matcher Description Example
x (Value itself) Was called with that value f `shouldBeCalled` (10 :: Int)
times n Exact count f `shouldBeCalled` (times 3 `with` "arg")
once Exactly once f `shouldBeCalled` (once `with` "arg")
never Never called f `shouldBeCalled` never
atLeast n n or more times f `shouldBeCalled` atLeast 2
atMost n n or fewer times f `shouldBeCalled` atMost 5
anything Any argument (count ignored) f `shouldBeCalled` anything
inOrderWith [...] Strict order f `shouldBeCalled` inOrderWith ["a", "b"]
inPartialOrderWith [...] Partial order (skips allowed) f `shouldBeCalled` inPartialOrderWith ["a", "c"]

Parameter Matchers (Definition)

Matcher Description Example
any Any value any ~> True
when pred label Condition when (>0) "positive" ~> True
when_ pred No label when_ (>0) ~> True

FAQ

Q. How are unevaluated lazy values handled? A. They are not counted. Mockcat records calls only "when the result is evaluated" (Honest Laziness). This prevents false positives from unneeded calculations.
Q. Can I use it in parallel tests? A. Yes. Internally uses `TVar` to count atomically, so it records accurately even when called in parallel via `mapConcurrently`, etc.
Q. Can I run tests with code coverage (HPC)? A. Yes (since v1.1.0.0). Mockcat's `expects` style is designed to be unaffected by the function wrapping performed by HPC, so it operates safely even under HPC. However, for the reasons mentioned above, we strongly recommend using the **`expects`** style (or `withMock`). The `shouldBeCalled` style cannot be used because HPC's mechanism makes it impossible to identify mock identity.
Q. What code does `makeMock` generate? A. It generates a `MockT m` instance for the specified typeclass, and stub generation function definitions named `_methodName` corresponding to each method.
Q. Isn't this strictly a Spy? A. Yes, according to definitions like xUnit Patterns, Mockcat's mocks which verify after execution are classified as **Test Spies**.
However, since many modern libraries (Jest, Mockito, etc.) group these under "Mock", and to avoid confusion from terminology proliferation, this library unifies them under the term **"Mock"**.
Q. Call counts are lower than expected in tests

A. Since mock is treated as a pure function, GHC's optimizations may cause evaluations to be shared. This is Mockcat's intended behavior—it accurately reflects the actual execution result after compilation.

Mockcat records "the actual number of evaluations" at runtime. Therefore, if the same expression is shared due to optimization, it is correctly counted as "1 call".

If you want to suppress evaluation sharing for testing purposes, you can add the following GHC pragmas to your test file.

{-# OPTIONS_GHC -fno-cse #-}
{-# OPTIONS_GHC -fno-full-laziness #-}

This is a test-only setting, useful when you want to verify behavior closer to the source-level call count.

Real-World Examples

Mockcat does not dictate production architecture. Use ordinary values and functions for pure code, and introduce Mockcat only at boundaries where a small, direct test double is useful.

Here are real-world test suites using mockcat:

Tips and Troubleshooting

Name collision with Prelude.any

The any parameter matcher from Test.MockCat may conflict with Prelude.any. To resolve this, hide any from Prelude or use a qualified name.

import Prelude hiding (any)
-- or
import qualified Test.MockCat as MC

Name collision with Control.Monad.when

Test.MockCat exports when (parameter matcher), which may conflict with Control.Monad.when. To avoid this, hide when from Test.MockCat or use qualified import.

import Test.MockCat hiding (when)
-- or
import Control.Monad hiding (when) -- if you want to use the matcher

Ambiguous types with OverloadedStrings

If you have OverloadedStrings enabled, string literals may cause ambiguity errors. Add explicit type annotations to resolve this.

mock (("value" :: String) ~> True)

Tested Versions

mockcat is continuously tested in CI across these configurations:

GHC Cabal OS
9.2.8 3.10.3.0 / 3.12.1.0 Ubuntu, macOS, Windows
9.4.8 3.10.3.0 / 3.12.1.0 Ubuntu, macOS, Windows
9.6.7 3.12.1.0 Ubuntu, macOS, Windows
9.8.4 3.12.1.0 Ubuntu, macOS, Windows
9.10.3 3.12.1.0 Ubuntu, macOS, Windows
9.12.2 3.12.1.0 Ubuntu, macOS, Windows

Happy Mocking! 🐱

About

Declarative mocking with a single arrow ~>. Architecture-agnostic and intent-driven.

Resources

Stars

7 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages