SendQuery.hs

-- | Coverage for 'Pqi.sendQuery': submitting a command without waiting,
-- then collecting its results, including multi-statement scripts, a mid-script
-- error, and the empty query.
module Pqi.Conformance.Operation.SendQuery
  ( 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 =
  describe "sendQuery" do
    let sendAndDrain sql conninfo =
          differential adapter conninfo \connection -> do
            sent <- connection.sendQuery sql
            results <- drainResults connection
            pure (sent, results)
    it "sends a query and collects its result"
      $ sendAndDrain "select i, i * 10 from generate_series (1, 3) as i"
    it "yields multiple results for multiple statements"
      $ sendAndDrain "select 1; select 2, 3; select 4"
    it "ends the results at a mid-script error"
      $ sendAndDrain "select 1; select 1 / 0; select 3"
    it "handles an empty string"
      $ sendAndDrain ""