registry-0.1.3.0: test/Test/Data/Registry/RIOSpec.hs

{-# LANGUAGE ScopedTypeVariables #-}
{-# OPTIONS_GHC -fno-warn-missing-signatures #-}
{-# OPTIONS_GHC -fno-warn-deprecations #-}

module Test.Data.Registry.RIOSpec where

import           Data.IORef
import           Data.Registry
import           Protolude             as P
import           Test.Tasty.Extensions

test_close_resource = prop "RIO resources must be closed, with warmup" $ do
  ref <- liftIO $ newIORef []
  let rio = do
        ref' <- allocate (pure ref) (\ref' -> modifyIORef ref' (<>["close"]))
        warmupWith (warmupOf ("test"::Text) $ modifyIORef ref (<>["start"]))
        pure ref'

  res <- liftIO $ withRIO rio $ \ref' -> modifyIORef ref' (<>["use"])
  content <- liftIO $ readIORef ref

  isSuccess res === True
  content === ["start", "use", "close" :: Text]

test_close_resource_no_warmup = prop "RIO resources must be closed" $ do

  let rio = do
        ref <- allocate (newIORef []) (\ref -> modifyIORef ref (<>["close"]))
        liftIO $ modifyIORef ref (<>["start"])
        pure ref

  ref <- liftIO $ withNoWarmupRIO rio $ \ref -> modifyIORef ref (<>["use"]) $> ref
  content <- liftIO $ readIORef ref
  content === ["start", "use", "close" :: Text]