module Test.Hspec.Wai.JSON (
json
, FromValue(..)
) where
import Control.Arrow (second)
import Data.List
import Data.ByteString.Lazy (ByteString)
import qualified Data.ByteString.Lazy as BL
import Data.Aeson (Value, encode)
import Data.Aeson.QQ
import qualified Data.CaseInsensitive as CI
import Language.Haskell.TH.Quote
import Test.Hspec.Wai
import Test.Hspec.Wai.Internal (formatHeader)
json :: QuasiQuoter
json = QuasiQuoter {
quoteExp = \input -> [|fromValue $(quoteExp aesonQQ input)|]
, quotePat = const $ error "No quotePat defined for Test.Hspec.Wai.JSON.json"
, quoteType = const $ error "No quoteType defined for Test.Hspec.Wai.JSON.json"
, quoteDec = const $ error "No quoteDec defined for Test.Hspec.Wai.JSON.json"
}
class FromValue a where
fromValue :: Value -> a
instance FromValue ResponseMatcher where
fromValue v = ResponseMatcher 200 [MatchHeader p] (Just body)
where
body = fromValue v
permissibleHeaders = addIfASCII ("Content-Type", "application/json") [("Content-Type", "application/json; charset=utf-8")]
addIfASCII h = if BL.all (< 128) body then (h :) else id
mkCI = map (second CI.mk)
p headers = if any (`elem` mkCI permissibleHeaders) (mkCI headers)
then Nothing
else (Just . unlines) ("missing header:" : (intersperse " OR" $ map formatHeader permissibleHeaders))
instance FromValue ByteString where
fromValue = encode