-- | Coverage for 'Pqi.flush': flushing queued output to the server in
-- non-blocking mode until it reports completion.
module Pqi.Conformance.Operation.Flush
  ( spec,
  )
where

import qualified Pqi
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import Pqi.Conformance.Scenario (drainResults, flushUntilDone)
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
"flush" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"flushes queued output to completion in non-blocking mode" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection
    -> IO (Bool, Bool, FlushStatus, [ResultObservation], Bool))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        setOk <- Connection -> Bool -> IO Bool
Pqi.setnonblocking Connection
connection Bool
True
        sent <- Pqi.sendQuery connection "select 1"
        flushed <- flushUntilDone (Pqi.flush connection)
        results <- drainResults connection
        restoreOk <- Pqi.setnonblocking connection False
        pure (setOk, sent, flushed, results, restoreOk)