-- | Coverage for 'Pqi.consumeInput': reading server input into the
-- driver's buffer so that 'Pqi.isBusy' can settle before collecting
-- results.
module Pqi.Conformance.Operation.ConsumeInput
  ( spec,
  )
where

import qualified Pqi
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import Pqi.Conformance.Scenario (drainResults)
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
"consumeInput" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"drives result collection together with isBusy" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Bool, Bool, [ResultObservation]))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        sent <- Connection -> ByteString -> IO Bool
Pqi.sendQuery Connection
connection ByteString
"select 42"
        let settle (Int
0 :: Int) = Bool -> IO Bool
forall a. a -> IO a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Bool
False
            settle Int
n = do
              consumed <- Connection -> IO Bool
Pqi.consumeInput Connection
connection
              busy <- Pqi.isBusy connection
              if busy then threadDelay 1000 >> settle (n - 1) else pure consumed
        consumed <- settle 10000
        results <- drainResults connection
        pure (sent, consumed, results)