-- | Coverage for 'Pqi.disableNoticeReporting': turning off notice
-- accumulation again so subsequently raised notices are not retained.
module Pqi.Conformance.Operation.DisableNoticeReporting
  ( spec,
  )
where

import qualified Pqi
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import Test.Hspec

spec :: Pqi.Adapter -> SpecWith ByteString
spec :: Adapter -> SpecWith ByteString
spec Adapter
adapter =
  String -> SpecWith ByteString -> SpecWith ByteString
forall a. HasCallStack => String -> SpecWith a -> SpecWith a
describe String
"disableNoticeReporting" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"stops retaining notices once disabled" \ByteString
conninfo ->
      Adapter -> ByteString -> (Connection -> IO (Bool, Bool)) -> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        Connection -> IO ()
Pqi.enableNoticeReporting Connection
connection
        whileEnabled <- Connection -> IO Bool
raiseNoticeAndCollect Connection
connection
        Pqi.disableNoticeReporting connection
        whileDisabled <- raiseNoticeAndCollect connection
        pure (whileEnabled, whileDisabled)

raiseNoticeAndCollect :: Pqi.Connection -> IO Bool
raiseNoticeAndCollect :: Connection -> IO Bool
raiseNoticeAndCollect Connection
connection = do
  _ <- Connection -> ByteString -> IO (Maybe Result)
Pqi.exec Connection
connection ByteString
"do $$ begin raise notice 'conformance notice'; end $$"
  isJust <$> Pqi.getNotice connection