-- | Coverage for 'Pqi.loExport': exporting a large object back to a file,
-- round-tripping its bytes.
--
-- @lo_import@\/@lo_export@ are self-contained and run in autocommit; they need
-- no explicit transaction block.
module Pqi.Conformance.Operation.LoExport
  ( 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
"loExport" do
    String
-> (ByteString -> IO ()) -> SpecWith (Arg (ByteString -> IO ()))
forall a.
(HasCallStack, Example a) =>
String -> a -> SpecWith (Arg a)
it String
"exports an imported object, round-tripping its bytes" \ByteString
conninfo ->
      Adapter
-> ByteString
-> (Connection -> IO (Maybe (Maybe ()), Bool))
-> IO ()
forall a.
(Eq a, Show a, HasCallStack) =>
Adapter -> ByteString -> (Connection -> IO a) -> IO ()
differential Adapter
adapter ByteString
conninfo \Connection
connection -> do
        (importPath, importHandle) <- String -> String -> IO (String, Handle)
openBinaryTempFile String
"/tmp" String
"pqi-conformance-export-in"
        ByteString.hPut importHandle payload
        hClose importHandle
        (exportPath, exportHandle) <- openBinaryTempFile "/tmp" "pqi-conformance-export-out"
        hClose exportHandle
        imported <- Pqi.loImport connection importPath
        exported <- for imported \Word32
o -> Connection -> Word32 -> String -> IO (Maybe ())
Pqi.loExport Connection
connection Word32
o String
exportPath
        traverse_ (Pqi.loUnlink connection) imported
        roundTripped <- ByteString.readFile exportPath
        removeFile importPath
        removeFile exportPath
        pure (exported, roundTripped == payload)
  where
    payload :: ByteString
payload = ByteString
"pqi conformance payload\n" ByteString -> ByteString -> ByteString
forall a. Semigroup a => a -> a -> a
<> [Word8] -> ByteString
ByteString.pack [Word8
0 .. Word8
255]