GetResult.hs

-- | Coverage for 'Pqi.getResult': collecting the results of an
-- asynchronous command one at a time, ending with the 'Nothing' terminator.
module Pqi.Conformance.Operation.GetResult
  ( spec,
  )
where

import qualified Pqi
import Pqi.Conformance.Harness
import qualified Pqi.Conformance.Operation.GetResult.LostConnection as LostConnection
import Pqi.Conformance.Prelude
import Pqi.Conformance.Scenario (takeResult)
import Test.Hspec

spec :: Pqi.Adapter -> SpecWith ByteString
spec adapter = do
  LostConnection.spec adapter

  describe "getResult" do
    it "yields each result then a Nothing terminator" \conninfo ->
      differential adapter conninfo \connection -> do
        _ <- Pqi.sendQuery connection "select 1 :: int4"
        first <- takeResult connection
        terminator <- takeResult connection
        afterTerminator <- takeResult connection
        pure (first, terminator, afterTerminator)