SubscriptionCheckpointInitializationMock.hs

module Test.SubscriptionCheckpointInitializationMock (spec) where

import Control.Monad.IO.Class (liftIO)
import Data.IORef (IORef, modifyIORef', newIORef, readIORef)
import Effectful (Eff, IOE, runEff, (:>))
import Effectful.Dispatch.Dynamic (interpret_)
import Kiroku.Store.Effect (Store (..))
import Kiroku.Store.Subscription (initializeSubscriptionCheckpoint)
import Kiroku.Store.Subscription.Types
import Kiroku.Store.Types (GlobalPosition (..))
import Test.Hspec

spec :: Spec
spec = describe "subscription checkpoint initialization mock interpreter" $ do
  it "carries the closed policy and typed result through one Store effect call" $ do
    calls <- newIORef (0 :: Int)
    let name = SubscriptionName "mock-initialization"
        key = SubscriptionCheckpointKey name 4
        expected = Right (InitializedCheckpoint FromCurrentHead key (GlobalPosition 23))
    actual <-
      runEff $
        runInitializationMock calls name 4 FromCurrentHead expected $
          initializeSubscriptionCheckpoint name 4 FromCurrentHead
    actual `shouldBe` expected
    readIORef calls `shouldReturn` 1

runInitializationMock ::
  (IOE :> es) =>
  IORef Int ->
  SubscriptionName ->
  Int ->
  MissingCheckpointPolicy ->
  Either SubscriptionCheckpointMissing CheckpointInitialization ->
  Eff (Store : es) a ->
  Eff es a
runInitializationMock calls expectedName expectedMember expectedPolicy expected = interpret_ $ \case
  InitializeSubscriptionCheckpoint actualName actualMember actualPolicy -> do
    liftIO $ do
      actualName `shouldBe` expectedName
      fromIntegral actualMember `shouldBe` expectedMember
      actualPolicy `shouldBe` expectedPolicy
      modifyIORef' calls (+ 1)
    pure expected
  _ -> error "unexpected Store operation in checkpoint initialization mock"