-- | Coverage for 'Pqi.loImportWithOid': importing a file as a new large
-- object with an explicitly requested OID.
--
-- Each run removes the object it creates, so the explicit OID is free for the
-- reference run and can be compared in full. @lo_import@ is self-contained and
-- runs in autocommit; it needs no explicit transaction block.
module Pqi.Conformance.Operation.LoImportWithOid
  ( 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
"loImportWithOid" 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 with an explicit OID" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe Word32, Maybe (Maybe ())))
-> 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-oid"
        ByteString.hPut handle "pqi conformance payload"
        hClose handle
        _ <- Pqi.loUnlink connection explicitOid
        imported <- Pqi.loImportWithOid connection path explicitOid
        unlinked <- for imported (Pqi.loUnlink connection)
        removeFile path
        pure (imported, unlinked)
  where
    explicitOid :: a
explicitOid = a
424243