-- | Coverage for 'Pqi.escapeStringConn': escaping strings for SQL
-- literals, the invalid-encoding error path, and a round-trip through a query.
module Pqi.Conformance.Operation.EscapeStringConn
  ( spec,
  )
where

import qualified Pqi
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import Pqi.Conformance.Scenario (execScenario)
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
"escapeStringConn" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"escapes a range of strings" \ByteString
conninfo ->
      Adapter
-> ByteString -> (Connection -> IO [Maybe ByteString]) -> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection ->
        (ByteString -> IO (Maybe ByteString))
-> [ByteString] -> IO [Maybe ByteString]
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) -> [a] -> f [b]
traverse (Connection -> ByteString -> IO (Maybe ByteString)
Pqi.escapeStringConn Connection
connection) [ByteString]
forall {a}. IsString a => [a]
stringCases

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"rejects invalid encoding" \ByteString
conninfo ->
      Adapter
-> ByteString -> (Connection -> IO (Maybe ByteString)) -> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection ->
        Connection -> ByteString -> IO (Maybe ByteString)
Pqi.escapeStringConn Connection
connection ByteString
"\255\254"

    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"produces literals that round-trip through a query" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe (Maybe ResultObservation)))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        escaped <- Connection -> ByteString -> IO (Maybe ByteString)
Pqi.escapeStringConn Connection
connection ByteString
"it's \\ tricky\nstuff"
        for escaped \ByteString
literal ->
          ByteString -> Connection -> IO (Maybe ResultObservation)
execScenario (ByteString
"select '" ByteString -> ByteString -> ByteString
forall a. Semigroup a => a -> a -> a
<> ByteString
literal ByteString -> ByteString -> ByteString
forall a. Semigroup a => a -> a -> a
<> ByteString
"' :: text") Connection
connection
  where
    stringCases :: [a]
stringCases =
      [ a
"",
        a
"plain",
        a
"it's",
        a
"back\\slash",
        a
"newline\nand\ttab",
        a
"héllo🐘",
        a
"double''single"
      ]