{-# LANGUAGE OverloadedStrings #-}
module ResponseSpec (main, spec) where
import Control.Concurrent (threadDelay)
import qualified Data.ByteString as S
import qualified Data.ByteString.Char8 as S8
import Data.Maybe (mapMaybe)
import Network (connectTo, PortID (PortNumber))
import Network.HTTP.Types
import Network.Wai
import Network.Wai.Handler.Warp
import Network.Wai.Handler.Warp.Response
import RunSpec (withApp)
import System.IO (hClose, hFlush)
import Test.Hspec
main :: IO ()
main = hspec spec
testRange :: S.ByteString
-> String
-> String
-> Spec
testRange range out crange = it title $ withApp defaultSettings app $ \port -> do
handle <- connectTo "127.0.0.1" $ PortNumber $ fromIntegral port
S.hPutStr handle "GET / HTTP/1.0\r\n"
S.hPutStr handle "Range: bytes="
S.hPutStr handle range
S.hPutStr handle "\r\n\r\n"
hFlush handle
threadDelay 10000
bss <- fmap (lines . filter (/= '\r') . S8.unpack) $ S.hGetSome handle 1024
hClose handle
out `shouldBe` last bss
let hs = mapMaybe toHeader bss
lookup "Content-Range" hs `shouldBe` Just ("bytes " ++ crange)
lookup "Content-Length" hs `shouldBe` Just (show $ length $ last bss)
where
app _ = ($ responseFile status200 [] "attic/hex" Nothing)
title = show (range, out, crange)
toHeader s =
case break (== ':') s of
(x, ':':y) -> Just (x, dropWhile (== ' ') y)
_ -> Nothing
testPartial :: Integer
-> Integer
-> Integer
-> String
-> Spec
testPartial size offset count out = it title $ withApp defaultSettings app $ \port -> do
handle <- connectTo "127.0.0.1" $ PortNumber $ fromIntegral port
S.hPutStr handle "GET / HTTP/1.0\r\n\r\n"
hFlush handle
threadDelay 10000
bss <- fmap (lines . filter (/= '\r') . S8.unpack) $ S.hGetSome handle 1024
hClose handle
out `shouldBe` last bss
let hs = mapMaybe toHeader bss
lookup "Content-Length" hs `shouldBe` Just (show $ length $ last bss)
lookup "Content-Range" hs `shouldBe` Just range
where
app _ = ($ responseFile status200 [] "attic/hex" $ Just $ FilePart offset count size)
title = show (offset, count, out)
toHeader s =
case break (== ':') s of
(x, ':':y) -> Just (x, dropWhile (== ' ') y)
_ -> Nothing
range = "bytes " ++ show offset ++ "-" ++ show (offset + count - 1) ++ "/" ++ show size
testFileRange :: String
-> Status -> ResponseHeaders -> FilePath
-> Maybe FilePart -> Maybe HeaderValue
-> Either String (Status, ResponseHeaders, Integer, Integer)
-> Spec
testFileRange desc s rsphdr file mPart mRange ans = it desc $ do
eres <- fileRange s rsphdr file mPart mRange
let res = case eres of
Left e -> Left $ show e
Right r -> Right r
res `shouldBe` ans
spec :: Spec
spec = do
describe "range requests" $ do
testRange "2-3" "23" "2-3/16"
testRange "5-" "56789abcdef" "5-15/16"
testRange "5-8" "5678" "5-8/16"
testRange "-3" "def" "13-15/16"
describe "partial files" $ do
testPartial 16 2 2 "23"
testPartial 16 0 2 "01"
testPartial 16 3 8 "3456789a"
describe "fileRange" $ do
testFileRange
"gets a file size from file system"
status200 [] "attic/hex" Nothing Nothing
$ Right (status200,[("Content-Length","16")],0,16)
testFileRange
"gets an error if a file does not exist"
status200 [] "attic/nonexist" Nothing Nothing
$ Left "attic/nonexist: getFileStatus: does not exist (No such file or directory)"
testFileRange
"changes status if FileParts is specified"
status200 [] "attic/hex" (Just (FilePart 2 10 16)) Nothing
$ Right (status206,[("Content-Range", "bytes 2-11/16"),("Content-Length","10")],2,10)
testFileRange
"does not change status and does not add Content-Range if FileParts means the entire"
status200 [] "attic/hex" (Just (FilePart 0 16 16)) Nothing
$ Right (status200,[("Content-Length","16")],0,16)
testFileRange
"gets a file size from file system and handles Range and returns Partical Content"
status200 [] "attic/hex" Nothing (Just "bytes=2-14")
$ Right (status206,[("Content-Range","bytes 2-14/16"),("Content-Length","13")],2,13)
testFileRange
"gets a file size from file system and handles Range and returns OK if Range means the entire"
status200 [] "attic/hex" Nothing (Just "bytes=0-15")
$ Right (status200,[("Content-Length","16")],0,16)
testFileRange
"igores Range if FilePart is specified"
status200 [] "attic/hex" (Just (FilePart 2 10 16)) (Just "bytes=8-9")
$ Right (status206,[("Content-Range", "bytes 2-11/16"),("Content-Length","10")],2,10)