hsaml2-0.2.0: test/XML/Canonical.hs
{-# LANGUAGE OverloadedStrings #-}
module XML.Canonical (tests) where
import qualified Data.ByteString as BS
import qualified Data.ByteString.UTF8 as BSU
import qualified Data.ByteString.Lazy.UTF8 as BSLU
import qualified Test.HUnit as U
import qualified Text.XML.HXT.Core as HXT
import System.IO.Unsafe (unsafePerformIO)
import Control.Monad
import Data.Either (isRight)
import SAML2.XML
import SAML2.XML.Canonical
canonicalizeXML :: CanonicalizationAlgorithm -> String -> Bool -> IO [BS.ByteString]
canonicalizeXML algo f ent = HXT.runX
(HXT.readDocument [HXT.withCheckNamespaces HXT.yes, HXT.withValidate HXT.no, HXT.withCanonicalize HXT.no, HXT.withSubstDTDEntities ent] f
HXT.>>> HXT.arrIO (canonicalize algo Nothing Nothing))
testC14N :: CanonicalizationAlgorithm -> String -> Bool -> BS.ByteString -> U.Test
testC14N algo f ent s = U.TestCase $
U.assertEqual (show algo ++ ' ' : f) [s] =<< canonicalizeXML algo f ent
testIdempotency :: CanonicalizationAlgorithm -> String -> U.Test
testIdempotency algo input = U.TestCase $ do
U.assertBool (show algo ++ "[1] " ++ input) (isRight (go input))
U.assertBool (show algo ++ "[2] " ++ input) (go input == (go >=> go) input)
where
go :: String -> Either String String
go = fmap (BSU.toString . unsafePerformIO . canonicalizeWithRoot algo Nothing Nothing . (: [])) . xmlToDocE . BSLU.fromString
tests :: U.Test
tests = U.test
[ testC14N (CanonicalXML10 False) "test/XML/noncanonical1.xml" False
"<?xml-stylesheet href=\"doc.xsl\"\n type=\"text/xsl\" ?>\n<doc>Hello, world!</doc>\n<?pi-without-data?>"
, testC14N (CanonicalXML10 True) "test/XML/noncanonical1.xml" False
"<?xml-stylesheet href=\"doc.xsl\"\n type=\"text/xsl\" ?>\n<doc>Hello, world!<!-- Comment 1 --></doc>\n<?pi-without-data?>\n<!-- Comment 2 -->\n<!-- Comment 3 -->"
, testC14N (CanonicalXML10 False) "test/XML/noncanonical2.xml" False
"<doc>\n <clean> </clean>\n <dirty> A B </dirty>\n <mixed>\n A\n <clean> </clean>\n B\n <dirty> A B </dirty>\n C\n </mixed>\n</doc>"
, testC14N (CanonicalXML10 False) "test/XML/noncanonical3.xml" False
"<doc>\n <e1></e1>\n <e2></e2>\n <e3 id=\"elem3\" name=\"elem3\"></e3>\n <e4 id=\"elem4\" name=\"elem4\"></e4>\n <e5 xmlns=\"http://example.org\" xmlns:a=\"http://www.w3.org\" xmlns:b=\"http://www.ietf.org\" attr=\"I'm\" attr2=\"all\" b:attr=\"sorted\" a:attr=\"out\"></e5>\n <e6 xmlns:a=\"http://www.w3.org\">\n <e7 xmlns=\"http://www.ietf.org\">\n <e8 xmlns=\"\">\n <e9 xmlns:a=\"http://www.ietf.org\" attr=\"default\"></e9>\n </e8>\n </e7>\n </e6>\n</doc>"
, testC14N (CanonicalXML10 False) "test/XML/noncanonical4.xml" False
"<doc>\n <text>First line
\nSecond line</text>\n <value>2</value>\n <compute>value>\"0\" && value<\"10\" ?\"valid\":\"error\"</compute>\n <compute expr=\"value>"0" && value<"10" ?"valid":"error"\">valid</compute>\n <norm attr=\" ' 
	 ' \"></norm>\n <normNames attr=\"A 
	 B\"></normNames>\n <normId id=\"' 
	 '\"></normId>\n</doc>"
, testC14N (CanonicalXML10 False) "test/XML/noncanonical5.xml" True
"<doc attrExtEnt=\"entExt\">\n Hello, world!\n</doc>"
, testC14N (CanonicalXML10 False) "test/XML/noncanonical6.xml" False
"<doc>\194\169</doc>"
-- , testC14N (CanonicalXML10 False) "test/XML/noncanonical7.xml"
, testIdempotency (CanonicalXMLExcl10 False) "<a>\n <b>\n\n\nwef </b></a>"
]