hasql-1.10: src/library-tests/Sharing/ByFeature/DecoderCompatibilityCheckSpec.hs
module Sharing.ByFeature.DecoderCompatibilityCheckSpec (spec) where
import Data.Either
import Data.Vector qualified as Vector
import Hasql.Connection qualified as Connection
import Hasql.Decoders qualified as Decoders
import Hasql.Errors qualified as Errors
import Hasql.Pipeline qualified as Pipeline
import Hasql.Session qualified as Session
import Hasql.Statement qualified as Statement
import Helpers.Scripts qualified as Scripts
import Test.Hspec
import Prelude
spec :: SpecWith (Text, Word16)
spec = parallel do
byPreparedStatusAndExecutor True "Session" (Session.statement ())
byPreparedStatusAndExecutor False "Session" (Session.statement ())
byPreparedStatusAndExecutor True "Pipeline" (Session.pipeline . Pipeline.statement ())
byPreparedStatusAndExecutor False "Pipeline" (Session.pipeline . Pipeline.statement ())
byPreparedStatusAndExecutor ::
Bool ->
Text ->
(forall a. (Show a) => Statement.Statement () a -> Session.Session a) ->
SpecWith (Text, Word16)
byPreparedStatusAndExecutor preparable executorName executor = do
describe (if preparable then "Preparable" else "Unpreparable") do
describe (toList executorName) do
describe "UnexpectedColumnCount" do
it "gets reported when result has more columns" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select 1, 2"
mempty
(Decoders.singleRow (Decoders.column (Decoders.nonNullable Decoders.int8)))
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnCountStatementError expected actual)) -> do
shouldBe expected 1
shouldBe actual 2
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
it "gets reported when result has fewer columns" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select 1"
mempty
( Decoders.singleRow
( (,)
<$> Decoders.column (Decoders.nonNullable Decoders.int8)
<*> Decoders.column (Decoders.nonNullable Decoders.int8)
)
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnCountStatementError expected actual)) -> do
shouldBe expected 2
shouldBe actual 1
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
describe "DecoderTypeMismatch" do
describe "singleRow" do
it "gets reported when column type mismatches decoder" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select 1::int8, 'text'"
mempty
( Decoders.singleRow
( (,)
<$> Decoders.column (Decoders.nonNullable Decoders.int8)
<*> Decoders.column (Decoders.nonNullable Decoders.int8)
)
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnTypeStatementError column expected actual)) -> do
shouldBe column 1
(expected, actual) `shouldBe` (20, 25)
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
describe "rowMaybe" do
it "gets reported when column type mismatches decoder" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select 1::int8, 'text'"
mempty
( Decoders.rowMaybe
( (,)
<$> Decoders.column (Decoders.nonNullable Decoders.int8)
<*> Decoders.column (Decoders.nonNullable Decoders.int8)
)
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnTypeStatementError column expected actual)) -> do
shouldBe column 1
(expected, actual) `shouldBe` (20, 25)
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
describe "rowVector" do
it "gets reported when column type mismatches decoder" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select int8 '1', text 'text'"
mempty
( Decoders.rowVector
( (,)
<$> Decoders.column (Decoders.nonNullable Decoders.int8)
<*> Decoders.column (Decoders.nonNullable Decoders.int8)
)
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnTypeStatementError column expected actual)) -> do
shouldBe column 1
(expected, actual) `shouldBe` (20, 25)
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
describe "array" do
describe "decoder:int8[]" do
describe "column:int8" do
it "reports properly" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select 1::int8"
mempty
( Decoders.singleRow
(Decoders.column (Decoders.nonNullable (Decoders.vectorArray @Vector (Decoders.nonNullable Decoders.int8))))
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnTypeStatementError column expected actual)) -> do
shouldBe column 0
(expected, actual) `shouldBe` (1016, 20)
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)
describe "decoder:int8[]" do
describe "column:int8[]" do
it "decodes properly" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select ARRAY[1::int8, 2::int8]"
mempty
( Decoders.singleRow
(Decoders.column (Decoders.nonNullable (Decoders.vectorArray @Vector (Decoders.nonNullable Decoders.int8))))
)
result <- Connection.use connection (executor statement)
shouldBe result (Right (Vector.fromList [1, 2]))
describe "decoder:int8" do
describe "column:int8[]" do
it "reports properly" \config -> do
Scripts.onPreparableConnection config \connection -> do
let statement =
(if preparable then Statement.preparable else Statement.unpreparable)
"select ARRAY[1::int8, 2::int8]"
mempty
( Decoders.singleRow
(Decoders.column (Decoders.nonNullable Decoders.int8))
)
result <- Connection.use connection (executor statement)
case result of
Left (Errors.StatementSessionError _ _ _ _ _ (Errors.UnexpectedColumnTypeStatementError column expected actual)) -> do
shouldBe column 0
(expected, actual) `shouldBe` (20, 1016)
Left err ->
expectationFailure ("Unexpected type of error: " <> show err)
result ->
expectationFailure ("Not an error: " <> show result)