6dd2418476
Moved `MediaBag` definition and functions from Shared: `lookupMedia`, `mediaDirectory`, `insertMedia`, `extractMediaBag`. Removed `emptyMediaBag`; use `mempty` instead, since `MediaBag` is a Monoid.
207 lines
7.3 KiB
Haskell
207 lines
7.3 KiB
Haskell
module Tests.Readers.Docx (tests) where
|
|
|
|
import Text.Pandoc.Options
|
|
import Text.Pandoc.Readers.Native
|
|
import Text.Pandoc.Definition
|
|
import Tests.Helpers
|
|
import Test.Framework
|
|
import qualified Data.ByteString as BS
|
|
import qualified Data.ByteString.Lazy as B
|
|
import qualified Data.ByteString.Char8 as B8
|
|
import qualified Data.ByteString.Base64 as B64
|
|
import Text.Pandoc.Readers.Docx
|
|
import Text.Pandoc.Writers.Native (writeNative)
|
|
import qualified Data.Map as M
|
|
import Text.Pandoc.MediaBag (lookupMedia)
|
|
|
|
-- We define a wrapper around pandoc that doesn't normalize in the
|
|
-- tests. Since we do our own normalization, we want to make sure
|
|
-- we're doing it right.
|
|
|
|
data NoNormPandoc = NoNormPandoc {unNoNorm :: Pandoc}
|
|
deriving Show
|
|
|
|
noNorm :: Pandoc -> NoNormPandoc
|
|
noNorm = NoNormPandoc
|
|
|
|
instance ToString NoNormPandoc where
|
|
toString d = writeNative def{ writerStandalone = s } $ toPandoc d
|
|
where s = case d of
|
|
NoNormPandoc (Pandoc (Meta m) _)
|
|
| M.null m -> False
|
|
| otherwise -> True
|
|
|
|
instance ToPandoc NoNormPandoc where
|
|
toPandoc = unNoNorm
|
|
|
|
compareOutput :: ReaderOptions
|
|
-> FilePath
|
|
-> FilePath
|
|
-> IO (NoNormPandoc, NoNormPandoc)
|
|
compareOutput opts docxFile nativeFile = do
|
|
df <- B.readFile docxFile
|
|
nf <- Prelude.readFile nativeFile
|
|
let (p, _) = readDocx opts df
|
|
return $ (noNorm p, noNorm (readNative nf))
|
|
|
|
testCompareWithOptsIO :: ReaderOptions -> String -> FilePath -> FilePath -> IO Test
|
|
testCompareWithOptsIO opts name docxFile nativeFile = do
|
|
(dp, np) <- compareOutput opts docxFile nativeFile
|
|
return $ test id name (dp, np)
|
|
|
|
testCompareWithOpts :: ReaderOptions -> String -> FilePath -> FilePath -> Test
|
|
testCompareWithOpts opts name docxFile nativeFile =
|
|
buildTest $ testCompareWithOptsIO opts name docxFile nativeFile
|
|
|
|
testCompare :: String -> FilePath -> FilePath -> Test
|
|
testCompare = testCompareWithOpts def
|
|
|
|
testCompareMediaIO :: String -> FilePath -> FilePath -> FilePath -> IO Test
|
|
testCompareMediaIO name docxFile mediaPath mediaFile = do
|
|
df <- B.readFile docxFile
|
|
mf <- B.readFile mediaFile
|
|
let (_, mb) = readDocx def df
|
|
dBytes = case lookupMedia mediaPath mb of
|
|
Just (_,bs) -> bs
|
|
Nothing -> error "Media file not found"
|
|
d64 = B8.unpack $ B64.encode $ BS.concat $ B.toChunks dBytes
|
|
m64 = B8.unpack $ B64.encode $ BS.concat $ B.toChunks mf
|
|
return $ test id name (d64, m64)
|
|
|
|
testCompareMedia :: String -> FilePath -> FilePath -> FilePath -> Test
|
|
testCompareMedia name docxFile mediaPath mediaFile =
|
|
buildTest $ testCompareMediaIO name docxFile mediaPath mediaFile
|
|
|
|
|
|
tests :: [Test]
|
|
tests = [ testGroup "inlines"
|
|
[ testCompare
|
|
"font formatting"
|
|
"docx.inline_formatting.docx"
|
|
"docx.inline_formatting.native"
|
|
, testCompare
|
|
"hyperlinks"
|
|
"docx.links.docx"
|
|
"docx.links.native"
|
|
, testCompare
|
|
"inline image with reference output"
|
|
"docx.image.docx"
|
|
"docx.image_no_embed.native"
|
|
, testCompare
|
|
"handling unicode input"
|
|
"docx.unicode.docx"
|
|
"docx.unicode.native"
|
|
, testCompare
|
|
"literal tabs"
|
|
"docx.tabs.docx"
|
|
"docx.tabs.native"
|
|
, testCompare
|
|
"normalizing inlines"
|
|
"docx.normalize.docx"
|
|
"docx.normalize.native"
|
|
, testCompare
|
|
"normalizing inlines deep inside blocks"
|
|
"docx.deep_normalize.docx"
|
|
"docx.deep_normalize.native"
|
|
, testCompare
|
|
"move trailing spaces outside of formatting"
|
|
"docx.trailing_spaces_in_formatting.docx"
|
|
"docx.trailing_spaces_in_formatting.native"
|
|
, testCompare
|
|
"inline code (with VerbatimChar style)"
|
|
"docx.inline_code.docx"
|
|
"docx.inline_code.native"
|
|
]
|
|
, testGroup "blocks"
|
|
[ testCompare
|
|
"headers"
|
|
"docx.headers.docx"
|
|
"docx.headers.native"
|
|
, testCompare
|
|
"headers already having auto identifiers"
|
|
"docx.already_auto_ident.docx"
|
|
"docx.already_auto_ident.native"
|
|
, testCompare
|
|
"lists"
|
|
"docx.lists.docx"
|
|
"docx.lists.native"
|
|
, testCompare
|
|
"definition lists"
|
|
"docx.definition_list.docx"
|
|
"docx.definition_list.native"
|
|
, testCompare
|
|
"footnotes and endnotes"
|
|
"docx.notes.docx"
|
|
"docx.notes.native"
|
|
, testCompare
|
|
"blockquotes (parsing indent as blockquote)"
|
|
"docx.block_quotes.docx"
|
|
"docx.block_quotes_parse_indent.native"
|
|
, testCompare
|
|
"hanging indents"
|
|
"docx.hanging_indent.docx"
|
|
"docx.hanging_indent.native"
|
|
, testCompare
|
|
"tables"
|
|
"docx.tables.docx"
|
|
"docx.tables.native"
|
|
, testCompare
|
|
"code block"
|
|
"docx.codeblock.docx"
|
|
"docx.codeblock.native"
|
|
|
|
]
|
|
, testGroup "track changes"
|
|
[ testCompare
|
|
"insertion (default)"
|
|
"docx.track_changes_insertion.docx"
|
|
"docx.track_changes_insertion_accept.native"
|
|
, testCompareWithOpts def{readerTrackChanges=AcceptChanges}
|
|
"insert insertion (accept)"
|
|
"docx.track_changes_insertion.docx"
|
|
"docx.track_changes_insertion_accept.native"
|
|
, testCompareWithOpts def{readerTrackChanges=RejectChanges}
|
|
"remove insertion (reject)"
|
|
"docx.track_changes_insertion.docx"
|
|
"docx.track_changes_insertion_reject.native"
|
|
, testCompare
|
|
"deletion (default)"
|
|
"docx.track_changes_deletion.docx"
|
|
"docx.track_changes_deletion_accept.native"
|
|
, testCompareWithOpts def{readerTrackChanges=AcceptChanges}
|
|
"remove deletion (accept)"
|
|
"docx.track_changes_deletion.docx"
|
|
"docx.track_changes_deletion_accept.native"
|
|
, testCompareWithOpts def{readerTrackChanges=RejectChanges}
|
|
"insert deletion (reject)"
|
|
"docx.track_changes_deletion.docx"
|
|
"docx.track_changes_deletion_reject.native"
|
|
, testCompareWithOpts def{readerTrackChanges=AllChanges}
|
|
"keep insertion (all)"
|
|
"docx.track_changes_deletion.docx"
|
|
"docx.track_changes_deletion_all.native"
|
|
, testCompareWithOpts def{readerTrackChanges=AllChanges}
|
|
"keep deletion (all)"
|
|
"docx.track_changes_deletion.docx"
|
|
"docx.track_changes_deletion_all.native"
|
|
]
|
|
, testGroup "media"
|
|
[ testCompareMedia
|
|
"image extraction"
|
|
"docx.image.docx"
|
|
"media/image1.jpeg"
|
|
"docx.image1.jpeg"
|
|
]
|
|
, testGroup "metadata"
|
|
[ testCompareWithOpts def{readerStandalone=True}
|
|
"metadata fields"
|
|
"docx.metadata.docx"
|
|
"docx.metadata.native"
|
|
, testCompareWithOpts def{readerStandalone=True}
|
|
"stop recording metadata with normal text"
|
|
"docx.metadata_after_normal.docx"
|
|
"docx.metadata_after_normal.native"
|
|
]
|
|
|
|
]
|
|
|