dfae661878
* Don't use hspecX, it's deprecated. * Do not import Test.Hspec.HUnit. It's no longer necessary and may be removed in the future.
51 lines
1.3 KiB
Haskell
51 lines
1.3 KiB
Haskell
{-# LANGUAGE QuasiQuotes, TypeFamilies, TemplateHaskell, MultiParamTypeClasses #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# LANGUAGE FlexibleInstances #-}
|
|
module YesodCoreTest.Exceptions (exceptionsTest, Widget) where
|
|
|
|
import Test.Hspec
|
|
|
|
import Yesod.Core hiding (Request)
|
|
import Network.Wai
|
|
import Network.Wai.Test
|
|
import Network.HTTP.Types (status301)
|
|
|
|
data Y = Y
|
|
mkYesod "Y" [parseRoutes|
|
|
/ RootR GET
|
|
/redirect RedirR GET
|
|
|]
|
|
|
|
instance Yesod Y where
|
|
approot = ApprootStatic "http://test"
|
|
errorHandler (InternalError e) = return $ chooseRep $ RepPlain $ toContent e
|
|
errorHandler x = defaultErrorHandler x
|
|
|
|
getRootR :: Handler ()
|
|
getRootR = error "FOOBAR" >> return ()
|
|
|
|
getRedirR :: Handler ()
|
|
getRedirR = do
|
|
setHeader "foo" "bar"
|
|
redirectWith status301 RootR
|
|
|
|
exceptionsTest :: Spec
|
|
exceptionsTest = describe "Test.Exceptions" $ do
|
|
it "500" case500
|
|
it "redirect keeps headers" caseRedirect
|
|
|
|
runner :: Session () -> IO ()
|
|
runner f = toWaiApp Y >>= runSession f
|
|
|
|
case500 :: IO ()
|
|
case500 = runner $ do
|
|
res <- request defaultRequest
|
|
assertStatus 500 res
|
|
assertBody "FOOBAR" res
|
|
|
|
caseRedirect :: IO ()
|
|
caseRedirect = runner $ do
|
|
res <- request defaultRequest { pathInfo = ["redirect"] }
|
|
assertStatus 301 res
|
|
assertHeader "foo" "bar" res
|