-- | Coverage for 'Pqi.pipelineStatus': the pipeline-mode status,
-- including the aborted state after an in-pipeline error and recovery at the
-- sync point.
module Pqi.Conformance.Operation.PipelineStatus
  ( spec,
  )
where

import qualified Pqi
import qualified Pqi as Lq
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import Pqi.Conformance.Scenario (takeCommandResults, takeResult)
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
"pipelineStatus" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"reports off, on, aborted, and recovery" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection
    -> IO
         (PipelineStatus, PipelineStatus, PipelineStatus, PipelineStatus))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        off <- Connection -> IO PipelineStatus
Lq.pipelineStatus Connection
connection
        _ <- Lq.enterPipelineMode connection
        on <- Lq.pipelineStatus connection
        _ <- traverse (\ByteString
sql -> Connection
-> ByteString
-> [Maybe (Word32, ByteString, Format)]
-> Format
-> IO Bool
Lq.sendQueryParams Connection
connection ByteString
sql [] Format
Lq.Text) ["select 1", "select 1 / 0", "select 3"]
        _ <- Lq.pipelineSync connection
        _ <- takeCommandResults connection
        _ <- takeCommandResults connection
        aborted <- Lq.pipelineStatus connection
        _ <- takeCommandResults connection
        _ <- takeResult connection
        recovered <- Lq.pipelineStatus connection
        _ <- Lq.exitPipelineMode connection
        pure (off, on, aborted, recovered)