servant/servant-server/test/Servant/Server/StaticFilesSpec.hs

72 lines
2.1 KiB
Haskell
Raw Permalink Normal View History

{-# LANGUAGE DataKinds #-}
2014-12-10 16:10:57 +01:00
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeOperators #-}
2014-12-10 16:10:57 +01:00
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Servant.Server.StaticFilesSpec where
2014-12-10 16:10:57 +01:00
2018-06-29 21:08:26 +02:00
import Control.Exception
(bracket)
import Data.Proxy
(Proxy (Proxy))
import Network.Wai
(Application)
import System.Directory
(createDirectory, getCurrentDirectory, setCurrentDirectory)
import System.IO.Temp
(withSystemTempDirectory)
import Test.Hspec
(Spec, around_, describe, it)
import Test.Hspec.Wai
(get, shouldRespondWith, with)
2014-12-10 16:10:57 +01:00
2018-06-29 21:08:26 +02:00
import Servant.API
((:<|>) ((:<|>)), (:>), Capture, Get, JSON, Raw)
import Servant.Server
(Server, serve)
import Servant.Server.StaticFiles
(serveDirectoryFileServer)
2018-06-29 21:08:26 +02:00
import Servant.ServerSpec
(Person (Person))
2014-12-10 16:10:57 +01:00
type Api =
2015-01-12 15:08:41 +01:00
"dummy_api" :> Capture "person_name" String :> Get '[JSON] Person
2014-12-10 16:10:57 +01:00
:<|> "static" :> Raw
api :: Proxy Api
api = Proxy
app :: Application
app = serve api server
2014-12-10 16:10:57 +01:00
server :: Server Api
server =
2015-01-06 17:26:37 +01:00
(\ name_ -> return (Person name_ 42))
2017-01-09 01:02:32 +01:00
:<|> serveDirectoryFileServer "static"
2014-12-10 16:10:57 +01:00
withStaticFiles :: IO () -> IO ()
withStaticFiles action = withSystemTempDirectory "servant-test" $ \ tmpDir ->
bracket (setup tmpDir) teardown (const action)
where
setup tmpDir = do
outer <- getCurrentDirectory
setCurrentDirectory tmpDir
createDirectory "static"
writeFile "static/foo.txt" "bar"
writeFile "static/index.html" "index"
return outer
teardown outer = do
setCurrentDirectory outer
spec :: Spec
spec = do
around_ withStaticFiles $ with (return app) $ do
describe "serveDirectory" $ do
it "successfully serves files" $ do
get "/static/foo.txt" `shouldRespondWith` "bar"
it "serves the contents of index.html when requesting the root of a directory" $ do
get "/static/" `shouldRespondWith` "index"