{-# LANGUAGE OverloadedStrings #-}
module YesodCoreTest.Streaming (specs) where

import Yesod.Core
import Test.Hspec
import Network.Wai.Test
import Data.Text (Text)
import Data.ByteString (ByteString)

app :: LiteApp
app = liteApp $ dispatchTo $ respondSource typeHtml $ do
    sendChunk ("Hello " :: String)
    sendChunk ("World" :: ByteString)
    sendChunk ("!\n" :: Text)
    sendChunkHtml "<&>"

test :: String
     -> (SResponse -> Session ())
     -> Spec
test name f = it name $ do
    wapp <- toWaiApp app
    flip runSession wapp $ do
        sres <- request defaultRequest
        f sres

specs :: Spec
specs = describe "Streaming" $ do
    test "works" $ \sres -> do
        assertStatus 200 sres
        assertBody "Hello World!\n&lt;&amp;&gt;" sres