-- | Coverage for 'Pqi.execPrepared': executing a prepared statement over
-- named and unnamed statements, parameter formats, and the unknown-statement
-- error path.
module Pqi.Conformance.Operation.ExecPrepared
  ( spec,
  )
where

import qualified Pqi as Lq
import Pqi.Conformance.Harness
import Pqi.Conformance.Observation
import Pqi.Conformance.Prelude
import Test.Hspec

spec :: Lq.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
"execPrepared" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"executes a prepared statement" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe ResultObservation))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        _ <- Connection
-> ByteString -> ByteString -> Maybe [Word32] -> IO (Maybe Result)
Lq.prepare Connection
connection ByteString
"conformance_stmt" ByteString
"select $1 :: text as a, $2 :: int4 as b" Maybe [Word32]
forall a. Maybe a
Nothing
        Lq.execPrepared connection "conformance_stmt" [Just ("hello", Lq.Text), Just ("7", Lq.Text)] Lq.Text
          >>= traverse observeResult

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"executes the unnamed prepared statement" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe ResultObservation))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        _ <- Connection
-> ByteString -> ByteString -> Maybe [Word32] -> IO (Maybe Result)
Lq.prepare Connection
connection ByteString
"" ByteString
"select $1 :: int4 + 1" Maybe [Word32]
forall a. Maybe a
Nothing
        Lq.execPrepared connection "" [Just ("41", Lq.Text)] Lq.Text
          >>= traverse observeResult

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"executes a zero-parameter statement" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe ResultObservation))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        _ <- Connection
-> ByteString -> ByteString -> Maybe [Word32] -> IO (Maybe Result)
Lq.prepare Connection
connection ByteString
"conformance_no_params" ByteString
"select 42" Maybe [Word32]
forall a. Maybe a
Nothing
        Lq.execPrepared connection "conformance_no_params" [] Lq.Text
          >>= traverse observeResult

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"binds null and binary parameters with a binary result" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe ResultObservation))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        _ <- Connection
-> ByteString -> ByteString -> Maybe [Word32] -> IO (Maybe Result)
Lq.prepare Connection
connection ByteString
"conformance_binary" ByteString
"select $1 :: text, $2 :: bytea" Maybe [Word32]
forall a. Maybe a
Nothing
        Lq.execPrepared connection "conformance_binary" [Nothing, Just ("\NUL\1\255", Lq.Binary)] Lq.Binary
          >>= traverse observeResult

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"rejects an unknown statement" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe ResultObservation))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection ->
        Connection
-> ByteString
-> [Maybe (ByteString, Format)]
-> Format
-> IO (Maybe Result)
Lq.execPrepared Connection
connection ByteString
"conformance_missing" [] Format
Lq.Text
          IO (Maybe Result)
-> (Maybe Result -> IO (Maybe ResultObservation))
-> IO (Maybe ResultObservation)
forall a b. IO a -> (a -> IO b) -> IO b
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= (Result -> IO ResultObservation)
-> Maybe Result -> IO (Maybe ResultObservation)
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> Maybe a -> f (Maybe b)
traverse Result -> IO ResultObservation
observeResult