-- | Coverage for 'Pqi.transactionStatus': in-transaction status tracking
-- across a successful command, a failed command, and a rollback.
module Pqi.Conformance.Operation.TransactionStatus
  ( 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
"transactionStatus" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"tracks status through a transaction" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection
    -> IO
         (TransactionStatus, TransactionStatus, TransactionStatus,
          TransactionStatus))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        idle <- Connection -> IO TransactionStatus
Pqi.transactionStatus Connection
connection
        _ <- Pqi.exec connection "begin"
        inTransaction <- Pqi.transactionStatus connection
        _ <- Pqi.exec connection "select 1 / 0"
        inError <- Pqi.transactionStatus connection
        _ <- Pqi.exec connection "rollback"
        afterRollback <- Pqi.transactionStatus connection
        pure (idle, inTransaction, inError, afterRollback)