-- | Coverage for 'Pqi.loImport': importing a file as a new large object.
--
-- The server-assigned OID differs between runs, so only its presence is
-- compared. @lo_import@ is self-contained and runs in autocommit; it needs no
-- explicit transaction block.
module Pqi.Conformance.Operation.LoImport
  ( spec,
  )
where

import qualified Data.ByteString as ByteString
import qualified Pqi
import Pqi.Conformance.Harness
import Pqi.Conformance.Prelude
import System.Directory (removeFile)
import System.IO (hClose, openBinaryTempFile)
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
"loImport" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"imports a file as a large object" \ByteString
conninfo ->
      Adapter -> ByteString -> (Connection -> IO Bool) -> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        (path, handle) <- String -> String -> IO (String, Handle)
openBinaryTempFile String
"/tmp" String
"pqi-conformance-import"
        ByteString.hPut handle "pqi conformance payload"
        hClose handle
        imported <- Pqi.loImport connection path
        traverse_ (Pqi.loUnlink connection) imported
        removeFile path
        pure (isJust imported)