1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{- |
Module : Tests.Lua.Writer
Copyright : © 2019-2023 Albert Krewinkel
License : GNU GPL, version 2 or above
Maintainer : Albert Krewinkel <albert@zeitkraut.de>
Tests for custom Lua writers.
-}
module Tests.Lua.Writer (tests) where
import Data.Default (Default (def))
import Data.Maybe (fromMaybe)
import Text.Pandoc.Class (runIOorExplode, readFileStrict)
import Text.Pandoc.Extensions (Extension (..), extensionsFromList)
import Text.Pandoc.Format (ExtensionsDiff (..), FlavoredFormat (..),
applyExtensionsDiff)
import Text.Pandoc.Lua (loadCustom)
import Text.Pandoc.Options (WriterOptions (..))
import Text.Pandoc.Readers (readNative)
import Text.Pandoc.Scripting (CustomComponents (..))
import Text.Pandoc.Writers (Writer (ByteStringWriter, TextWriter))
import Test.Tasty (TestTree)
import Test.Tasty.Golden (goldenVsString)
import Test.Tasty.HUnit (testCase, (@?=))
import qualified Data.ByteString.Lazy as BL
import qualified Text.Pandoc.Builder as B
import qualified Text.Pandoc.UTF8 as UTF8
tests :: [TestTree]
tests =
[ goldenVsString "default testsuite"
"writer.custom"
(runIOorExplode $ do
source <- UTF8.toText <$> readFileStrict "testsuite.native"
doc <- readNative def source
txt <- customWriter <$> loadCustom "sample.lua" >>= \case
Just (TextWriter f) -> f def doc
_ -> error "Expected a text writer"
pure $ BL.fromStrict (UTF8.fromText txt))
, goldenVsString "tables testsuite"
"tables.custom"
(runIOorExplode $ do
source <- UTF8.toText <$> readFileStrict "tables.native"
doc <- readNative def source
txt <- writeCustom "sample.lua" >>= \case
(TextWriter f, _, _) -> f def doc
_ -> error "Expected a text writer"
pure $ BL.fromStrict (UTF8.fromText txt))
, goldenVsString "bytestring writer"
"bytestring.bin"
(runIOorExplode $
writeCustom "bytestring.lua" >>= \case
(ByteStringWriter f, _, _) -> f def mempty
_ -> error "Expected a bytestring writer")
, goldenVsString "template"
"writer-template.out.txt"
(runIOorExplode $ do
(_, _, template) <- writeCustom "writer-template.lua"
pure . BL.fromStrict . UTF8.fromText $ fromMaybe "" template)
, testCase "preset extensions" $ do
let format = FlavoredFormat "extensions.lua" mempty
result <- runIOorExplode $ writeCustom "extensions.lua" >>= \case
(TextWriter write, extsConf, _) -> do
exts <- applyExtensionsDiff extsConf format
write def{writerExtensions = exts} (B.doc mempty)
_ -> error "Expected a text writer"
result @?= "smart extension is enabled;\ncitations extension is disabled\n"
, testCase "modified extensions" $ do
let ediff = ExtensionsDiff
{ extsToEnable = extensionsFromList [Ext_citations]
, extsToDisable = mempty
}
let format = FlavoredFormat "extensions.lua" ediff
result <- runIOorExplode $ writeCustom "extensions.lua" >>= \case
(TextWriter write, extsConf, _) -> do
exts <- applyExtensionsDiff extsConf format
write def{writerExtensions = exts} (B.doc mempty)
_ -> error "Expected a text writer"
result @?= "smart extension is enabled;\ncitations extension is enabled\n"
]
where
writeCustom fp = do
components <- loadCustom fp
let exts = fromMaybe mempty (customExtensions components)
case customWriter components of
Nothing -> error "Expected a writer to be defined"
Just w -> return (w, exts, customTemplate components)
|
