Lua: support custom bytestring readers.

1 files changed, 34 insertions, 0 deletions

diff --git a/pandoc-lua-engine/test/Tests/Lua/Reader.hs b/pandoc-lua-engine/test/Tests/Lua/Reader.hs
new file mode 100644
index 000000000..16474bd91
--- /dev/null
+++ b/pandoc-lua-engine/test/Tests/Lua/Reader.hs
@@ -0,0 +1,34 @@
+{-# LANGUAGE LambdaCase        #-}
+{- |
+Module      : Tests.Lua.Reader
+Copyright   : © 2022 Albert Krewinkel
+License     : GPL-2.0-or-later
+Maintainer  : Albert Krewinkel <pandoc@tarleb.com>
+
+Tests for custom Lua readers.
+-}
+module Tests.Lua.Reader (tests) where
+
+import Data.Char (chr)
+import Data.Default (Default (def))
+import Text.Pandoc.Class (runIOorExplode)
+import Text.Pandoc.Lua (readCustom)
+import Text.Pandoc.Readers (Reader (ByteStringReader, TextReader))
+import Test.Tasty (TestTree)
+import Test.Tasty.HUnit ((@?=), testCase)
+
+import qualified Data.ByteString.Lazy as BL
+import qualified Data.Text as T
+import qualified Text.Pandoc.Builder as B
+
+tests :: [TestTree]
+tests =
+  [ testCase "read binary to code block" $ do
+    input <- BL.readFile "bytestring.bin"
+    doc <- runIOorExplode $
+      readCustom "bytestring-reader.lua" >>= \case
+        ByteStringReader f -> f def input
+        TextReader {}      -> error "Expected a bytestring reader"
+    let bytes = mconcat $ map (B.str . T.singleton . chr) [0..255]
+    doc @?= B.doc (B.plain bytes)
+  ]