131 lines
4.2 KiB
Haskell
131 lines
4.2 KiB
Haskell
{-
|
|
|
|
gemcap
|
|
|
|
Cooyright (C) Jonathan Lamothe <jonathan@jlamothe.net>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public
|
|
License along with this program. If not, see
|
|
<https://www.gnu.org/licenses/>.
|
|
|
|
-}
|
|
|
|
{-# LANGUAGE OverloadedStrings, RecordWildCards #-}
|
|
|
|
module Network.Gemini.Capsule.InternalSpec (spec) where
|
|
|
|
import qualified Data.ByteString as BS
|
|
import Data.Char (ord)
|
|
import Data.Connection (Connection (..))
|
|
import System.IO.Streams (nullInput, unRead)
|
|
import Test.Hspec (Spec, context, describe, it, shouldReturn, xit)
|
|
|
|
import Network.Gemini.Capsule.Types
|
|
import Network.Gemini.Capsule.Internal
|
|
|
|
spec :: Spec
|
|
spec = describe "Internal" $ do
|
|
readURLSpec
|
|
sendResponseSpec
|
|
strFromConnSpec
|
|
readMaxSpec
|
|
stripCRLFSpec
|
|
|
|
readURLSpec :: Spec
|
|
readURLSpec = describe "readURL" $ mapM_
|
|
( \(desc, ioConn, expect) -> context desc $
|
|
xit ("should return " ++ show expect) $
|
|
do
|
|
conn <- ioConn
|
|
readURL conn `shouldReturn` expect
|
|
)
|
|
|
|
-- description, connection, expected result
|
|
[ ( "valid URL", validConn, Just validExp )
|
|
, ( "long URL", longConn, Just longExp )
|
|
, ( "too long URL", tooLongConn, Nothing )
|
|
, ( "gibberish input", gibConn, Nothing )
|
|
]
|
|
|
|
where
|
|
validConn = mkInConn ["gemini://example.com/\r\n"]
|
|
longConn = mkInConn [longBS]
|
|
tooLongConn = mkInConn [tooLongBS]
|
|
gibConn = mkInConn ["aosidjfwoeinboijwefr"]
|
|
longBS = BS.pack (take 1024 bytes) <> "\r\n"
|
|
tooLongBS = BS.pack (take 1025 bytes) <> "\r\n"
|
|
bytes = BS.unpack prefix ++ repeat (fromIntegral $ ord 'A')
|
|
validExp = newGemURL "example.com"
|
|
longExp = validExp { gemPath = [longDir] }
|
|
longDir = replicate (1024 - BS.length prefix) 'A'
|
|
prefix = "gemini://example.com/"
|
|
|
|
sendResponseSpec :: Spec
|
|
sendResponseSpec = describe "sendResponse" $ return ()
|
|
|
|
strFromConnSpec :: Spec
|
|
strFromConnSpec = describe "strFromConn" $ mapM_
|
|
( \(desc, maxLen, ioConn, expect) -> context desc $
|
|
xit ("should return " ++ show expect) $ do
|
|
conn <- ioConn
|
|
strFromConn maxLen conn `shouldReturn` expect
|
|
)
|
|
|
|
-- description, max size, connection, expected
|
|
[ ( "valid string", 100, mkInConn ["foo\r\n"], Just "foo" )
|
|
, ( "long string", 5, mkInConn ["too long\r\n"], Nothing )
|
|
, ( "no CR/LF", 100, mkInConn ["foo"], Nothing )
|
|
, ( "bad UTF-8", 100, mkInConn ["foo\xff\r\n"], Nothing )
|
|
, ( "non-ASCII", 100, mkInConn ["\xc3\xa9\r\n"], Just "\xe9" )
|
|
]
|
|
|
|
readMaxSpec :: Spec
|
|
readMaxSpec = describe "readMax" $ mapM_
|
|
( \(desc, maxLen, ioConn, expect) -> context desc $
|
|
it ("should return " ++ show expect) $ do
|
|
conn <- ioConn
|
|
readMax maxLen conn `shouldReturn` expect
|
|
)
|
|
|
|
-- description, max length, connection, expected
|
|
[ ( "single input", 1024, singleConn, Just singleBS )
|
|
, ( "multi input", 1024, multiConn, Just multiBS )
|
|
, ( "long input", longLen, longConn, Just longBS )
|
|
, ( "too long", pred longLen, longConn, Nothing )
|
|
, ( "empty input", 1024, mkInConn [], Just "" )
|
|
]
|
|
|
|
where
|
|
singleConn = mkInConn ["foo"]
|
|
multiConn = mkInConn ["foo", "bar", "baz"]
|
|
longConn = mkInConn [longBS]
|
|
longLen = BS.length longBS
|
|
singleBS = "foo"
|
|
multiBS = "foobarbaz"
|
|
longBS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
stripCRLFSpec :: Spec
|
|
stripCRLFSpec = describe "stripCRLF" $ return ()
|
|
|
|
mkInConn :: [BS.ByteString] -> IO (Connection ())
|
|
mkInConn bss = do
|
|
source <- nullInput
|
|
mapM_ (`unRead` source) (reverse bss)
|
|
let
|
|
send = const $ return ()
|
|
close = return ()
|
|
connExtraInfo = ()
|
|
return Connection {..}
|
|
|
|
--jl
|